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

1from .concat_and_cache_mla import concat_and_cache_mla 

2from .cross_entropy_loss import cross_entropy_loss 

3from .flash_mla import flash_mla 

4from .fused_add_rms_norm import fused_add_rms_norm 

5from .geglu import dgeglu, geglu 

6from .gelu_and_mul import gelu_and_mul 

7from .instance_norm import instance_norm 

8from .moe_align_block_size import moe_align_block_size, moe_align_block_size_triton 

9from .outer import outer 

10from .reglu import dreglu, reglu 

11from .reshape_and_cache import reshape_and_cache 

12from .reshape_and_cache_flash import reshape_and_cache_flash 

13from .rotary_embedding import apply_rotary_pos_emb 

14from .rwkv_ka_fusion import rwkv_ka_fusion 

15from .rwkv_mm_sparsity import rwkv_mm_sparsity 

16from .silu_and_mul import silu_and_mul, silu_and_mul_out 

17from .skip_layernorm import skip_layer_norm 

18from .topk_softmax import topk_softmax 

19from .weight_norm import weight_norm 

20 

21__all__ = [ 

22 "apply_rotary_pos_emb", 

23 "skip_layer_norm", 

24 "fused_add_rms_norm", 

25 "silu_and_mul", 

26 "silu_and_mul_out", 

27 "geglu", 

28 "dgeglu", 

29 "gelu_and_mul", 

30 "cross_entropy_loss", 

31 "outer", 

32 "instance_norm", 

33 "weight_norm", 

34 "concat_and_cache_mla", 

35 "reshape_and_cache", 

36 "moe_align_block_size", 

37 "moe_align_block_size_triton", 

38 "reshape_and_cache_flash", 

39 "flash_mla", 

40 "topk_softmax", 

41 "rwkv_ka_fusion", 

42 "rwkv_mm_sparsity", 

43 "dreglu", 

44 "reglu", 

45]