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

1from backend_utils import VendorInfoBase # noqa: E402 

2 

3vendor_info = VendorInfoBase( 

4 vendor_name="nvidia", device_name="cuda", device_query_cmd="nvidia-smi" 

5) 

6ARCH_MAP = {"9": "hopper", "8": "ampere"} 

7CUSTOMIZED_UNUSED_OPS = ("cumsum", "cos", "add") 

8 

9 

10__all__ = ["*"]