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

1from backend_utils import VendorInfoBase # noqa: E402 

2 

3vendor_info = VendorInfoBase( 

4 vendor_name="kunlunxin", 

5 device_name="cuda", 

6 device_query_cmd="xpu-smi", 

7 triton_extra_name="xpu", 

8) 

9 

10CUSTOMIZED_UNUSED_OPS = ( 

11 "cumsum", 

12 "unique", 

13 "randperm", 

14 "cummin", 

15 "topk", 

16 "sort", 

17) 

18 

19 

20__all__ = ["*"]