Coverage for src/flag_gems/runtime/backend/_mthreads/ops/utils.py: 0%

1import torch 

2import triton.language as tl 

3 

4 

5def get_triton_dtype(dtype): 

6 dtype_map = { 

7 torch.float16: tl.float16, 

8 torch.bfloat16: tl.bfloat16, 

9 torch.float32: tl.float32, 

10 } 

11 return dtype_map.get(dtype, None)