_math.py 564 B

1234567891011121314151617181920
  1. import triton.language.math as tl_math
  2. from ._core import builtin
  3. umulhi = builtin(tl_math.umulhi)
  4. exp = builtin(tl_math.exp)
  5. exp2 = builtin(tl_math.exp2)
  6. fma = builtin(tl_math.fma)
  7. log = builtin(tl_math.log)
  8. log2 = builtin(tl_math.log2)
  9. cos = builtin(tl_math.cos)
  10. rsqrt = builtin(tl_math.rsqrt)
  11. sin = builtin(tl_math.sin)
  12. sqrt = builtin(tl_math.sqrt)
  13. sqrt_rn = builtin(tl_math.sqrt_rn)
  14. abs = builtin(tl_math.abs)
  15. fdiv = builtin(tl_math.fdiv)
  16. div_rn = builtin(tl_math.div_rn)
  17. erf = builtin(tl_math.erf)
  18. floor = builtin(tl_math.floor)
  19. ceil = builtin(tl_math.ceil)