Coverage for src/flag_gems/utils/triton_driver_helper.py: 50%

1try: 

2 from flag_gems.runtime import torch_device_fn 

3 

4 get_device_properties = torch_device_fn.get_device_properties 

5except AttributeError: 

6 import triton 

7 

8 get_device_properties = triton.runtime.driver.active.utils.get_device_properties