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

1from .cross_entropy_loss import cross_entropy_loss 

2from .fused_add_rms_norm import fused_add_rms_norm 

3from .fused_moe import ( 

4 dispatch_fused_moe_kernel, 

5 fused_experts_impl, 

6 inplace_fused_experts, 

7 invoke_fused_moe_triton_kernel, 

8 outplace_fused_experts, 

9) 

10from .moe_align_block_size import moe_align_block_size, moe_align_block_size_triton 

11from .moe_sum import moe_sum 

12from .rotary_embedding import apply_rotary_pos_emb 

13from .skip_layernorm import skip_layer_norm 

14from .sparse_attention import sparse_attn_triton 

15 

16__all__ = [ 

17 "cross_entropy_loss", 

18 "apply_rotary_pos_emb", 

19 "fused_add_rms_norm", 

20 "skip_layer_norm", 

21 "sparse_attn_triton", 

22 "moe_align_block_size", 

23 "moe_align_block_size_triton", 

24 "moe_sum", 

25 "dispatch_fused_moe_kernel", 

26 "fused_experts_impl", 

27 "inplace_fused_experts", 

28 "invoke_fused_moe_triton_kernel", 

29 "outplace_fused_experts", 

30]