Coverage for src/flag_gems/runtime/backend/_cambricon/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 .gelu_and_mul import gelu_and_mul 

5from .outer import outer 

6from .silu_and_mul import silu_and_mul, silu_and_mul_out 

7from .skip_layernorm import skip_layer_norm 

8from .weight_norm import weight_norm 

9 

10__all__ = [ 

11 "skip_layer_norm", 

12 "fused_add_rms_norm", 

13 "silu_and_mul", 

14 "silu_and_mul_out", 

15 "gelu_and_mul", 

16 "cross_entropy_loss", 

17 "outer", 

18 "weight_norm", 

19 "flash_mla", 

20]