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

5 statements  

« prev     ^ index     » next       coverage.py v7.6.9, created at 2026-03-13 10:08 +0800

1from .cross_entropy_loss import cross_entropy_loss 

2from .fused_add_rms_norm import fused_add_rms_norm 

3from .rotary_embedding import apply_rotary_pos_emb 

4from .skip_layernorm import skip_layer_norm 

5 

6__all__ = [ 

7 "cross_entropy_loss", 

8 "apply_rotary_pos_emb", 

9 "fused_add_rms_norm", 

10 "skip_layer_norm", 

11]