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

1from .bincount import bincount 

2from .concat_and_cache_mla import concat_and_cache_mla 

3from .cross_entropy_loss import cross_entropy_loss 

4from .flash_mla import flash_mla 

5from .fused_add_rms_norm import fused_add_rms_norm 

6from .geglu import dgeglu, geglu 

7from .gelu_and_mul import gelu_and_mul 

8from .instance_norm import instance_norm 

9from .moe_align_block_size import moe_align_block_size, moe_align_block_size_triton 

10from .outer import outer 

11from .reglu import dreglu, reglu 

12from .reshape_and_cache import reshape_and_cache 

13from .reshape_and_cache_flash import reshape_and_cache_flash 

14from .rotary_embedding import apply_rotary_pos_emb 

15from .rwkv_ka_fusion import rwkv_ka_fusion 

16from .rwkv_mm_sparsity import rwkv_mm_sparsity 

17from .silu_and_mul import silu_and_mul, silu_and_mul_out 

18from .skip_layernorm import skip_layer_norm 

19from .sparse_attention import sparse_attn_triton 

20from .topk_softmax import topk_softmax 

21from .weight_norm import weight_norm 

22 

23__all__ = [ 

24 "apply_rotary_pos_emb", 

25 "skip_layer_norm", 

26 "fused_add_rms_norm", 

27 "silu_and_mul", 

28 "silu_and_mul_out", 

29 "geglu", 

30 "dgeglu", 

31 "gelu_and_mul", 

32 "cross_entropy_loss", 

33 "outer", 

34 "instance_norm", 

35 "weight_norm", 

36 "concat_and_cache_mla", 

37 "reshape_and_cache", 

38 "moe_align_block_size", 

39 "moe_align_block_size_triton", 

40 "reshape_and_cache_flash", 

41 "flash_mla", 

42 "topk_softmax", 

43 "rwkv_ka_fusion", 

44 "rwkv_mm_sparsity", 

45 "dreglu", 

46 "reglu", 

47 "sparse_attn_triton", 

48 "bincount", 

49]