Coverage for src/flag_gems/runtime/backend/_sunrise/fused/__init__.py: 0%

1from .bincount import bincount 

2from .flash_mla import flash_mla 

3from .fused_add_rms_norm import fused_add_rms_norm 

4from .reshape_and_cache_flash import reshape_and_cache_flash 

5from .skip_layernorm import skip_layer_norm 

6from .sparse_attention import sparse_attn_triton 

7 

8__all__ = [ 

9 "bincount", 

10 "flash_mla", 

11 "fused_add_rms_norm", 

12 "skip_layer_norm", 

13 "reshape_and_cache_flash", 

14 "sparse_attn_triton", 

15]