| src/flag_gems/__init__.py |
use_gems |
21 |
2 |
0 |
90% |
| src/flag_gems/__init__.py |
(no class) |
52 |
6 |
0 |
88% |
| src/flag_gems/config.py |
(no class) |
77 |
33 |
2 |
57% |
| src/flag_gems/experimental_ops/__init__.py |
(no class) |
2 |
0 |
0 |
100% |
| src/flag_gems/experimental_ops/_adaptive_avg_pool3d.py |
(no class) |
114 |
114 |
0 |
0% |
| src/flag_gems/experimental_ops/_log_softmax_backward_data.py |
(no class) |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/_unsafe_view.py |
(no class) |
76 |
76 |
0 |
0% |
| src/flag_gems/experimental_ops/abs.py |
(no class) |
101 |
101 |
0 |
0% |
| src/flag_gems/experimental_ops/abs_.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/absolute_.py |
(no class) |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/addcdiv.py |
(no class) |
61 |
61 |
0 |
0% |
| src/flag_gems/experimental_ops/addcmul_.py |
(no class) |
60 |
60 |
0 |
0% |
| src/flag_gems/experimental_ops/amin.py |
(no class) |
135 |
135 |
0 |
0% |
| src/flag_gems/experimental_ops/arccosh.py |
(no class) |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/arctanh.py |
(no class) |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/atanh_.py |
(no class) |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/celu.py |
(no class) |
67 |
67 |
0 |
0% |
| src/flag_gems/experimental_ops/celu_.py |
(no class) |
24 |
24 |
0 |
0% |
| src/flag_gems/experimental_ops/copy_.py |
(no class) |
74 |
74 |
0 |
0% |
| src/flag_gems/experimental_ops/cos_.py |
(no class) |
28 |
28 |
0 |
0% |
| src/flag_gems/experimental_ops/cosh_.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/deg2rad.py |
(no class) |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/deg2rad_.py |
(no class) |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/diag.py |
(no class) |
112 |
112 |
0 |
0% |
| src/flag_gems/experimental_ops/elu.py |
(no class) |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/erf_.py |
(no class) |
43 |
43 |
0 |
0% |
| src/flag_gems/experimental_ops/erfinv.py |
(no class) |
78 |
78 |
0 |
0% |
| src/flag_gems/experimental_ops/erfinv_.py |
(no class) |
46 |
46 |
0 |
0% |
| src/flag_gems/experimental_ops/exp2.py |
(no class) |
42 |
42 |
0 |
0% |
| src/flag_gems/experimental_ops/exp2_.py |
(no class) |
30 |
30 |
0 |
0% |
| src/flag_gems/experimental_ops/exp_.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/expand.py |
(no class) |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/eye.py |
(no class) |
53 |
53 |
0 |
0% |
| src/flag_gems/experimental_ops/fft_ifftshift.py |
(no class) |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/fix.py |
(no class) |
46 |
46 |
0 |
0% |
| src/flag_gems/experimental_ops/fix_.py |
(no class) |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/frac.py |
(no class) |
55 |
55 |
0 |
0% |
| src/flag_gems/experimental_ops/gelu_.py |
(no class) |
74 |
74 |
0 |
0% |
| src/flag_gems/experimental_ops/glu.py |
(no class) |
78 |
78 |
0 |
0% |
| src/flag_gems/experimental_ops/hardshrink.py |
(no class) |
44 |
44 |
0 |
0% |
| src/flag_gems/experimental_ops/hardsigmoid_.py |
(no class) |
42 |
42 |
0 |
0% |
| src/flag_gems/experimental_ops/hardswish.py |
(no class) |
74 |
74 |
0 |
0% |
| src/flag_gems/experimental_ops/hardtanh.py |
(no class) |
49 |
49 |
0 |
0% |
| src/flag_gems/experimental_ops/hardtanh_.py |
(no class) |
51 |
51 |
0 |
0% |
| src/flag_gems/experimental_ops/heaviside.py |
(no class) |
53 |
53 |
0 |
0% |
| src/flag_gems/experimental_ops/heaviside_.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/experimental_ops/huber_loss.py |
(no class) |
102 |
102 |
0 |
0% |
| src/flag_gems/experimental_ops/hypot_.py |
(no class) |
56 |
56 |
0 |
0% |
| src/flag_gems/experimental_ops/im2col.py |
(no class) |
114 |
114 |
0 |
0% |
| src/flag_gems/experimental_ops/leaky_relu.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/leaky_relu_.py |
(no class) |
44 |
44 |
0 |
0% |
| src/flag_gems/experimental_ops/lift.py |
(no class) |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/log10_.py |
(no class) |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/log2_.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/log_.py |
(no class) |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/logaddexp2.py |
(no class) |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/logical_xor_.py |
(no class) |
41 |
41 |
0 |
0% |
| src/flag_gems/experimental_ops/masked_scatter.py |
(no class) |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/masked_select.py |
(no class) |
85 |
85 |
0 |
0% |
| src/flag_gems/experimental_ops/maximum.py |
(no class) |
103 |
103 |
0 |
0% |
| src/flag_gems/experimental_ops/mse_loss.py |
(no class) |
134 |
134 |
0 |
0% |
| src/flag_gems/experimental_ops/multiply.py |
(no class) |
91 |
91 |
0 |
0% |
| src/flag_gems/experimental_ops/mv.py |
(no class) |
58 |
58 |
0 |
0% |
| src/flag_gems/experimental_ops/native_dropout_backward.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/neg_.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/negative.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/experimental_ops/negative_.py |
(no class) |
25 |
25 |
0 |
0% |
| src/flag_gems/experimental_ops/permute.py |
(no class) |
64 |
64 |
0 |
0% |
| src/flag_gems/experimental_ops/permute_copy.py |
(no class) |
74 |
74 |
0 |
0% |
| src/flag_gems/experimental_ops/pixel_shuffle.py |
(no class) |
86 |
86 |
0 |
0% |
| src/flag_gems/experimental_ops/rad2deg_.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/reciprocal.py |
(no class) |
55 |
55 |
0 |
0% |
| src/flag_gems/experimental_ops/reciprocal_.py |
(no class) |
24 |
24 |
0 |
0% |
| src/flag_gems/experimental_ops/relu.py |
(no class) |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/relu_.py |
(no class) |
28 |
28 |
0 |
0% |
| src/flag_gems/experimental_ops/replication_pad2d.py |
(no class) |
78 |
78 |
0 |
0% |
| src/flag_gems/experimental_ops/replication_pad3d.py |
(no class) |
94 |
94 |
0 |
0% |
| src/flag_gems/experimental_ops/rmsnorm.py |
(no class) |
46 |
40 |
0 |
13% |
| src/flag_gems/experimental_ops/rsqrt_.py |
(no class) |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/sgn.py |
(no class) |
60 |
60 |
0 |
0% |
| src/flag_gems/experimental_ops/sigmoid.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/experimental_ops/sigmoid_.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/sign.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/silu.py |
(no class) |
64 |
64 |
0 |
0% |
| src/flag_gems/experimental_ops/silu_.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/sin_.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/sinc.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/experimental_ops/sinc_.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/slice_backward.py |
(no class) |
76 |
76 |
0 |
0% |
| src/flag_gems/experimental_ops/slice_scatter.py |
(no class) |
87 |
87 |
0 |
0% |
| src/flag_gems/experimental_ops/smooth_l1_loss.py |
(no class) |
149 |
149 |
0 |
0% |
| src/flag_gems/experimental_ops/softplus.py |
(no class) |
69 |
69 |
0 |
0% |
| src/flag_gems/experimental_ops/special_xlog1py.py |
(no class) |
90 |
90 |
0 |
0% |
| src/flag_gems/experimental_ops/take.py |
(no class) |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/threshold.py |
(no class) |
55 |
55 |
0 |
0% |
| src/flag_gems/experimental_ops/threshold_.py |
(no class) |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/trace.py |
(no class) |
61 |
61 |
0 |
0% |
| src/flag_gems/experimental_ops/triu.py |
(no class) |
73 |
73 |
0 |
0% |
| src/flag_gems/experimental_ops/trunc.py |
(no class) |
85 |
85 |
0 |
0% |
| src/flag_gems/experimental_ops/unsqueeze.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/experimental_ops/unsqueeze_copy.py |
(no class) |
60 |
60 |
0 |
0% |
| src/flag_gems/experimental_ops/upsample_nearest1d.py |
(no class) |
70 |
70 |
0 |
0% |
| src/flag_gems/experimental_ops/upsample_nearest3d.py |
(no class) |
121 |
121 |
0 |
0% |
| src/flag_gems/experimental_ops/xlogy.py |
(no class) |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/xlogy_.py |
(no class) |
77 |
77 |
0 |
0% |
| src/flag_gems/experimental_ops/zeros_like.py |
(no class) |
79 |
79 |
0 |
0% |
| src/flag_gems/fused/FLA/__init__.py |
(no class) |
3 |
0 |
0 |
100% |
| src/flag_gems/fused/FLA/chunk.py |
(no class) |
18 |
9 |
0 |
50% |
| src/flag_gems/fused/FLA/chunk_delta_h.py |
(no class) |
163 |
149 |
0 |
9% |
| src/flag_gems/fused/FLA/chunk_o.py |
(no class) |
69 |
54 |
0 |
22% |
| src/flag_gems/fused/FLA/chunk_scaled_dot_kkt.py |
(no class) |
46 |
46 |
0 |
0% |
| src/flag_gems/fused/FLA/cumsum.py |
(no class) |
90 |
90 |
0 |
0% |
| src/flag_gems/fused/FLA/fused_cumsum_kkt_solve_tril.py |
(no class) |
157 |
142 |
0 |
10% |
| src/flag_gems/fused/FLA/fused_recurrent.py |
(no class) |
174 |
161 |
0 |
7% |
| src/flag_gems/fused/FLA/index.py |
(no class) |
13 |
4 |
0 |
69% |
| src/flag_gems/fused/FLA/solve_tril.py |
(no class) |
226 |
198 |
0 |
12% |
| src/flag_gems/fused/FLA/triton_ops_helper.py |
(no class) |
15 |
6 |
0 |
60% |
| src/flag_gems/fused/FLA/utils.py |
(no class) |
68 |
13 |
0 |
81% |
| src/flag_gems/fused/FLA/wy_fast.py |
(no class) |
49 |
38 |
0 |
22% |
| src/flag_gems/fused/__init__.py |
(no class) |
28 |
0 |
0 |
100% |
| src/flag_gems/fused/apply_repetition_penalties.py |
(no class) |
37 |
16 |
0 |
57% |
| src/flag_gems/fused/bincount.py |
(no class) |
120 |
59 |
0 |
51% |
| src/flag_gems/fused/concat_and_cache_mla.py |
ConcatAndCacheMla |
26 |
4 |
0 |
85% |
| src/flag_gems/fused/concat_and_cache_mla.py |
(no class) |
54 |
35 |
0 |
35% |
| src/flag_gems/fused/cross_entropy_loss.py |
CrossEntropyLoss |
64 |
18 |
0 |
72% |
| src/flag_gems/fused/cross_entropy_loss.py |
(no class) |
312 |
268 |
0 |
14% |
| src/flag_gems/fused/cutlass_scaled_mm.py |
(no class) |
194 |
158 |
0 |
19% |
| src/flag_gems/fused/flash_mla.py |
(no class) |
107 |
72 |
0 |
33% |
| src/flag_gems/fused/fused_add_rms_norm.py |
(no class) |
40 |
17 |
0 |
58% |
| src/flag_gems/fused/fused_moe.py |
MoEActivation |
9 |
4 |
0 |
56% |
| src/flag_gems/fused/fused_moe.py |
(no class) |
656 |
377 |
0 |
43% |
| src/flag_gems/fused/geglu.py |
(no class) |
70 |
34 |
0 |
51% |
| src/flag_gems/fused/gelu_and_mul.py |
GeluAndMul |
14 |
7 |
0 |
50% |
| src/flag_gems/fused/gelu_and_mul.py |
(no class) |
56 |
28 |
0 |
50% |
| src/flag_gems/fused/grouped_topk.py |
(no class) |
133 |
123 |
0 |
8% |
| src/flag_gems/fused/instance_norm.py |
InstanceNorm |
67 |
19 |
0 |
72% |
| src/flag_gems/fused/instance_norm.py |
(no class) |
241 |
194 |
0 |
20% |
| src/flag_gems/fused/moe_align_block_size.py |
(no class) |
348 |
259 |
0 |
26% |
| src/flag_gems/fused/moe_sum.py |
(no class) |
30 |
15 |
0 |
50% |
| src/flag_gems/fused/outer.py |
Outer |
15 |
6 |
0 |
60% |
| src/flag_gems/fused/outer.py |
(no class) |
11 |
0 |
0 |
100% |
| src/flag_gems/fused/reglu.py |
(no class) |
80 |
37 |
0 |
54% |
| src/flag_gems/fused/reshape_and_cache.py |
(no class) |
41 |
20 |
0 |
51% |
| src/flag_gems/fused/reshape_and_cache_flash.py |
(no class) |
44 |
19 |
0 |
57% |
| src/flag_gems/fused/rotary_embedding.py |
(no class) |
128 |
84 |
0 |
34% |
| src/flag_gems/fused/rwkv_ka_fusion.py |
(no class) |
40 |
21 |
0 |
48% |
| src/flag_gems/fused/rwkv_mm_sparsity.py |
(no class) |
33 |
14 |
0 |
58% |
| src/flag_gems/fused/silu_and_mul.py |
SiluAndMul |
7 |
4 |
0 |
43% |
| src/flag_gems/fused/silu_and_mul.py |
(no class) |
32 |
13 |
0 |
59% |
| src/flag_gems/fused/skip_layernorm.py |
SkipLayerNorm |
13 |
0 |
0 |
100% |
| src/flag_gems/fused/skip_layernorm.py |
(no class) |
37 |
20 |
0 |
46% |
| src/flag_gems/fused/swiglu.py |
(no class) |
74 |
35 |
0 |
53% |
| src/flag_gems/fused/topk_softmax.py |
(no class) |
54 |
46 |
0 |
15% |
| src/flag_gems/fused/weight_norm.py |
WeightNorm |
17 |
17 |
0 |
0% |
| src/flag_gems/fused/weight_norm.py |
(no class) |
107 |
80 |
0 |
25% |
| src/flag_gems/logging_utils.py |
LogOncePerLocationFilter |
7 |
7 |
0 |
0% |
| src/flag_gems/logging_utils.py |
(no class) |
31 |
1 |
0 |
97% |
| src/flag_gems/modules/__init__.py |
(no class) |
5 |
0 |
0 |
100% |
| src/flag_gems/modules/activation.py |
GemsSiluAndMul |
2 |
2 |
0 |
0% |
| src/flag_gems/modules/activation.py |
(no class) |
13 |
2 |
0 |
85% |
| src/flag_gems/modules/normalization.py |
GemsRMSNorm |
15 |
15 |
0 |
0% |
| src/flag_gems/modules/normalization.py |
(no class) |
34 |
13 |
0 |
62% |
| src/flag_gems/modules/rotary_embedding.py |
GemsRope |
17 |
17 |
0 |
0% |
| src/flag_gems/modules/rotary_embedding.py |
GemsDeepseekYarnRoPE |
19 |
19 |
0 |
0% |
| src/flag_gems/modules/rotary_embedding.py |
(no class) |
43 |
20 |
0 |
53% |
| src/flag_gems/ops/__init__.py |
(no class) |
222 |
0 |
0 |
100% |
| src/flag_gems/ops/_functional_sym_constrain_range_for_size.py |
(no class) |
35 |
11 |
0 |
69% |
| src/flag_gems/ops/_safe_softmax.py |
(no class) |
55 |
15 |
0 |
73% |
| src/flag_gems/ops/_upsample_nearest_exact1d.py |
(no class) |
133 |
63 |
0 |
53% |
| src/flag_gems/ops/abs.py |
(no class) |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/absolute.py |
(no class) |
43 |
23 |
0 |
47% |
| src/flag_gems/ops/acos.py |
(no class) |
14 |
1 |
0 |
93% |
| src/flag_gems/ops/add.py |
(no class) |
37 |
7 |
0 |
81% |
| src/flag_gems/ops/addcdiv.py |
(no class) |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/addcmul.py |
(no class) |
18 |
6 |
0 |
67% |
| src/flag_gems/ops/addmm.py |
(no class) |
64 |
24 |
0 |
62% |
| src/flag_gems/ops/addmv.py |
(no class) |
57 |
21 |
0 |
63% |
| src/flag_gems/ops/addr.py |
(no class) |
44 |
19 |
0 |
57% |
| src/flag_gems/ops/alias_copy.py |
(no class) |
49 |
27 |
0 |
45% |
| src/flag_gems/ops/all.py |
(no class) |
103 |
33 |
0 |
68% |
| src/flag_gems/ops/amax.py |
(no class) |
92 |
50 |
0 |
46% |
| src/flag_gems/ops/angle.py |
(no class) |
30 |
8 |
0 |
73% |
| src/flag_gems/ops/any.py |
(no class) |
103 |
33 |
0 |
68% |
| src/flag_gems/ops/arange.py |
(no class) |
48 |
12 |
0 |
75% |
| src/flag_gems/ops/arcsinh.py |
(no class) |
53 |
17 |
0 |
68% |
| src/flag_gems/ops/arcsinh_.py |
(no class) |
34 |
15 |
0 |
56% |
| src/flag_gems/ops/arctanh_.py |
(no class) |
42 |
22 |
0 |
48% |
| src/flag_gems/ops/argmax.py |
(no class) |
163 |
92 |
0 |
44% |
| src/flag_gems/ops/argmin.py |
(no class) |
161 |
89 |
0 |
45% |
| src/flag_gems/ops/asinh_.py |
(no class) |
45 |
25 |
0 |
44% |
| src/flag_gems/ops/atan.py |
(no class) |
18 |
1 |
0 |
94% |
| src/flag_gems/ops/attention.py |
ScaleDotProductAttention |
13 |
6 |
0 |
54% |
| src/flag_gems/ops/attention.py |
(no class) |
386 |
278 |
0 |
28% |
| src/flag_gems/ops/avg_pool2d.py |
(no class) |
150 |
86 |
0 |
43% |
| src/flag_gems/ops/baddbmm.py |
BaddbmmFunction |
29 |
12 |
0 |
59% |
| src/flag_gems/ops/baddbmm.py |
(no class) |
117 |
91 |
0 |
22% |
| src/flag_gems/ops/batch_norm.py |
(no class) |
154 |
100 |
0 |
35% |
| src/flag_gems/ops/bitwise_and.py |
(no class) |
27 |
2 |
0 |
93% |
| src/flag_gems/ops/bitwise_left_shift.py |
(no class) |
11 |
3 |
0 |
73% |
| src/flag_gems/ops/bitwise_not.py |
(no class) |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/bitwise_or.py |
(no class) |
27 |
2 |
0 |
93% |
| src/flag_gems/ops/bitwise_right_shift.py |
(no class) |
11 |
3 |
0 |
73% |
| src/flag_gems/ops/bmm.py |
(no class) |
93 |
58 |
0 |
38% |
| src/flag_gems/ops/cat.py |
(no class) |
109 |
35 |
0 |
68% |
| src/flag_gems/ops/ceil.py |
(no class) |
22 |
2 |
0 |
91% |
| src/flag_gems/ops/celu.py |
(no class) |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/clamp.py |
(no class) |
75 |
18 |
0 |
76% |
| src/flag_gems/ops/contiguous.py |
(no class) |
11 |
6 |
0 |
45% |
| src/flag_gems/ops/conv1d.py |
(no class) |
27 |
3 |
0 |
89% |
| src/flag_gems/ops/conv2d.py |
Conv2d |
73 |
38 |
0 |
48% |
| src/flag_gems/ops/conv2d.py |
(no class) |
105 |
65 |
0 |
38% |
| src/flag_gems/ops/conv3d.py |
(no class) |
104 |
45 |
0 |
57% |
| src/flag_gems/ops/conv_depthwise2d.py |
(no class) |
10 |
6 |
0 |
40% |
| src/flag_gems/ops/copy.py |
(no class) |
60 |
19 |
0 |
68% |
| src/flag_gems/ops/cos.py |
(no class) |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/count_nonzero.py |
(no class) |
95 |
36 |
0 |
62% |
| src/flag_gems/ops/cummax.py |
(no class) |
242 |
153 |
0 |
37% |
| src/flag_gems/ops/cummin.py |
(no class) |
242 |
153 |
0 |
37% |
| src/flag_gems/ops/cumsum.py |
(no class) |
328 |
198 |
0 |
40% |
| src/flag_gems/ops/diag.py |
(no class) |
62 |
20 |
0 |
68% |
| src/flag_gems/ops/diag_embed.py |
(no class) |
29 |
1 |
0 |
97% |
| src/flag_gems/ops/diagonal.py |
(no class) |
15 |
6 |
0 |
60% |
| src/flag_gems/ops/digamma_.py |
(no class) |
54 |
38 |
0 |
30% |
| src/flag_gems/ops/div.py |
(no class) |
174 |
70 |
0 |
60% |
| src/flag_gems/ops/dot.py |
(no class) |
63 |
32 |
0 |
49% |
| src/flag_gems/ops/dropout.py |
(no class) |
90 |
50 |
0 |
44% |
| src/flag_gems/ops/elu.py |
(no class) |
31 |
6 |
0 |
81% |
| src/flag_gems/ops/embedding.py |
(no class) |
98 |
45 |
0 |
54% |
| src/flag_gems/ops/embedding_dense_backward.py |
(no class) |
64 |
30 |
0 |
53% |
| src/flag_gems/ops/eq.py |
(no class) |
33 |
6 |
0 |
82% |
| src/flag_gems/ops/erf.py |
(no class) |
16 |
2 |
0 |
88% |
| src/flag_gems/ops/exp.py |
(no class) |
18 |
1 |
0 |
94% |
| src/flag_gems/ops/exp2.py |
(no class) |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/exponential_.py |
(no class) |
120 |
71 |
0 |
41% |
| src/flag_gems/ops/eye.py |
(no class) |
21 |
2 |
0 |
90% |
| src/flag_gems/ops/eye_m.py |
(no class) |
35 |
12 |
0 |
66% |
| src/flag_gems/ops/fill.py |
(no class) |
61 |
10 |
0 |
84% |
| src/flag_gems/ops/flash_api.py |
fwd_params |
59 |
0 |
0 |
100% |
| src/flag_gems/ops/flash_api.py |
(no class) |
318 |
34 |
0 |
89% |
| src/flag_gems/ops/flash_kernel.py |
(no class) |
574 |
497 |
0 |
13% |
| src/flag_gems/ops/flip.py |
(no class) |
32 |
1 |
0 |
97% |
| src/flag_gems/ops/floor_.py |
(no class) |
41 |
19 |
0 |
54% |
| src/flag_gems/ops/fmin.py |
(no class) |
87 |
26 |
0 |
70% |
| src/flag_gems/ops/full.py |
(no class) |
41 |
4 |
0 |
90% |
| src/flag_gems/ops/full_like.py |
(no class) |
16 |
0 |
0 |
100% |
| src/flag_gems/ops/gather.py |
GatherFunction |
17 |
0 |
0 |
100% |
| src/flag_gems/ops/gather.py |
(no class) |
113 |
3 |
0 |
97% |
| src/flag_gems/ops/ge.py |
(no class) |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/gelu.py |
(no class) |
56 |
17 |
0 |
70% |
| src/flag_gems/ops/get_scheduler_metadata.py |
(no class) |
333 |
309 |
0 |
7% |
| src/flag_gems/ops/glu.py |
(no class) |
34 |
7 |
0 |
79% |
| src/flag_gems/ops/groupnorm.py |
(no class) |
139 |
88 |
0 |
37% |
| src/flag_gems/ops/gt.py |
(no class) |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/hardsigmoid.py |
(no class) |
30 |
16 |
0 |
47% |
| src/flag_gems/ops/hardswish_.py |
(no class) |
44 |
19 |
0 |
57% |
| src/flag_gems/ops/hstack.py |
(no class) |
106 |
34 |
0 |
68% |
| src/flag_gems/ops/hypot.py |
(no class) |
75 |
33 |
0 |
56% |
| src/flag_gems/ops/i0.py |
(no class) |
61 |
29 |
0 |
52% |
| src/flag_gems/ops/i0_.py |
(no class) |
46 |
25 |
0 |
46% |
| src/flag_gems/ops/index.py |
IndexFunction |
25 |
1 |
0 |
96% |
| src/flag_gems/ops/index.py |
(no class) |
256 |
22 |
0 |
91% |
| src/flag_gems/ops/index_add.py |
IndexAddFunction |
19 |
0 |
0 |
100% |
| src/flag_gems/ops/index_add.py |
(no class) |
138 |
0 |
0 |
100% |
| src/flag_gems/ops/index_put.py |
IndexPutFunction |
25 |
1 |
0 |
96% |
| src/flag_gems/ops/index_put.py |
(no class) |
212 |
20 |
0 |
91% |
| src/flag_gems/ops/index_select.py |
(no class) |
50 |
16 |
0 |
68% |
| src/flag_gems/ops/isclose.py |
(no class) |
45 |
20 |
0 |
56% |
| src/flag_gems/ops/isfinite.py |
(no class) |
20 |
4 |
0 |
80% |
| src/flag_gems/ops/isin.py |
(no class) |
130 |
53 |
0 |
59% |
| src/flag_gems/ops/isinf.py |
(no class) |
13 |
1 |
0 |
92% |
| src/flag_gems/ops/isnan.py |
(no class) |
13 |
1 |
0 |
92% |
| src/flag_gems/ops/kron.py |
(no class) |
171 |
65 |
0 |
62% |
| src/flag_gems/ops/layernorm.py |
(no class) |
241 |
163 |
0 |
32% |
| src/flag_gems/ops/le.py |
(no class) |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/lerp.py |
(no class) |
35 |
5 |
0 |
86% |
| src/flag_gems/ops/lift_fresh_copy.py |
(no class) |
55 |
34 |
0 |
38% |
| src/flag_gems/ops/linspace.py |
(no class) |
34 |
10 |
0 |
71% |
| src/flag_gems/ops/log.py |
(no class) |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/log1p_.py |
(no class) |
38 |
15 |
0 |
61% |
| src/flag_gems/ops/log_sigmoid.py |
(no class) |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/log_softmax.py |
(no class) |
98 |
50 |
0 |
49% |
| src/flag_gems/ops/logaddexp.py |
(no class) |
75 |
22 |
0 |
71% |
| src/flag_gems/ops/logical_and.py |
(no class) |
20 |
2 |
0 |
90% |
| src/flag_gems/ops/logical_not.py |
(no class) |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/logical_or.py |
(no class) |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/logical_xor.py |
(no class) |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/logit.py |
(no class) |
80 |
41 |
0 |
49% |
| src/flag_gems/ops/logit_.py |
(no class) |
63 |
30 |
0 |
52% |
| src/flag_gems/ops/logspace.py |
(no class) |
37 |
9 |
0 |
76% |
| src/flag_gems/ops/lt.py |
(no class) |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/margin_ranking_loss.py |
(no class) |
70 |
26 |
0 |
63% |
| src/flag_gems/ops/masked_fill.py |
(no class) |
33 |
2 |
0 |
94% |
| src/flag_gems/ops/masked_scatter.py |
(no class) |
122 |
61 |
0 |
50% |
| src/flag_gems/ops/masked_select.py |
(no class) |
105 |
54 |
0 |
49% |
| src/flag_gems/ops/max.py |
(no class) |
97 |
40 |
0 |
59% |
| src/flag_gems/ops/max_pool2d_with_indices.py |
(no class) |
141 |
73 |
0 |
48% |
| src/flag_gems/ops/maximum.py |
(no class) |
18 |
4 |
0 |
78% |
| src/flag_gems/ops/mean.py |
(no class) |
193 |
103 |
0 |
47% |
| src/flag_gems/ops/min.py |
(no class) |
96 |
40 |
0 |
58% |
| src/flag_gems/ops/minimum.py |
(no class) |
18 |
4 |
0 |
78% |
| src/flag_gems/ops/mm.py |
(no class) |
104 |
59 |
0 |
43% |
| src/flag_gems/ops/mm_streamk.py |
(no class) |
234 |
182 |
0 |
22% |
| src/flag_gems/ops/mse_loss.py |
Reduction |
0 |
0 |
0 |
100% |
| src/flag_gems/ops/mse_loss.py |
(no class) |
61 |
21 |
0 |
66% |
| src/flag_gems/ops/mul.py |
(no class) |
62 |
12 |
0 |
81% |
| src/flag_gems/ops/multinomial.py |
(no class) |
52 |
16 |
0 |
69% |
| src/flag_gems/ops/mv.py |
(no class) |
39 |
17 |
0 |
56% |
| src/flag_gems/ops/nan_to_num.py |
(no class) |
26 |
7 |
0 |
73% |
| src/flag_gems/ops/ne.py |
(no class) |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/neg.py |
(no class) |
14 |
1 |
0 |
93% |
| src/flag_gems/ops/nll_loss_nd.py |
(no class) |
138 |
121 |
0 |
12% |
| src/flag_gems/ops/nllloss.py |
(no class) |
196 |
119 |
0 |
39% |
| src/flag_gems/ops/nonzero.py |
(no class) |
46 |
13 |
0 |
72% |
| src/flag_gems/ops/normal.py |
(no class) |
61 |
4 |
0 |
93% |
| src/flag_gems/ops/one_hot.py |
(no class) |
35 |
14 |
0 |
60% |
| src/flag_gems/ops/ones.py |
(no class) |
31 |
6 |
0 |
81% |
| src/flag_gems/ops/ones_like.py |
(no class) |
18 |
0 |
0 |
100% |
| src/flag_gems/ops/pad.py |
PadFunction |
19 |
0 |
0 |
100% |
| src/flag_gems/ops/pad.py |
(no class) |
260 |
1 |
0 |
99% |
| src/flag_gems/ops/per_token_group_quant_fp8.py |
(no class) |
72 |
38 |
0 |
47% |
| src/flag_gems/ops/pixel_unshuffle.py |
(no class) |
87 |
32 |
0 |
63% |
| src/flag_gems/ops/polar.py |
(no class) |
17 |
3 |
0 |
82% |
| src/flag_gems/ops/pow.py |
(no class) |
33 |
3 |
0 |
91% |
| src/flag_gems/ops/prelu.py |
(no class) |
62 |
22 |
0 |
65% |
| src/flag_gems/ops/prod.py |
(no class) |
86 |
31 |
0 |
64% |
| src/flag_gems/ops/quantile.py |
(no class) |
153 |
85 |
0 |
44% |
| src/flag_gems/ops/rand.py |
(no class) |
48 |
22 |
0 |
54% |
| src/flag_gems/ops/rand_like.py |
(no class) |
22 |
0 |
0 |
100% |
| src/flag_gems/ops/randn.py |
(no class) |
77 |
47 |
0 |
39% |
| src/flag_gems/ops/randn_like.py |
(no class) |
22 |
0 |
0 |
100% |
| src/flag_gems/ops/randperm.py |
(no class) |
266 |
143 |
0 |
46% |
| src/flag_gems/ops/reciprocal.py |
(no class) |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/reflection_pad1d.py |
(no class) |
78 |
36 |
0 |
54% |
| src/flag_gems/ops/reflection_pad2d.py |
(no class) |
92 |
50 |
0 |
46% |
| src/flag_gems/ops/relu.py |
(no class) |
21 |
2 |
0 |
90% |
| src/flag_gems/ops/relu6.py |
(no class) |
28 |
9 |
0 |
68% |
| src/flag_gems/ops/repeat.py |
RepeatFunction |
19 |
0 |
0 |
100% |
| src/flag_gems/ops/repeat.py |
(no class) |
232 |
2 |
0 |
99% |
| src/flag_gems/ops/repeat_interleave.py |
(no class) |
85 |
22 |
0 |
74% |
| src/flag_gems/ops/replication_pad1d.py |
(no class) |
79 |
27 |
0 |
66% |
| src/flag_gems/ops/replication_pad3d.py |
(no class) |
50 |
26 |
0 |
48% |
| src/flag_gems/ops/resolve_conj.py |
(no class) |
103 |
86 |
0 |
17% |
| src/flag_gems/ops/resolve_neg.py |
(no class) |
7 |
0 |
0 |
100% |
| src/flag_gems/ops/rms_norm.py |
RmsNorm |
10 |
5 |
0 |
50% |
| src/flag_gems/ops/rms_norm.py |
(no class) |
109 |
70 |
0 |
36% |
| src/flag_gems/ops/rrelu_with_noise_backward.py |
(no class) |
41 |
18 |
0 |
56% |
| src/flag_gems/ops/rsqrt.py |
(no class) |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/scaled_softmax.py |
(no class) |
126 |
74 |
0 |
41% |
| src/flag_gems/ops/scatter.py |
ScatterFunction |
19 |
0 |
0 |
100% |
| src/flag_gems/ops/scatter.py |
(no class) |
232 |
1 |
0 |
99% |
| src/flag_gems/ops/scatter_add_.py |
ScatterFunction |
19 |
0 |
0 |
100% |
| src/flag_gems/ops/scatter_add_.py |
(no class) |
236 |
46 |
0 |
81% |
| src/flag_gems/ops/select_backward.py |
(no class) |
49 |
20 |
0 |
59% |
| src/flag_gems/ops/select_scatter.py |
(no class) |
45 |
14 |
0 |
69% |
| src/flag_gems/ops/selu.py |
(no class) |
45 |
22 |
0 |
51% |
| src/flag_gems/ops/selu_.py |
(no class) |
43 |
21 |
0 |
51% |
| src/flag_gems/ops/sgn_.py |
(no class) |
41 |
18 |
0 |
56% |
| src/flag_gems/ops/sigmoid.py |
(no class) |
29 |
5 |
0 |
83% |
| src/flag_gems/ops/silu.py |
(no class) |
32 |
8 |
0 |
75% |
| src/flag_gems/ops/sin.py |
(no class) |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/sinh_.py |
(no class) |
40 |
15 |
0 |
62% |
| src/flag_gems/ops/slice_backward.py |
(no class) |
31 |
10 |
0 |
68% |
| src/flag_gems/ops/slice_scatter.py |
(no class) |
51 |
16 |
0 |
69% |
| src/flag_gems/ops/soft_margin_loss.py |
(no class) |
125 |
78 |
0 |
38% |
| src/flag_gems/ops/softmax.py |
(no class) |
228 |
153 |
0 |
33% |
| src/flag_gems/ops/softplus.py |
(no class) |
17 |
5 |
0 |
71% |
| src/flag_gems/ops/softshrink.py |
(no class) |
52 |
20 |
0 |
62% |
| src/flag_gems/ops/sort.py |
(no class) |
210 |
123 |
0 |
41% |
| src/flag_gems/ops/special_i0e.py |
(no class) |
45 |
21 |
0 |
53% |
| src/flag_gems/ops/special_i1.py |
(no class) |
74 |
47 |
0 |
36% |
| src/flag_gems/ops/sqrt.py |
(no class) |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/stack.py |
(no class) |
86 |
34 |
0 |
60% |
| src/flag_gems/ops/std.py |
(no class) |
110 |
60 |
0 |
45% |
| src/flag_gems/ops/sub.py |
(no class) |
31 |
4 |
0 |
87% |
| src/flag_gems/ops/sum.py |
(no class) |
232 |
128 |
0 |
45% |
| src/flag_gems/ops/t_copy.py |
(no class) |
68 |
32 |
0 |
53% |
| src/flag_gems/ops/tan.py |
(no class) |
17 |
2 |
0 |
88% |
| src/flag_gems/ops/tanh.py |
(no class) |
28 |
3 |
0 |
89% |
| src/flag_gems/ops/threshold.py |
(no class) |
21 |
2 |
0 |
90% |
| src/flag_gems/ops/tile.py |
TileFunction |
19 |
0 |
0 |
100% |
| src/flag_gems/ops/tile.py |
(no class) |
236 |
2 |
0 |
99% |
| src/flag_gems/ops/to.py |
(no class) |
49 |
6 |
0 |
88% |
| src/flag_gems/ops/topk.py |
(no class) |
183 |
109 |
0 |
40% |
| src/flag_gems/ops/trace.py |
(no class) |
55 |
23 |
0 |
58% |
| src/flag_gems/ops/tril.py |
(no class) |
41 |
20 |
0 |
51% |
| src/flag_gems/ops/triu.py |
(no class) |
106 |
36 |
0 |
66% |
| src/flag_gems/ops/unfold_backward.py |
(no class) |
45 |
16 |
0 |
64% |
| src/flag_gems/ops/uniform.py |
(no class) |
41 |
20 |
0 |
51% |
| src/flag_gems/ops/unique.py |
(no class) |
287 |
146 |
0 |
49% |
| src/flag_gems/ops/upsample_bicubic2d.py |
(no class) |
127 |
92 |
0 |
28% |
| src/flag_gems/ops/upsample_bicubic2d_aa.py |
(no class) |
178 |
141 |
0 |
21% |
| src/flag_gems/ops/upsample_linear1d.py |
(no class) |
50 |
23 |
0 |
54% |
| src/flag_gems/ops/upsample_nearest1d.py |
(no class) |
51 |
23 |
0 |
55% |
| src/flag_gems/ops/upsample_nearest2d.py |
(no class) |
58 |
28 |
0 |
52% |
| src/flag_gems/ops/upsample_nearest3d.py |
(no class) |
63 |
32 |
0 |
49% |
| src/flag_gems/ops/var_mean.py |
(no class) |
115 |
59 |
0 |
49% |
| src/flag_gems/ops/vdot.py |
(no class) |
140 |
66 |
0 |
53% |
| src/flag_gems/ops/vector_norm.py |
(no class) |
259 |
146 |
0 |
44% |
| src/flag_gems/ops/vstack.py |
(no class) |
76 |
20 |
0 |
74% |
| src/flag_gems/ops/w8a8_block_fp8_matmul.py |
(no class) |
82 |
42 |
0 |
49% |
| src/flag_gems/ops/weightnorm.py |
(no class) |
160 |
114 |
0 |
29% |
| src/flag_gems/ops/where.py |
(no class) |
48 |
5 |
0 |
90% |
| src/flag_gems/ops/zero.py |
(no class) |
47 |
18 |
0 |
62% |
| src/flag_gems/ops/zeros.py |
(no class) |
35 |
6 |
0 |
83% |
| src/flag_gems/ops/zeros_like.py |
(no class) |
18 |
0 |
0 |
100% |
| src/flag_gems/patches/__init__.py |
(no class) |
3 |
0 |
0 |
100% |
| src/flag_gems/patches/patch_util.py |
(no class) |
59 |
21 |
0 |
64% |
| src/flag_gems/patches/patch_vllm_all.py |
(no class) |
185 |
160 |
0 |
14% |
| src/flag_gems/runtime/__init__.py |
(no class) |
30 |
10 |
0 |
67% |
| src/flag_gems/runtime/backend/__init__.py |
BackendArchEvent |
78 |
18 |
0 |
77% |
| src/flag_gems/runtime/backend/__init__.py |
(no class) |
139 |
21 |
0 |
85% |
| src/flag_gems/runtime/backend/_aipu/__init__.py |
(no class) |
6 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_aipu/fused/__init__.py |
(no class) |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_aipu/heuristics_config_utils.py |
(no class) |
136 |
136 |
0 |
0% |
| src/flag_gems/runtime/backend/_aipu/ops/__init__.py |
(no class) |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_aipu/ops/cumsum.py |
(no class) |
258 |
258 |
0 |
0% |
| src/flag_gems/runtime/backend/_aipu/ops/multinomial.py |
(no class) |
52 |
52 |
0 |
0% |
| src/flag_gems/runtime/backend/_amd/__init__.py |
(no class) |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_amd/heuristics_config_utils.py |
(no class) |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_amd/ops/__init__.py |
(no class) |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/__init__.py |
(no class) |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/heuristics_config_utils.py |
(no class) |
122 |
122 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/__init__.py |
(no class) |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/add.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/gelu.py |
Gelu |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/gelu.py |
(no class) |
46 |
46 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/__init__.py |
(no class) |
14 |
3 |
0 |
79% |
| src/flag_gems/runtime/backend/_ascend/fla/__init__.py |
(no class) |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/chunk.py |
(no class) |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/chunk_delta_h.py |
(no class) |
112 |
112 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/chunk_o.py |
(no class) |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/chunk_scaled_dot_kkt.py |
(no class) |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/cumsum.py |
(no class) |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/fused_qkvzba_split_reshape.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/fused_recurrent.py |
(no class) |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/layernorm_guard.py |
LayerNormFn |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/layernorm_guard.py |
(no class) |
82 |
82 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/sigmoid_gating.py |
(no class) |
95 |
95 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/solve_tril.py |
(no class) |
180 |
180 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/utils.py |
(no class) |
13 |
13 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/wy_fast.py |
(no class) |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/__init__.py |
(no class) |
5 |
5 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/cross_entropy_loss.py |
CrossEntropyLoss |
64 |
64 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/cross_entropy_loss.py |
(no class) |
316 |
316 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/fused_add_rms_norm.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/rotary_embedding.py |
(no class) |
86 |
86 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/skip_layernorm.py |
SkipLayerNorm |
13 |
13 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/skip_layernorm.py |
(no class) |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/heuristics_config_utils.py |
(no class) |
143 |
143 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/__init__.py |
(no class) |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/addmm.py |
(no class) |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/all.py |
(no class) |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/amax.py |
(no class) |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/angle.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/any.py |
(no class) |
106 |
106 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/arange.py |
(no class) |
38 |
38 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/argmax.py |
(no class) |
148 |
148 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/argmin.py |
(no class) |
101 |
101 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/bmm.py |
(no class) |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/cat.py |
(no class) |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/count_nonzero.py |
(no class) |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/cumsum.py |
(no class) |
321 |
321 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/diag.py |
(no class) |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/diag_embed.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/diagonal.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/dot.py |
(no class) |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/embedding.py |
Embedding |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/embedding.py |
(no class) |
74 |
74 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/exponential_.py |
(no class) |
86 |
86 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/fill.py |
(no class) |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/flip.py |
(no class) |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/full.py |
(no class) |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/full_like.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/gather.py |
(no class) |
81 |
81 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/groupnorm.py |
(no class) |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/hstack.py |
(no class) |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index.py |
(no class) |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index_add.py |
IndexAddFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index_add.py |
(no class) |
126 |
126 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index_select.py |
(no class) |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/isin.py |
(no class) |
132 |
132 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/linspace.py |
(no class) |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/log_softmax.py |
(no class) |
99 |
99 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/masked_fill.py |
(no class) |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/masked_select.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/max.py |
(no class) |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/mean.py |
(no class) |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/min.py |
(no class) |
101 |
101 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/mm.py |
(no class) |
76 |
76 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/multinomial.py |
(no class) |
52 |
52 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/ones.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/ones_like.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/outer.py |
Outer |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/outer.py |
(no class) |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/polar.py |
(no class) |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/pow.py |
(no class) |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/randperm.py |
(no class) |
273 |
273 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/repeat_interleave.py |
(no class) |
39 |
39 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/resolve_neg.py |
(no class) |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/rms_norm.py |
RmsNorm |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/rms_norm.py |
(no class) |
88 |
88 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/select_scatter.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/slice_scatter.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/softmax.py |
(no class) |
213 |
213 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/sort.py |
(no class) |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/stack.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/threshold.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/triu.py |
(no class) |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/unique.py |
(no class) |
314 |
314 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/upsample_nearest2d.py |
(no class) |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/var_mean.py |
(no class) |
155 |
155 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/vector_norm.py |
(no class) |
267 |
267 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/vstack.py |
(no class) |
80 |
80 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/where.py |
(no class) |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/zeros.py |
(no class) |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/zeros_like.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/utils/__init__.py |
(no class) |
8 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_cambricon/__init__.py |
(no class) |
18 |
16 |
0 |
11% |
| src/flag_gems/runtime/backend/_cambricon/fused/__init__.py |
(no class) |
9 |
9 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/cross_entropy_loss.py |
CrossEntropyLoss |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/cross_entropy_loss.py |
(no class) |
417 |
417 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/flash_mla.py |
(no class) |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/fused_add_rms_norm.py |
(no class) |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/gelu_and_mul.py |
GeluAndMul |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/gelu_and_mul.py |
(no class) |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/outer.py |
Outer |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/outer.py |
(no class) |
64 |
64 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/silu_and_mul.py |
SiluAndMul |
6 |
6 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/silu_and_mul.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/skip_layernorm.py |
SkipLayerNorm |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/skip_layernorm.py |
(no class) |
102 |
102 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/weight_norm.py |
WeightNorm |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/weight_norm.py |
(no class) |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/heuristics_config_utils.py |
(no class) |
110 |
110 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/__init__.py |
(no class) |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/abs.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/acos.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/add.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/addcdiv.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/addcmul.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/addmm.py |
(no class) |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/all.py |
(no class) |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/amax.py |
(no class) |
133 |
133 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/any.py |
(no class) |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/arange.py |
(no class) |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/argmax.py |
(no class) |
133 |
133 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/atan.py |
(no class) |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/attention.py |
ScaleDotProductAttention |
13 |
13 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/attention.py |
(no class) |
384 |
384 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/avg_pool2d.py |
(no class) |
177 |
177 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_and.py |
(no class) |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_left_shift.py |
(no class) |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_not.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_or.py |
(no class) |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_right_shift.py |
(no class) |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bmm.py |
(no class) |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cat.py |
CatKernelGenerator |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cat.py |
(no class) |
74 |
74 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ceil.py |
(no class) |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/celu.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/clamp.py |
(no class) |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/contiguous.py |
(no class) |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/copy.py |
(no class) |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cos.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/count_nonzero.py |
(no class) |
113 |
113 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cummin.py |
(no class) |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cumsum.py |
(no class) |
325 |
325 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/diag.py |
(no class) |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/diag_embed.py |
(no class) |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/diagonal.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/div.py |
(no class) |
182 |
182 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/dropout.py |
(no class) |
84 |
84 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/elu.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/embedding.py |
(no class) |
84 |
84 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/eq.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/erf.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/exp.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/exp2.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/exponential_.py |
(no class) |
69 |
69 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/fill.py |
(no class) |
69 |
69 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/flip.py |
FlipKernelCode |
250 |
250 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/flip.py |
(no class) |
25 |
25 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/full.py |
(no class) |
70 |
70 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/full_like.py |
(no class) |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/gather.py |
GatherFunction |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/gather.py |
(no class) |
135 |
135 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ge.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/gelu.py |
(no class) |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/glu.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/groupnorm.py |
(no class) |
343 |
343 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/gt.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/hstack.py |
(no class) |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/index_add.py |
IndexAddFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/index_add.py |
(no class) |
174 |
174 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/index_select.py |
(no class) |
167 |
167 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isclose.py |
(no class) |
40 |
40 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isfinite.py |
(no class) |
24 |
24 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isin.py |
(no class) |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isinf.py |
(no class) |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isnan.py |
(no class) |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/kron.py |
(no class) |
121 |
121 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/layernorm.py |
(no class) |
341 |
341 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/le.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/linspace.py |
(no class) |
39 |
39 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/log.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/log_sigmoid.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/log_softmax.py |
(no class) |
556 |
556 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logical_and.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logical_not.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logical_or.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logical_xor.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logspace.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/lt.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/masked_fill.py |
(no class) |
73 |
73 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/masked_select.py |
(no class) |
73 |
73 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/max.py |
(no class) |
177 |
177 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/max_pool2d_with_indices.py |
(no class) |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/maximum.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/mean.py |
(no class) |
82 |
82 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/min.py |
(no class) |
170 |
170 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/minimum.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/mm.py |
(no class) |
97 |
97 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/mul.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/multinomial.py |
(no class) |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/mv.py |
(no class) |
67 |
67 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ne.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/neg.py |
(no class) |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/nonzero.py |
(no class) |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/normal.py |
(no class) |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ones.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ones_like.py |
(no class) |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/pad.py |
PadFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/pad.py |
(no class) |
325 |
325 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/per_token_group_quant_fp8.py |
(no class) |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/pow.py |
(no class) |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/prod.py |
(no class) |
97 |
97 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/quantile.py |
(no class) |
225 |
225 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rand.py |
(no class) |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rand_like.py |
(no class) |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/randn.py |
(no class) |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/randn_like.py |
(no class) |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/randperm.py |
(no class) |
329 |
329 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/reciprocal.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/relu.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/repeat.py |
RepeatFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/repeat.py |
(no class) |
284 |
284 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/repeat_interleave.py |
RepeatInterleaveSelfIntFn |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/repeat_interleave.py |
(no class) |
81 |
81 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/resolve_conj.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/resolve_neg.py |
(no class) |
7 |
7 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rms_norm.py |
RmsNorm |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rms_norm.py |
(no class) |
189 |
189 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rsqrt.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/scatter.py |
ScatterFunction |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/scatter.py |
(no class) |
162 |
162 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/select_scatter.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sigmoid.py |
(no class) |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/silu.py |
(no class) |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sin.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/slice_scatter.py |
(no class) |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/softmax.py |
(no class) |
546 |
546 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/softplus.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sort.py |
(no class) |
207 |
207 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sqrt.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/stack.py |
StackKernelCode |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/stack.py |
(no class) |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sub.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sum.py |
(no class) |
142 |
142 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/tan.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/tanh.py |
(no class) |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/tile.py |
TileFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/tile.py |
(no class) |
287 |
287 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/to.py |
(no class) |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/topk.py |
(no class) |
174 |
174 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/triu.py |
(no class) |
140 |
140 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/uniform.py |
(no class) |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/unique.py |
(no class) |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/upsample_nearest2d.py |
(no class) |
108 |
108 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/var_mean.py |
(no class) |
160 |
160 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/vector_norm.py |
(no class) |
308 |
308 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/vstack.py |
VstackKernelCode |
160 |
160 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/vstack.py |
(no class) |
24 |
24 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/weightnorm.py |
(no class) |
228 |
228 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/where.py |
(no class) |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/zeros.py |
(no class) |
40 |
40 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/zeros_like.py |
(no class) |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/__init__.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/pointwise_dynamic.py |
FunctionSchema |
82 |
82 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/pointwise_dynamic.py |
KernelGenerator |
458 |
458 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/pointwise_dynamic.py |
WrapperGenerator |
235 |
235 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/pointwise_dynamic.py |
ModuleGenerator |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/pointwise_dynamic.py |
PointwiseDynamicFunction |
85 |
85 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/pointwise_dynamic.py |
(no class) |
115 |
115 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/reduce_utils.py |
(no class) |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/__init__.py |
(no class) |
15 |
9 |
0 |
40% |
| src/flag_gems/runtime/backend/_enflame/fused/__init__.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/heuristics_config_utils.py |
(no class) |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/ops/__init__.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/__init__.py |
(no class) |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_hygon/fused/__init__.py |
(no class) |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/heuristics_config_utils.py |
(no class) |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/__init__.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/all.py |
(no class) |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/any.py |
(no class) |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/attention.py |
ScaleDotProductAttention |
13 |
13 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/attention.py |
(no class) |
386 |
386 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/div.py |
(no class) |
178 |
178 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/exponential_.py |
(no class) |
93 |
93 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/fill.py |
(no class) |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/flash_api.py |
fwd_params |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/flash_api.py |
(no class) |
317 |
317 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/flash_kernel.py |
(no class) |
551 |
551 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/gelu.py |
Gelu |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/gelu.py |
InplaceGelu |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/gelu.py |
(no class) |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/isclose.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/isin.py |
(no class) |
132 |
132 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/mm.py |
(no class) |
94 |
94 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/pow.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/randperm.py |
(no class) |
266 |
266 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/silu.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/sort.py |
(no class) |
182 |
182 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/unique.py |
(no class) |
287 |
287 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/upsample_nearest2d.py |
(no class) |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/__init__.py |
(no class) |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_iluvatar/heuristics_config_utils.py |
(no class) |
131 |
131 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/__init__.py |
(no class) |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/div.py |
(no class) |
166 |
166 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/__init__.py |
(no class) |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/__init__.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/concat_and_cache_mla.py |
ConcatAndCacheMla |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/concat_and_cache_mla.py |
(no class) |
53 |
53 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/cross_entropy_loss.py |
CrossEntropyLoss |
116 |
116 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/cross_entropy_loss.py |
(no class) |
315 |
315 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/flash_mla.py |
(no class) |
99 |
99 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/fused_add_rms_norm.py |
(no class) |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/geglu.py |
(no class) |
68 |
68 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/gelu_and_mul.py |
GeluAndMul |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/gelu_and_mul.py |
(no class) |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/instance_norm.py |
InstanceNorm |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/instance_norm.py |
(no class) |
287 |
287 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/moe_align_block_size.py |
(no class) |
76 |
76 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/outer.py |
Outer |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/outer.py |
(no class) |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/reglu.py |
(no class) |
83 |
83 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/reshape_and_cache.py |
(no class) |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/reshape_and_cache_flash.py |
(no class) |
39 |
39 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/rotary_embedding.py |
(no class) |
128 |
128 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/rwkv_ka_fusion.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/rwkv_mm_sparsity.py |
(no class) |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/silu_and_mul.py |
SiluAndMul |
6 |
6 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/silu_and_mul.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/skip_layernorm.py |
SkipLayerNorm |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/skip_layernorm.py |
(no class) |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/topk_softmax.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/weight_norm.py |
WeightNorm |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/weight_norm.py |
(no class) |
110 |
110 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/heuristics_config_utils.py |
(no class) |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/__init__.py |
(no class) |
170 |
170 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/abs.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/acos.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/add.py |
(no class) |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addcdiv.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addcmul.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addmm.py |
(no class) |
85 |
85 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addmv.py |
(no class) |
61 |
61 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addr.py |
(no class) |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/all.py |
(no class) |
160 |
160 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/amax.py |
(no class) |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/angle.py |
(no class) |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/any.py |
(no class) |
160 |
160 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/arange.py |
(no class) |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/argmax.py |
(no class) |
137 |
137 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/argmin.py |
(no class) |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/atan.py |
(no class) |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/attention.py |
ScaleDotProductAttention |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/attention.py |
(no class) |
338 |
338 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/avg_pool2d.py |
(no class) |
150 |
150 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/baddbmm.py |
BaddbmmFunction |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/baddbmm.py |
(no class) |
119 |
119 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/batch_norm.py |
(no class) |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_and.py |
(no class) |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_left_shift.py |
(no class) |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_not.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_or.py |
(no class) |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_right_shift.py |
(no class) |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bmm.py |
(no class) |
102 |
102 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cat.py |
(no class) |
53 |
53 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/celu.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/clamp.py |
(no class) |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/contiguous.py |
(no class) |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv1d.py |
(no class) |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv2d.py |
Conv2d |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv2d.py |
(no class) |
109 |
109 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv3d.py |
(no class) |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv_depthwise2d.py |
(no class) |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/copy.py |
(no class) |
64 |
64 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cos.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/count_nonzero.py |
(no class) |
140 |
140 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cummax.py |
(no class) |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cummin.py |
(no class) |
243 |
243 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cumsum.py |
(no class) |
269 |
269 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/diag.py |
(no class) |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/diag_embed.py |
(no class) |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/diagonal.py |
(no class) |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/div.py |
(no class) |
179 |
179 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/dot.py |
(no class) |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/dropout.py |
(no class) |
97 |
97 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/elu.py |
(no class) |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/embedding.py |
(no class) |
98 |
98 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/eq.py |
(no class) |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/erf.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/exp.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/exp2.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/exponential_.py |
(no class) |
88 |
88 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/eye.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/eye_m.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/fill.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/flash_api.py |
fwd_params |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/flash_api.py |
(no class) |
306 |
306 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/flash_kernel.py |
(no class) |
534 |
534 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/flip.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/full.py |
(no class) |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/full_like.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/gather.py |
GatherFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/gather.py |
(no class) |
170 |
170 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ge.py |
(no class) |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/gelu.py |
(no class) |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/get_scheduler_metadata.py |
(no class) |
277 |
277 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/glu.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/groupnorm.py |
(no class) |
187 |
187 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/gt.py |
(no class) |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/hstack.py |
(no class) |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index.py |
IndexFunction |
25 |
25 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index.py |
(no class) |
268 |
268 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_add.py |
(no class) |
80 |
80 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_put.py |
IndexPutFunction |
25 |
25 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_put.py |
(no class) |
227 |
227 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_select.py |
(no class) |
46 |
46 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isclose.py |
(no class) |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isfinite.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isin.py |
(no class) |
138 |
138 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isinf.py |
(no class) |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isnan.py |
(no class) |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/kron.py |
(no class) |
117 |
117 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/layernorm.py |
(no class) |
295 |
295 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/le.py |
(no class) |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/lerp.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/linspace.py |
(no class) |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/log.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/log_sigmoid.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/log_softmax.py |
(no class) |
105 |
105 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logical_and.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logical_not.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logical_or.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logical_xor.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logspace.py |
(no class) |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/lt.py |
(no class) |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/masked_fill.py |
(no class) |
84 |
84 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/masked_scatter.py |
(no class) |
122 |
122 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/masked_select.py |
(no class) |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/max.py |
(no class) |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/max_pool2d_with_indices.py |
(no class) |
145 |
145 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/maximum.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mean.py |
(no class) |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/min.py |
(no class) |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/minimum.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mm.py |
(no class) |
108 |
108 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mse_loss.py |
Reduction |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mse_loss.py |
(no class) |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mul.py |
(no class) |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/multinomial.py |
(no class) |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mv.py |
(no class) |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/nan_to_num.py |
(no class) |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ne.py |
(no class) |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/neg.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/nllloss.py |
(no class) |
171 |
171 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/nonzero.py |
(no class) |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/normal.py |
(no class) |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ones.py |
(no class) |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ones_like.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/pad.py |
PadFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/pad.py |
(no class) |
263 |
263 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/per_token_group_quant_fp8.py |
(no class) |
67 |
67 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/polar.py |
(no class) |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/pow.py |
(no class) |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/prod.py |
(no class) |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/quantile.py |
(no class) |
83 |
83 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rand.py |
(no class) |
131 |
131 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rand_like.py |
(no class) |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/randn.py |
(no class) |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/randn_like.py |
(no class) |
24 |
24 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/randperm.py |
(no class) |
269 |
269 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/reciprocal.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/relu.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/repeat.py |
RepeatFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/repeat.py |
(no class) |
232 |
232 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/repeat_interleave.py |
(no class) |
83 |
83 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/resolve_conj.py |
(no class) |
6 |
6 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/resolve_neg.py |
(no class) |
7 |
7 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rms_norm.py |
RmsNorm |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rms_norm.py |
(no class) |
203 |
203 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rsqrt.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rsub.py |
(no class) |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/scaled_softmax.py |
(no class) |
122 |
122 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/scatter.py |
ScatterFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/scatter.py |
(no class) |
220 |
220 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/scatter_add_.py |
ScatterFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/scatter_add_.py |
(no class) |
236 |
236 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/select_scatter.py |
(no class) |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sigmoid.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/silu.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sin.py |
(no class) |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/slice_scatter.py |
(no class) |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/softmax.py |
(no class) |
180 |
180 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/softplus.py |
(no class) |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sort.py |
(no class) |
307 |
307 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sqrt.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/stack.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/std.py |
(no class) |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sub.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sum.py |
(no class) |
148 |
148 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/tan.py |
(no class) |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/tanh.py |
(no class) |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/threshold.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/tile.py |
TileFunction |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/tile.py |
(no class) |
236 |
236 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/to.py |
(no class) |
67 |
67 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/topk.py |
(no class) |
187 |
187 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/trace.py |
(no class) |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/triu.py |
(no class) |
68 |
68 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/uniform.py |
(no class) |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/unique.py |
(no class) |
524 |
524 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/upsample_bicubic2d_aa.py |
(no class) |
200 |
200 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/upsample_nearest1d.py |
(no class) |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/upsample_nearest2d.py |
(no class) |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/var_mean.py |
(no class) |
120 |
120 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/vdot.py |
(no class) |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/vector_norm.py |
(no class) |
263 |
263 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/vstack.py |
(no class) |
77 |
77 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/weightnorm.py |
(no class) |
171 |
171 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/where.py |
(no class) |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/zeros.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/zeros_like.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/__init__.py |
(no class) |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_metax/fused/__init__.py |
(no class) |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/fused/flash_mla.py |
(no class) |
104 |
104 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/heuristics_config_utils.py |
(no class) |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/__init__.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/addmm.py |
(no class) |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/amax.py |
(no class) |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/arange.py |
(no class) |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/bmm.py |
(no class) |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/exponential_.py |
(no class) |
176 |
176 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/full.py |
(no class) |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/full_like.py |
(no class) |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/groupnorm.py |
GroupNorm |
40 |
40 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/groupnorm.py |
(no class) |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/index.py |
IndexFunction |
25 |
25 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/index.py |
(no class) |
256 |
256 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/index_put.py |
IndexPutFunction |
25 |
25 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/index_put.py |
(no class) |
207 |
207 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/index_select.py |
(no class) |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/isin.py |
(no class) |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/log_softmax.py |
(no class) |
98 |
98 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/masked_fill.py |
(no class) |
65 |
65 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/min.py |
(no class) |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/mm.py |
(no class) |
116 |
116 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/nonzero.py |
NonzeroFunction |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/nonzero.py |
(no class) |
109 |
109 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/ones.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/ones_like.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/outer.py |
Outer |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/outer.py |
(no class) |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/polar.py |
(no class) |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/prod.py |
(no class) |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/repeat_interleave.py |
(no class) |
117 |
117 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/resolve_conj.py |
(no class) |
7 |
7 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/sigmoid.py |
Sigmoid |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/sigmoid.py |
(no class) |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/tanh.py |
Tanh |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/tanh.py |
(no class) |
40 |
40 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/unique.py |
(no class) |
284 |
284 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/upsample_nearest2d.py |
(no class) |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/zeros.py |
(no class) |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/zeros_like.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/__init__.py |
(no class) |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_mthreads/fused/__init__.py |
(no class) |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/fused/cross_entropy_loss.py |
CrossEntropyLoss |
64 |
64 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/fused/cross_entropy_loss.py |
(no class) |
312 |
312 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/heuristics_config_utils.py |
(no class) |
239 |
239 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/__init__.py |
(no class) |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/addmm.py |
(no class) |
120 |
120 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/all.py |
(no class) |
140 |
140 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/amax.py |
(no class) |
143 |
143 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/any.py |
(no class) |
135 |
135 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/arange.py |
(no class) |
82 |
82 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/argmin.py |
(no class) |
121 |
121 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/batch_norm.py |
(no class) |
283 |
283 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/bmm.py |
(no class) |
137 |
137 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/celu.py |
(no class) |
82 |
82 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/conv2d.py |
Conv2d |
73 |
73 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/conv2d.py |
(no class) |
88 |
88 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/dropout.py |
(no class) |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/gather.py |
(no class) |
71 |
71 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/gelu.py |
Gelu |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/gelu.py |
(no class) |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/index_add.py |
(no class) |
79 |
79 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/index_put.py |
IndexPutFunction |
25 |
25 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/index_put.py |
(no class) |
219 |
219 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/index_select.py |
(no class) |
101 |
101 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/isin.py |
(no class) |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/log.py |
(no class) |
57 |
57 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/log_softmax.py |
(no class) |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/max.py |
(no class) |
150 |
150 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/min.py |
(no class) |
123 |
123 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/mm.py |
(no class) |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/normal.py |
(no class) |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/one_hot.py |
(no class) |
104 |
104 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/ones.py |
(no class) |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/ones_like.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/prod.py |
(no class) |
188 |
188 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/rand.py |
(no class) |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/rand_like.py |
(no class) |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/randn.py |
(no class) |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/randn_like.py |
(no class) |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/randperm.py |
(no class) |
266 |
266 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/repeat.py |
(no class) |
176 |
176 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/repeat_interleave.py |
(no class) |
225 |
225 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/resolve_conj.py |
(no class) |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/sort.py |
(no class) |
210 |
210 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/tanh.py |
Tanh |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/tanh.py |
(no class) |
23 |
23 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/tile.py |
(no class) |
174 |
174 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/unique.py |
(no class) |
284 |
284 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/utils.py |
(no class) |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/zeros.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/zeros_like.py |
(no class) |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/__init__.py |
(no class) |
5 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/ampere/__init__.py |
(no class) |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/ampere/ops/__init__.py |
(no class) |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/fused/__init__.py |
(no class) |
2 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/fused/fused_add_rms_norm.py |
(no class) |
37 |
25 |
0 |
32% |
| src/flag_gems/runtime/backend/_nvidia/heuristics_config_utils.py |
(no class) |
243 |
41 |
0 |
83% |
| src/flag_gems/runtime/backend/_nvidia/hopper/__init__.py |
(no class) |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/hopper/grouped_topk.py |
(no class) |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/heuristics_config_utils.py |
(no class) |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/__init__.py |
(no class) |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/mm.py |
(no class) |
265 |
265 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/ops/__init__.py |
(no class) |
3 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/ops/add.py |
(no class) |
20 |
14 |
0 |
30% |
| src/flag_gems/runtime/backend/_nvidia/ops/gelu.py |
Gelu |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/ops/gelu.py |
(no class) |
46 |
18 |
0 |
61% |
| src/flag_gems/runtime/backend/_sunrise/__init__.py |
(no class) |
5 |
4 |
0 |
20% |
| src/flag_gems/runtime/backend/_sunrise/fused/__init__.py |
(no class) |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/heuristics_config_utils.py |
(no class) |
174 |
174 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/__init__.py |
(no class) |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/__init__.py |
TxdaDeviceProperties |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/__init__.py |
(no class) |
27 |
22 |
0 |
19% |
| src/flag_gems/runtime/backend/_tsingmicro/fused/__init__.py |
(no class) |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/fused/cross_entropy_loss.py |
CrossEntropyLoss |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/fused/cross_entropy_loss.py |
(no class) |
416 |
416 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/heuristics_config_utils.py |
(no class) |
215 |
215 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/__init__.py |
(no class) |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/argmax.py |
(no class) |
132 |
132 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/cat.py |
(no class) |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/count_nonzero.py |
(no class) |
113 |
113 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/hstack.py |
(no class) |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/isin.py |
(no class) |
126 |
126 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/kron.py |
(no class) |
120 |
120 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/masked_select.py |
(no class) |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/mm.py |
(no class) |
94 |
94 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/rms_norm.py |
RmsNorm |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/rms_norm.py |
(no class) |
130 |
130 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/stack.py |
(no class) |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/unique.py |
(no class) |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/backend_utils.py |
VendorInfoBase |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/backend_utils.py |
(no class) |
30 |
5 |
0 |
83% |
| src/flag_gems/runtime/backend/device.py |
DeviceDetector |
62 |
8 |
0 |
87% |
| src/flag_gems/runtime/backend/device.py |
(no class) |
21 |
0 |
0 |
100% |
| src/flag_gems/runtime/common.py |
vendors |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/common.py |
(no class) |
17 |
0 |
0 |
100% |
| src/flag_gems/runtime/configloader.py |
ConfigLoader |
104 |
15 |
0 |
86% |
| src/flag_gems/runtime/configloader.py |
(no class) |
19 |
0 |
0 |
100% |
| src/flag_gems/runtime/error.py |
(no class) |
8 |
4 |
0 |
50% |
| src/flag_gems/runtime/register.py |
Register |
64 |
24 |
0 |
62% |
| src/flag_gems/runtime/register.py |
(no class) |
15 |
0 |
0 |
100% |
| src/flag_gems/testing/__init__.py |
(no class) |
29 |
3 |
0 |
90% |
| src/flag_gems/utils/__init__.py |
(no class) |
6 |
0 |
0 |
100% |
| src/flag_gems/utils/code_cache.py |
(no class) |
26 |
3 |
0 |
88% |
| src/flag_gems/utils/code_utils.py |
IndentedBuffer |
30 |
8 |
0 |
73% |
| src/flag_gems/utils/code_utils.py |
NameSpace |
29 |
29 |
0 |
0% |
| src/flag_gems/utils/code_utils.py |
(no class) |
37 |
1 |
0 |
97% |
| src/flag_gems/utils/codegen_config_utils.py |
CodeGenConfig |
2 |
0 |
0 |
100% |
| src/flag_gems/utils/codegen_config_utils.py |
(no class) |
56 |
22 |
0 |
61% |
| src/flag_gems/utils/device_info.py |
DeviceInfo |
0 |
0 |
0 |
100% |
| src/flag_gems/utils/device_info.py |
(no class) |
59 |
18 |
0 |
69% |
| src/flag_gems/utils/libentry.py |
Cache |
3 |
0 |
0 |
100% |
| src/flag_gems/utils/libentry.py |
ConfigCache |
9 |
3 |
0 |
67% |
| src/flag_gems/utils/libentry.py |
BenchmarkCache |
10 |
10 |
0 |
0% |
| src/flag_gems/utils/libentry.py |
LibCache |
34 |
13 |
0 |
62% |
| src/flag_gems/utils/libentry.py |
LibTuner |
93 |
31 |
0 |
67% |
| src/flag_gems/utils/libentry.py |
LibTuner.register_policy.decorator.AnonymousLibTunerImpl |
2 |
1 |
0 |
50% |
| src/flag_gems/utils/libentry.py |
LibEntry |
116 |
10 |
0 |
91% |
| src/flag_gems/utils/libentry.py |
(no class) |
112 |
6 |
0 |
95% |
| src/flag_gems/utils/limits.py |
(no class) |
29 |
23 |
0 |
21% |
| src/flag_gems/utils/models/__init__.py |
(no class) |
3 |
0 |
0 |
100% |
| src/flag_gems/utils/models/model.py |
PersistantModel |
11 |
9 |
0 |
18% |
| src/flag_gems/utils/models/model.py |
(no class) |
26 |
0 |
0 |
100% |
| src/flag_gems/utils/models/session.py |
RollbackSession |
5 |
4 |
0 |
20% |
| src/flag_gems/utils/models/session.py |
(no class) |
7 |
0 |
0 |
100% |
| src/flag_gems/utils/models/sql.py |
Base |
0 |
0 |
0 |
100% |
| src/flag_gems/utils/models/sql.py |
SQLPersistantModel |
73 |
46 |
0 |
37% |
| src/flag_gems/utils/models/sql.py |
(no class) |
29 |
0 |
0 |
100% |
| src/flag_gems/utils/pointwise_dynamic.py |
FunctionSchema |
82 |
11 |
0 |
87% |
| src/flag_gems/utils/pointwise_dynamic.py |
KernelGenerator |
286 |
7 |
0 |
98% |
| src/flag_gems/utils/pointwise_dynamic.py |
WrapperGenerator |
189 |
6 |
0 |
97% |
| src/flag_gems/utils/pointwise_dynamic.py |
ModuleGenerator |
85 |
5 |
0 |
94% |
| src/flag_gems/utils/pointwise_dynamic.py |
KernelInfo |
0 |
0 |
0 |
100% |
| src/flag_gems/utils/pointwise_dynamic.py |
PointwiseDynamicFunction |
96 |
8 |
0 |
92% |
| src/flag_gems/utils/pointwise_dynamic.py |
(no class) |
124 |
5 |
0 |
96% |
| src/flag_gems/utils/pointwise_dynamic_backup.py |
FunctionSchema |
82 |
82 |
0 |
0% |
| src/flag_gems/utils/pointwise_dynamic_backup.py |
KernelGenerator |
287 |
287 |
0 |
0% |
| src/flag_gems/utils/pointwise_dynamic_backup.py |
WrapperGenerator |
189 |
189 |
0 |
0% |
| src/flag_gems/utils/pointwise_dynamic_backup.py |
ModuleGenerator |
28 |
28 |
0 |
0% |
| src/flag_gems/utils/pointwise_dynamic_backup.py |
KernelInfo |
0 |
0 |
0 |
100% |
| src/flag_gems/utils/pointwise_dynamic_backup.py |
PointwiseDynamicFunction |
96 |
96 |
0 |
0% |
| src/flag_gems/utils/pointwise_dynamic_backup.py |
(no class) |
123 |
123 |
0 |
0% |
| src/flag_gems/utils/random_utils.py |
(no class) |
60 |
29 |
0 |
52% |
| src/flag_gems/utils/shape_utils.py |
MemOverlap |
0 |
0 |
0 |
100% |
| src/flag_gems/utils/shape_utils.py |
(no class) |
245 |
73 |
0 |
70% |
| src/flag_gems/utils/tensor_wrapper.py |
TypedPtr |
6 |
1 |
0 |
83% |
| src/flag_gems/utils/tensor_wrapper.py |
StridedBuffer |
36 |
19 |
0 |
47% |
| src/flag_gems/utils/tensor_wrapper.py |
(no class) |
23 |
0 |
0 |
100% |
| src/flag_gems/utils/triton_driver_helper.py |
(no class) |
6 |
3 |
0 |
50% |
| src/flag_gems/utils/triton_lang_extension.py |
(no class) |
56 |
29 |
0 |
48% |
| src/flag_gems/utils/triton_lang_helper.py |
(no class) |
26 |
9 |
0 |
65% |
| src/flag_gems/utils/type_utils.py |
(no class) |
8 |
0 |
0 |
100% |