From 244e8d00d596b3d93b5bb92f3d2e76b71c719070 Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Wed, 11 Feb 2026 14:23:09 -0500 Subject: [PATCH] [Cute-DSL] Add cute.arch.fmin by calling nvvm --- python/CuTeDSL/cutlass/cute/arch/nvvm_wrappers.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/python/CuTeDSL/cutlass/cute/arch/nvvm_wrappers.py b/python/CuTeDSL/cutlass/cute/arch/nvvm_wrappers.py index 69f4477d..92a393fb 100644 --- a/python/CuTeDSL/cutlass/cute/arch/nvvm_wrappers.py +++ b/python/CuTeDSL/cutlass/cute/arch/nvvm_wrappers.py @@ -956,6 +956,20 @@ def fmax( ) +@dsl_user_op +def fmin( + a: Union[float, Float32], b: Union[float, Float32], *, loc=None, ip=None +) -> Float32: + return Float32( + nvvm.fmin( + Float32(a).ir_value(loc=loc, ip=ip), + Float32(b).ir_value(loc=loc, ip=ip), + loc=loc, + ip=ip, + ) + ) + + @dsl_user_op def rcp_approx(a: Union[float, Float32], *, loc=None, ip=None): return Float32(