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

1from .cross_entropy_loss import cross_entropy_loss 

2from .flash_mla import flash_mla 

3from .fused_add_rms_norm import fused_add_rms_norm 

4from .fused_moe import ( 

5 dispatch_fused_moe_kernel, 

6 fused_experts_impl, 

7 inplace_fused_experts, 

8 invoke_fused_moe_triton_kernel, 

9 outplace_fused_experts, 

10) 

11from .moe_align_block_size import moe_align_block_size, moe_align_block_size_triton 

12from .moe_sum import moe_sum 

13from .rotary_embedding import apply_rotary_pos_emb 

14from .skip_layernorm import skip_layer_norm 

15from .sparse_attention import sparse_attn_triton 

16 

17__all__ = [ 

18 "cross_entropy_loss", 

19 "apply_rotary_pos_emb", 

20 "flash_mla", 

21 "fused_add_rms_norm", 

22 "skip_layer_norm", 

23 "sparse_attn_triton", 

24 "moe_align_block_size", 

25 "moe_align_block_size_triton", 

26 "moe_sum", 

27 "dispatch_fused_moe_kernel", 

28 "fused_experts_impl", 

29 "inplace_fused_experts", 

30 "invoke_fused_moe_triton_kernel", 

31 "outplace_fused_experts", 

32]