Coverage for src/flag_gems/ops/rad2deg.py: 94%

1# Generated by KernelGen: https://github.com/flagos-ai/KernelGen 

2import logging 

3 

4import triton 

5import triton.language as tl 

6 

7from flag_gems.utils import pointwise_dynamic 

8 

9logger = logging.getLogger(__name__) 

10 

11 

12@pointwise_dynamic(promotion_methods=[(0, "DEFAULT")]) 

13@triton.jit 

14def rad2deg_func(x): 

15 # 180.0 / pi 

16 return (x.to(tl.float32) * 57.29577951308232).to(x.dtype) 

17 

18 

19def rad2deg(A): 

20 logger.debug("GEMS RAD2DEG") 

21 return rad2deg_func(A) 

22 

23 

24def rad2deg_(A): 

25 logger.debug("GEMS RAD2DEG_") 

26 rad2deg_func(A, out0=A) 

27 return A