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

1from backend_utils import VendorInfoBase 

2 

3vendor_info = VendorInfoBase( 

4 vendor_name="amd", 

5 device_name="cuda", 

6 device_query_cmd="rocm-smi", 

7) 

8 

9CUSTOMIZED_UNUSED_OPS = ( 

10 "add", 

11 "cos", 

12 "cumsum", 

13) 

14 

15 

16__all__ = ["*"]