[Cute-DSL] Add cute.arch.fmin by calling nvvm

2026-04-19 22:38:56 +00:00 · 2026-02-11 14:23:09 -05:00
parent 8dbce01473
commit 244e8d00d5
1 changed files with 14 additions and 0 deletions
--- 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(