triton-windows 3.3.1.post19__cp310-cp310-win_amd64.whl → 3.5.0.post21__cp310-cp310-win_amd64.whl

triton/language/extra/hip/libdevice.py

@@ -2,474 +2,490 @@ from triton.language import core
 
 
 @core.extern
-def abs(arg0, _builder=None):
+def abs(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("int32"), ): ("__triton_hip_iabs", core.dtype("int32")),
             (core.dtype("int64"), ): ("__triton_hip_iabs", core.dtype("int64")),
             (core.dtype("fp32"), ): ("__triton_hip_fabs", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__triton_hip_fabs", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def floor(arg0, _builder=None):
+def floor(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_floor_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_floor_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def rsqrt(arg0, _builder=None):
+def rsqrt(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_rsqrt_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_rsqrt_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def ceil(arg0, _builder=None):
+def ceil(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_ceil_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_ceil_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def trunc(arg0, _builder=None):
+def trunc(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_trunc_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_trunc_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def exp2(arg0, _builder=None):
+def exp2(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_exp2_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_exp2_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def exp(arg0, _builder=None):
+def exp(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_exp_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_exp_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def fast_expf(arg0, _builder=None):
+def fast_expf(arg0, _semantic=None):
     return core.extern_elementwise("", "", [arg0], {
         (core.dtype("fp32"), ): ("__triton_hip_fast_expf", core.dtype("fp32")),
-    }, is_pure=True, _builder=_builder)
+    }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def fast_dividef(arg0, arg1, _builder=None):
+def fast_tanhf(arg0, _semantic=None):
+    return core.extern_elementwise("", "", [arg0], {
+        (core.dtype("fp32"), ): ("__triton_hip_fast_tanhf", core.dtype("fp32")),
+    }, is_pure=True, _semantic=_semantic)
+
+
+@core.extern
+def fast_dividef(arg0, arg1, _semantic=None):
     return core.extern_elementwise("", "", [arg0, arg1], {
         (core.dtype("fp32"), core.dtype("fp32")): ("__triton_hip_fast_fdividef", core.dtype("fp32")),
-    }, is_pure=True, _builder=_builder)
+    }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def sqrt(arg0, _builder=None):
+def sqrt(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_sqrt_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_sqrt_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def llrint(arg0, _builder=None):
+def llrint(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__triton_hip_llrint", core.dtype("int64")),
             (core.dtype("fp64"), ): ("__triton_hip_llrint", core.dtype("int64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def nearbyint(arg0, _builder=None):
+def nearbyint(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [
             arg0,
         ], {
             (core.dtype("fp32"), ): ("__ocml_nearbyint_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_nearbyint_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def isnan(arg0, _builder=None):
+def isnan(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [
             arg0,
         ], {
             (core.dtype("fp32"), ): ("__ocml_isnan_f32", core.dtype("int32")),
             (core.dtype("fp64"), ): ("__ocml_isnan_f64", core.dtype("int32")),
-        }, is_pure=True, _builder=_builder).to(core.int1, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic).to(core.int1, _semantic=_semantic)
 
 
 @core.extern
-def signbit(arg0, _builder=None):
+def signbit(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [
             arg0,
         ], {
             (core.dtype("fp32"), ): ("__ocml_signbit_f32", core.dtype("int32")),
             (core.dtype("fp64"), ): ("__ocml_signbit_f64", core.dtype("int32")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def copysign(arg0, arg1, _builder=None):
+def copysign(arg0, arg1, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0, arg1], {
             (core.dtype("fp32"), core.dtype("fp32")): ("__ocml_copysign_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("fp64")): ("__ocml_copysign_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def isinf(arg0, _builder=None):
+def isinf(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_isinf_f32", core.dtype("int32")),
             (core.dtype("fp64"), ): ("__ocml_isinf_f64", core.dtype("int32")),
-        }, is_pure=True, _builder=_builder).to(core.int1, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic).to(core.int1, _semantic=_semantic)
 
 
 @core.extern
-def nextafter(arg0, arg1, _builder=None):
+def nextafter(arg0, arg1, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0, arg1], {
             (core.dtype("fp32"), core.dtype("fp32")): ("__ocml_nextafter_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("fp64")): ("__ocml_nextafter_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def sin(arg0, _builder=None):
+def sin(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_sin_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_sin_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def cos(arg0, _builder=None):
+def cos(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_cos_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_cos_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def tan(arg0, _builder=None):
+def tan(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_tan_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_tan_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def log2(arg0, _builder=None):
+def log2(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_log2_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_log2_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def cosh(arg0, _builder=None):
+def cosh(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_cosh_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_cosh_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def sinh(arg0, _builder=None):
+def sinh(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_sinh_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_sinh_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def tanh(arg0, _builder=None):
+def tanh(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_tanh_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_tanh_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def atan2(arg0, arg1, _builder=None):
+def atan2(arg0, arg1, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0, arg1], {
             (core.dtype("fp32"), core.dtype("fp32")): ("__ocml_atan2_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("fp64")): ("__ocml_atan2_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def atan(arg0, _builder=None):
+def atan(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_atan_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_atan_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def asin(arg0, _builder=None):
+def asin(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_asin_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_asin_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def acos(arg0, _builder=None):
+def acos(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_acos_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_acos_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def log(arg0, _builder=None):
+def log(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_log_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_log_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def log10(arg0, _builder=None):
+def log10(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_log10_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_log10_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def log1p(arg0, _builder=None):
+def log1p(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_log1p_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_log1p_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def acosh(arg0, _builder=None):
+def acosh(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_acosh_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_acosh_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def asinh(arg0, _builder=None):
+def asinh(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_asinh_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_asinh_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def atanh(arg0, _builder=None):
+def atanh(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_atanh_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_atanh_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def expm1(arg0, _builder=None):
+def expm1(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_expm1_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_expm1_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def hypot(arg0, arg1, _builder=None):
+def hypot(arg0, arg1, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0, arg1], {
             (core.dtype("fp32"), core.dtype("fp32")): ("__ocml_hypot_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("fp64")): ("__ocml_hypot_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def j0(arg0, _builder=None):
+def j0(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_j0_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_j0_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def j1(arg0, _builder=None):
+def j1(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_j1_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_j1_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def y0(arg0, _builder=None):
+def y0(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_y0_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_y0_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def y1(arg0, _builder=None):
+def y1(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_y1_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_y1_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def cyl_bessel_i0(arg0, _builder=None):
+def cyl_bessel_i0(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_i0_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_i0_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def cyl_bessel_i1(arg0, _builder=None):
+def cyl_bessel_i1(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_i1_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_i1_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def erf(arg0, _builder=None):
+def erf(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_erf_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_erf_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def erfinv(arg0, _builder=None):
+def erfinv(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_erfinv_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_erfinv_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def erfc(arg0, _builder=None):
+def erfc(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_erfc_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_erfc_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def erfcx(arg0, _builder=None):
+def erfcx(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_erfcx_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_erfcx_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def lgamma(arg0, _builder=None):
+def lgamma(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_lgamma_f32", core.dtype("fp32")),
             (core.dtype("fp64"), ): ("__ocml_lgamma_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def ldexp(arg0, arg1, _builder=None):
+def ldexp(arg0, arg1, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0, arg1], {
             (core.dtype("fp32"), core.dtype("int32")): ("__ocml_ldexp_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("int32")): ("__ocml_ldexp_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def fmod(arg0, arg1, _builder=None):
+def fmod(arg0, arg1, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0, arg1], {
             (core.dtype("fp32"), core.dtype("fp32")): ("__ocml_fmod_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("fp64")): ("__ocml_fmod_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def fma(arg0, arg1, arg2, _builder=None):
+def fma(arg0, arg1, arg2, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0, arg1, arg2], {
             (core.dtype("fp32"), core.dtype("fp32"), core.dtype("fp32")): ("__ocml_fma_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("fp64"), core.dtype("fp64")): ("__ocml_fma_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def pow(arg0, arg1, _builder=None):
+def pow(arg0, arg1, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0, arg1], {
             (core.dtype("fp32"), core.dtype("int32")): ("__ocml_pown_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("int32")): ("__ocml_pown_f64", core.dtype("fp64")),
             (core.dtype("fp32"), core.dtype("fp32")): ("__ocml_pow_f32", core.dtype("fp32")),
             (core.dtype("fp64"), core.dtype("fp64")): ("__ocml_pow_f64", core.dtype("fp64")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
 
 
 @core.extern
-def ilogb(arg0, _builder=None):
+def ilogb(arg0, _semantic=None):
     return core.extern_elementwise(
         "", "", [arg0], {
             (core.dtype("fp32"), ): ("__ocml_ilogb_f32", core.dtype("int32")),
             (core.dtype("fp64"), ): ("__ocml_ilogb_f64", core.dtype("int32")),
-        }, is_pure=True, _builder=_builder)
+        }, is_pure=True, _semantic=_semantic)
+
+
+@core.extern
+def round(arg0, _semantic=None):
+    return core.extern_elementwise(
+        "", "", [arg0], {
+            (core.dtype("fp32"), ): ("__ocml_round_f32", core.dtype("fp32")),
+            (core.dtype("fp64"), ): ("__ocml_round_f64", core.dtype("fp64")),
+        }, is_pure=True, _semantic=_semantic)

triton/language/extra/hip/utils.py

@@ -0,0 +1,35 @@
+from triton.language import core
+
+
+@core.extern
+def memrealtime(_semantic=None):
+    """
+    Returns a 64-bit real time-counter value
+    """
+    target_arch = _semantic.builder.options.arch
+    if 'gfx11' in target_arch or 'gfx12' in target_arch:
+        return core.inline_asm_elementwise(
+            """
+            s_sendmsg_rtn_b64 $0, sendmsg(MSG_RTN_GET_REALTIME)
+            s_waitcnt lgkmcnt(0)
+            """,
+            "=r",
+            [],
+            dtype=core.int64,
+            is_pure=False,
+            pack=1,
+            _semantic=_semantic,
+        )
+    else:
+        return core.inline_asm_elementwise(
+            """
+            s_memrealtime $0
+            s_waitcnt vmcnt(0)
+            """,
+            "=r",
+            [],
+            dtype=core.int64,
+            is_pure=False,
+            pack=1,
+            _semantic=_semantic,
+        )

triton/language/extra/libdevice.py

@@ -466,6 +466,10 @@ def fast_expf(arg0):
     ...
 
 
+def fast_tanhf(arg0):
+    ...
+
+
 def fast_tanf(arg0):
     ...