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 .fused_moe import fused_experts_impl, inplace_fused_experts, outplace_fused_experts 

5from .fused_recurrent import fused_recurrent_gated_delta_rule_fwd 

6from .hc_head_fused_kernel import hc_head_fused_kernel, hc_head_fused_kernel_ref 

7from .reshape_and_cache_flash import reshape_and_cache_flash 

8from .skip_layernorm import skip_layer_norm 

9from .sparse_attention import sparse_attn_triton 

10 

11__all__ = [ 

12 "bincount", 

13 "flash_mla", 

14 "fused_add_rms_norm", 

15 "fused_experts_impl", 

16 "fused_recurrent_gated_delta_rule_fwd", 

17 "hc_head_fused_kernel", 

18 "hc_head_fused_kernel_ref", 

19 "inplace_fused_experts", 

20 "outplace_fused_experts", 

21 "skip_layer_norm", 

22 "reshape_and_cache_flash", 

23 "sparse_attn_triton", 

24]