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

1from backend_utils import VendorInfoBase 

2 

3vendor_info = VendorInfoBase( 

4 vendor_name="nvidia", 

5 device_name="cuda", 

6 device_query_cmd="nvidia-smi", 

7) 

8 

9ARCH_MAP = { 

10 "9": "hopper", 

11 "8": "ampere", 

12} 

13 

14CUSTOMIZED_UNUSED_OPS = ( 

15 "add", 

16 "cos", 

17 "cumsum", 

18) 

19 

20__all__ = ["*"]