Merge pull request #3022 from tridao/nvvm_fmin

[Cute-DSL] Add cute.arch.fmin by calling nvvm
This commit is contained in:
drazi
2026-02-13 20:45:08 +08:00
committed by GitHub

View File

@@ -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(