Coverage for src/flag_gems/runtime/backend/_aipu/__init__.py: 100%

1from backend_utils import VendorInfoBase # noqa: E402 

2from triton.runtime import driver # noqa: E402 

3 

4vendor_info = VendorInfoBase( 

5 vendor_name="aipu", 

6 device_name="aipu", 

7 device_query_cmd="aipu", 

8 dispatch_key="PrivateUse1", 

9) 

10 

11# The aipu backend is loaded dynamically, so here need to active first. 

12driver.active.get_active_torch_device() 

13 

14CUSTOMIZED_UNUSED_OPS = () 

15 

16__all__ = ["*"]