| src/flag_gems/__init__.py |
73 |
8 |
0 |
89% |
| src/flag_gems/config.py |
77 |
33 |
2 |
57% |
| src/flag_gems/experimental_ops/__init__.py |
2 |
0 |
0 |
100% |
| src/flag_gems/experimental_ops/_adaptive_avg_pool3d.py |
114 |
114 |
0 |
0% |
| src/flag_gems/experimental_ops/_functional_sym_constrain_range_for_size.py |
30 |
30 |
0 |
0% |
| src/flag_gems/experimental_ops/_log_softmax_backward_data.py |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/_safe_softmax.py |
53 |
53 |
0 |
0% |
| src/flag_gems/experimental_ops/_unsafe_view.py |
76 |
76 |
0 |
0% |
| src/flag_gems/experimental_ops/_upsample_nearest_exact1d.py |
128 |
128 |
0 |
0% |
| src/flag_gems/experimental_ops/abs.py |
101 |
101 |
0 |
0% |
| src/flag_gems/experimental_ops/abs_.py |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/absolute.py |
54 |
54 |
0 |
0% |
| src/flag_gems/experimental_ops/absolute_.py |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/addcdiv.py |
61 |
61 |
0 |
0% |
| src/flag_gems/experimental_ops/addcmul_.py |
60 |
60 |
0 |
0% |
| src/flag_gems/experimental_ops/alias_copy.py |
46 |
46 |
0 |
0% |
| src/flag_gems/experimental_ops/amin.py |
135 |
135 |
0 |
0% |
| src/flag_gems/experimental_ops/arccosh.py |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/arcsinh.py |
49 |
49 |
0 |
0% |
| src/flag_gems/experimental_ops/arcsinh_.py |
30 |
30 |
0 |
0% |
| src/flag_gems/experimental_ops/arctanh.py |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/arctanh_.py |
43 |
43 |
0 |
0% |
| src/flag_gems/experimental_ops/asinh_.py |
45 |
45 |
0 |
0% |
| src/flag_gems/experimental_ops/atanh_.py |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/celu.py |
67 |
67 |
0 |
0% |
| src/flag_gems/experimental_ops/celu_.py |
24 |
24 |
0 |
0% |
| src/flag_gems/experimental_ops/copy_.py |
74 |
74 |
0 |
0% |
| src/flag_gems/experimental_ops/cos_.py |
28 |
28 |
0 |
0% |
| src/flag_gems/experimental_ops/cosh_.py |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/deg2rad.py |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/deg2rad_.py |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/diag.py |
112 |
112 |
0 |
0% |
| src/flag_gems/experimental_ops/digamma_.py |
54 |
54 |
0 |
0% |
| src/flag_gems/experimental_ops/elu.py |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/erf_.py |
43 |
43 |
0 |
0% |
| src/flag_gems/experimental_ops/erfinv.py |
78 |
78 |
0 |
0% |
| src/flag_gems/experimental_ops/erfinv_.py |
46 |
46 |
0 |
0% |
| src/flag_gems/experimental_ops/exp2.py |
42 |
42 |
0 |
0% |
| src/flag_gems/experimental_ops/exp2_.py |
30 |
30 |
0 |
0% |
| src/flag_gems/experimental_ops/exp_.py |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/expand.py |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/eye.py |
53 |
53 |
0 |
0% |
| src/flag_gems/experimental_ops/fft_ifftshift.py |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/fix.py |
46 |
46 |
0 |
0% |
| src/flag_gems/experimental_ops/fix_.py |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/floor_.py |
44 |
44 |
0 |
0% |
| src/flag_gems/experimental_ops/fmin.py |
82 |
82 |
0 |
0% |
| src/flag_gems/experimental_ops/frac.py |
55 |
55 |
0 |
0% |
| src/flag_gems/experimental_ops/gelu_.py |
74 |
74 |
0 |
0% |
| src/flag_gems/experimental_ops/glu.py |
78 |
78 |
0 |
0% |
| src/flag_gems/experimental_ops/hardshrink.py |
44 |
44 |
0 |
0% |
| src/flag_gems/experimental_ops/hardsigmoid.py |
29 |
29 |
0 |
0% |
| src/flag_gems/experimental_ops/hardsigmoid_.py |
42 |
42 |
0 |
0% |
| src/flag_gems/experimental_ops/hardswish.py |
74 |
74 |
0 |
0% |
| src/flag_gems/experimental_ops/hardswish_.py |
42 |
42 |
0 |
0% |
| src/flag_gems/experimental_ops/hardtanh.py |
49 |
49 |
0 |
0% |
| src/flag_gems/experimental_ops/hardtanh_.py |
51 |
51 |
0 |
0% |
| src/flag_gems/experimental_ops/heaviside.py |
53 |
53 |
0 |
0% |
| src/flag_gems/experimental_ops/heaviside_.py |
35 |
35 |
0 |
0% |
| src/flag_gems/experimental_ops/huber_loss.py |
102 |
102 |
0 |
0% |
| src/flag_gems/experimental_ops/hypot.py |
76 |
76 |
0 |
0% |
| src/flag_gems/experimental_ops/hypot_.py |
56 |
56 |
0 |
0% |
| src/flag_gems/experimental_ops/i0.py |
57 |
57 |
0 |
0% |
| src/flag_gems/experimental_ops/i0_.py |
44 |
44 |
0 |
0% |
| src/flag_gems/experimental_ops/im2col.py |
114 |
114 |
0 |
0% |
| src/flag_gems/experimental_ops/leaky_relu.py |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/leaky_relu_.py |
44 |
44 |
0 |
0% |
| src/flag_gems/experimental_ops/lift.py |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/lift_fresh_copy.py |
51 |
51 |
0 |
0% |
| src/flag_gems/experimental_ops/log10_.py |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/log1p_.py |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/log2_.py |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/log_.py |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/logaddexp.py |
70 |
70 |
0 |
0% |
| src/flag_gems/experimental_ops/logaddexp2.py |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/logical_xor_.py |
41 |
41 |
0 |
0% |
| src/flag_gems/experimental_ops/logit.py |
81 |
81 |
0 |
0% |
| src/flag_gems/experimental_ops/logit_.py |
61 |
61 |
0 |
0% |
| src/flag_gems/experimental_ops/margin_ranking_loss.py |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/masked_scatter.py |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/masked_select.py |
85 |
85 |
0 |
0% |
| src/flag_gems/experimental_ops/maximum.py |
103 |
103 |
0 |
0% |
| src/flag_gems/experimental_ops/mse_loss.py |
134 |
134 |
0 |
0% |
| src/flag_gems/experimental_ops/multiply.py |
91 |
91 |
0 |
0% |
| src/flag_gems/experimental_ops/mv.py |
58 |
58 |
0 |
0% |
| src/flag_gems/experimental_ops/native_dropout_backward.py |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/neg_.py |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/negative.py |
31 |
31 |
0 |
0% |
| src/flag_gems/experimental_ops/negative_.py |
25 |
25 |
0 |
0% |
| src/flag_gems/experimental_ops/permute.py |
64 |
64 |
0 |
0% |
| src/flag_gems/experimental_ops/permute_copy.py |
74 |
74 |
0 |
0% |
| src/flag_gems/experimental_ops/pixel_shuffle.py |
86 |
86 |
0 |
0% |
| src/flag_gems/experimental_ops/pixel_unshuffle.py |
82 |
82 |
0 |
0% |
| src/flag_gems/experimental_ops/prelu.py |
59 |
59 |
0 |
0% |
| src/flag_gems/experimental_ops/rad2deg_.py |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/reciprocal.py |
55 |
55 |
0 |
0% |
| src/flag_gems/experimental_ops/reciprocal_.py |
24 |
24 |
0 |
0% |
| src/flag_gems/experimental_ops/reflection_pad1d.py |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/relu.py |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/relu6.py |
26 |
26 |
0 |
0% |
| src/flag_gems/experimental_ops/relu_.py |
28 |
28 |
0 |
0% |
| src/flag_gems/experimental_ops/replication_pad1d.py |
76 |
76 |
0 |
0% |
| src/flag_gems/experimental_ops/replication_pad2d.py |
78 |
78 |
0 |
0% |
| src/flag_gems/experimental_ops/replication_pad3d.py |
94 |
94 |
0 |
0% |
| src/flag_gems/experimental_ops/rmsnorm.py |
47 |
40 |
0 |
15% |
| src/flag_gems/experimental_ops/rrelu_with_noise_backward.py |
41 |
41 |
0 |
0% |
| src/flag_gems/experimental_ops/rsqrt_.py |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/selu.py |
42 |
42 |
0 |
0% |
| src/flag_gems/experimental_ops/selu_.py |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/sgn.py |
60 |
60 |
0 |
0% |
| src/flag_gems/experimental_ops/sgn_.py |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/sigmoid.py |
45 |
45 |
0 |
0% |
| src/flag_gems/experimental_ops/sigmoid_.py |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/sign.py |
37 |
37 |
0 |
0% |
| src/flag_gems/experimental_ops/silu.py |
64 |
64 |
0 |
0% |
| src/flag_gems/experimental_ops/silu_.py |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/sin_.py |
36 |
36 |
0 |
0% |
| src/flag_gems/experimental_ops/sinc.py |
31 |
31 |
0 |
0% |
| src/flag_gems/experimental_ops/sinc_.py |
32 |
32 |
0 |
0% |
| src/flag_gems/experimental_ops/sinh_.py |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/slice_backward.py |
76 |
76 |
0 |
0% |
| src/flag_gems/experimental_ops/slice_scatter.py |
87 |
87 |
0 |
0% |
| src/flag_gems/experimental_ops/smooth_l1_loss.py |
149 |
149 |
0 |
0% |
| src/flag_gems/experimental_ops/soft_margin_loss.py |
121 |
121 |
0 |
0% |
| src/flag_gems/experimental_ops/softplus.py |
69 |
69 |
0 |
0% |
| src/flag_gems/experimental_ops/softshrink.py |
54 |
54 |
0 |
0% |
| src/flag_gems/experimental_ops/special_i0e.py |
45 |
45 |
0 |
0% |
| src/flag_gems/experimental_ops/special_i1.py |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/special_xlog1py.py |
90 |
90 |
0 |
0% |
| src/flag_gems/experimental_ops/t_copy.py |
64 |
64 |
0 |
0% |
| src/flag_gems/experimental_ops/take.py |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/threshold.py |
55 |
55 |
0 |
0% |
| src/flag_gems/experimental_ops/threshold_.py |
34 |
34 |
0 |
0% |
| src/flag_gems/experimental_ops/trace.py |
61 |
61 |
0 |
0% |
| src/flag_gems/experimental_ops/tril.py |
43 |
43 |
0 |
0% |
| src/flag_gems/experimental_ops/triu.py |
73 |
73 |
0 |
0% |
| src/flag_gems/experimental_ops/trunc.py |
85 |
85 |
0 |
0% |
| src/flag_gems/experimental_ops/unsqueeze.py |
45 |
45 |
0 |
0% |
| src/flag_gems/experimental_ops/unsqueeze_copy.py |
60 |
60 |
0 |
0% |
| src/flag_gems/experimental_ops/upsample_nearest1d.py |
70 |
70 |
0 |
0% |
| src/flag_gems/experimental_ops/upsample_nearest3d.py |
121 |
121 |
0 |
0% |
| src/flag_gems/experimental_ops/xlogy.py |
75 |
75 |
0 |
0% |
| src/flag_gems/experimental_ops/xlogy_.py |
77 |
77 |
0 |
0% |
| src/flag_gems/experimental_ops/zero.py |
43 |
43 |
0 |
0% |
| src/flag_gems/experimental_ops/zeros_like.py |
79 |
79 |
0 |
0% |
| src/flag_gems/fused/FLA/__init__.py |
3 |
0 |
0 |
100% |
| src/flag_gems/fused/FLA/chunk.py |
19 |
10 |
0 |
47% |
| src/flag_gems/fused/FLA/chunk_delta_h.py |
163 |
149 |
0 |
9% |
| src/flag_gems/fused/FLA/chunk_o.py |
69 |
54 |
0 |
22% |
| src/flag_gems/fused/FLA/chunk_scaled_dot_kkt.py |
46 |
34 |
0 |
26% |
| src/flag_gems/fused/FLA/cumsum.py |
90 |
69 |
0 |
23% |
| src/flag_gems/fused/FLA/fused_recurrent.py |
173 |
160 |
0 |
8% |
| src/flag_gems/fused/FLA/index.py |
13 |
4 |
0 |
69% |
| src/flag_gems/fused/FLA/solve_tril.py |
226 |
198 |
0 |
12% |
| src/flag_gems/fused/FLA/triton_ops_helper.py |
15 |
6 |
0 |
60% |
| src/flag_gems/fused/FLA/utils.py |
68 |
13 |
0 |
81% |
| src/flag_gems/fused/FLA/wy_fast.py |
49 |
38 |
0 |
22% |
| src/flag_gems/fused/__init__.py |
26 |
0 |
0 |
100% |
| src/flag_gems/fused/apply_repetition_penalties.py |
34 |
16 |
0 |
53% |
| src/flag_gems/fused/concat_and_cache_mla.py |
80 |
39 |
0 |
51% |
| src/flag_gems/fused/cross_entropy_loss.py |
376 |
286 |
0 |
24% |
| src/flag_gems/fused/cutlass_scaled_mm.py |
191 |
157 |
0 |
18% |
| src/flag_gems/fused/flash_mla.py |
107 |
72 |
0 |
33% |
| src/flag_gems/fused/fused_add_rms_norm.py |
40 |
17 |
0 |
58% |
| src/flag_gems/fused/geglu.py |
68 |
34 |
0 |
50% |
| src/flag_gems/fused/gelu_and_mul.py |
70 |
35 |
0 |
50% |
| src/flag_gems/fused/grouped_topk.py |
130 |
122 |
0 |
6% |
| src/flag_gems/fused/instance_norm.py |
308 |
213 |
0 |
31% |
| src/flag_gems/fused/moe_align_block_size.py |
131 |
86 |
0 |
34% |
| src/flag_gems/fused/moe_sum.py |
27 |
15 |
0 |
44% |
| src/flag_gems/fused/outer.py |
26 |
6 |
0 |
77% |
| src/flag_gems/fused/reglu.py |
78 |
37 |
0 |
53% |
| src/flag_gems/fused/reshape_and_cache.py |
41 |
20 |
0 |
51% |
| src/flag_gems/fused/reshape_and_cache_flash.py |
44 |
19 |
0 |
57% |
| src/flag_gems/fused/rotary_embedding.py |
128 |
84 |
0 |
34% |
| src/flag_gems/fused/rwkv_ka_fusion.py |
37 |
21 |
0 |
43% |
| src/flag_gems/fused/rwkv_mm_sparsity.py |
30 |
14 |
0 |
53% |
| src/flag_gems/fused/silu_and_mul.py |
37 |
15 |
0 |
59% |
| src/flag_gems/fused/skip_layernorm.py |
50 |
20 |
0 |
60% |
| src/flag_gems/fused/swiglu.py |
72 |
35 |
0 |
51% |
| src/flag_gems/fused/topk_softmax.py |
51 |
45 |
0 |
12% |
| src/flag_gems/fused/weight_norm.py |
124 |
97 |
0 |
22% |
| src/flag_gems/logging_utils.py |
38 |
8 |
0 |
79% |
| src/flag_gems/modules/__init__.py |
5 |
0 |
0 |
100% |
| src/flag_gems/modules/activation.py |
15 |
4 |
0 |
73% |
| src/flag_gems/modules/normalization.py |
49 |
28 |
0 |
43% |
| src/flag_gems/modules/rotary_embedding.py |
79 |
56 |
0 |
29% |
| src/flag_gems/ops/__init__.py |
171 |
0 |
0 |
100% |
| src/flag_gems/ops/abs.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/acos.py |
14 |
1 |
0 |
93% |
| src/flag_gems/ops/add.py |
37 |
7 |
0 |
81% |
| src/flag_gems/ops/addcdiv.py |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/addcmul.py |
18 |
6 |
0 |
67% |
| src/flag_gems/ops/addmm.py |
64 |
24 |
0 |
62% |
| src/flag_gems/ops/addmv.py |
57 |
21 |
0 |
63% |
| src/flag_gems/ops/addr.py |
44 |
19 |
0 |
57% |
| src/flag_gems/ops/all.py |
103 |
33 |
0 |
68% |
| src/flag_gems/ops/amax.py |
92 |
50 |
0 |
46% |
| src/flag_gems/ops/angle.py |
30 |
8 |
0 |
73% |
| src/flag_gems/ops/any.py |
103 |
33 |
0 |
68% |
| src/flag_gems/ops/arange.py |
38 |
9 |
0 |
76% |
| src/flag_gems/ops/argmax.py |
163 |
92 |
0 |
44% |
| src/flag_gems/ops/argmin.py |
161 |
89 |
0 |
45% |
| src/flag_gems/ops/atan.py |
18 |
1 |
0 |
94% |
| src/flag_gems/ops/attention.py |
399 |
284 |
0 |
29% |
| src/flag_gems/ops/avg_pool2d.py |
150 |
86 |
0 |
43% |
| src/flag_gems/ops/baddbmm.py |
146 |
103 |
0 |
29% |
| src/flag_gems/ops/batch_norm.py |
154 |
100 |
0 |
35% |
| src/flag_gems/ops/bitwise_and.py |
27 |
2 |
0 |
93% |
| src/flag_gems/ops/bitwise_left_shift.py |
11 |
3 |
0 |
73% |
| src/flag_gems/ops/bitwise_not.py |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/bitwise_or.py |
27 |
2 |
0 |
93% |
| src/flag_gems/ops/bitwise_right_shift.py |
10 |
2 |
0 |
80% |
| src/flag_gems/ops/bmm.py |
93 |
58 |
0 |
38% |
| src/flag_gems/ops/cat.py |
109 |
35 |
0 |
68% |
| src/flag_gems/ops/ceil.py |
22 |
2 |
0 |
91% |
| src/flag_gems/ops/celu.py |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/clamp.py |
75 |
18 |
0 |
76% |
| src/flag_gems/ops/contiguous.py |
11 |
6 |
0 |
45% |
| src/flag_gems/ops/conv1d.py |
27 |
3 |
0 |
89% |
| src/flag_gems/ops/conv2d.py |
178 |
103 |
0 |
42% |
| src/flag_gems/ops/conv3d.py |
104 |
45 |
0 |
57% |
| src/flag_gems/ops/conv_depthwise2d.py |
10 |
6 |
0 |
40% |
| src/flag_gems/ops/copy.py |
60 |
20 |
0 |
67% |
| src/flag_gems/ops/cos.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/count_nonzero.py |
95 |
36 |
0 |
62% |
| src/flag_gems/ops/cummax.py |
242 |
153 |
0 |
37% |
| src/flag_gems/ops/cummin.py |
242 |
153 |
0 |
37% |
| src/flag_gems/ops/cumsum.py |
328 |
198 |
0 |
40% |
| src/flag_gems/ops/diag.py |
62 |
20 |
0 |
68% |
| src/flag_gems/ops/diag_embed.py |
29 |
1 |
0 |
97% |
| src/flag_gems/ops/diagonal.py |
15 |
6 |
0 |
60% |
| src/flag_gems/ops/div.py |
172 |
70 |
0 |
59% |
| src/flag_gems/ops/dot.py |
63 |
32 |
0 |
49% |
| src/flag_gems/ops/dropout.py |
90 |
50 |
0 |
44% |
| src/flag_gems/ops/elu.py |
31 |
6 |
0 |
81% |
| src/flag_gems/ops/embedding.py |
98 |
45 |
0 |
54% |
| src/flag_gems/ops/eq.py |
33 |
6 |
0 |
82% |
| src/flag_gems/ops/erf.py |
16 |
2 |
0 |
88% |
| src/flag_gems/ops/exp.py |
18 |
1 |
0 |
94% |
| src/flag_gems/ops/exp2.py |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/exponential_.py |
120 |
71 |
0 |
41% |
| src/flag_gems/ops/eye.py |
21 |
2 |
0 |
90% |
| src/flag_gems/ops/eye_m.py |
35 |
12 |
0 |
66% |
| src/flag_gems/ops/fill.py |
43 |
6 |
0 |
86% |
| src/flag_gems/ops/flash_api.py |
373 |
33 |
0 |
91% |
| src/flag_gems/ops/flash_kernel.py |
555 |
478 |
0 |
14% |
| src/flag_gems/ops/flip.py |
32 |
1 |
0 |
97% |
| src/flag_gems/ops/full.py |
41 |
4 |
0 |
90% |
| src/flag_gems/ops/full_like.py |
16 |
0 |
0 |
100% |
| src/flag_gems/ops/gather.py |
130 |
3 |
0 |
98% |
| src/flag_gems/ops/ge.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/gelu.py |
56 |
17 |
0 |
70% |
| src/flag_gems/ops/get_scheduler_metadata.py |
332 |
308 |
0 |
7% |
| src/flag_gems/ops/glu.py |
34 |
7 |
0 |
79% |
| src/flag_gems/ops/groupnorm.py |
139 |
88 |
0 |
37% |
| src/flag_gems/ops/gt.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/hstack.py |
106 |
34 |
0 |
68% |
| src/flag_gems/ops/index.py |
281 |
23 |
0 |
92% |
| src/flag_gems/ops/index_add.py |
157 |
0 |
0 |
100% |
| src/flag_gems/ops/index_put.py |
232 |
8 |
0 |
97% |
| src/flag_gems/ops/index_select.py |
50 |
16 |
0 |
68% |
| src/flag_gems/ops/isclose.py |
45 |
20 |
0 |
56% |
| src/flag_gems/ops/isfinite.py |
20 |
4 |
0 |
80% |
| src/flag_gems/ops/isin.py |
130 |
53 |
0 |
59% |
| src/flag_gems/ops/isinf.py |
13 |
1 |
0 |
92% |
| src/flag_gems/ops/isnan.py |
13 |
1 |
0 |
92% |
| src/flag_gems/ops/kron.py |
171 |
65 |
0 |
62% |
| src/flag_gems/ops/layernorm.py |
241 |
163 |
0 |
32% |
| src/flag_gems/ops/le.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/lerp.py |
35 |
5 |
0 |
86% |
| src/flag_gems/ops/linspace.py |
34 |
10 |
0 |
71% |
| src/flag_gems/ops/log.py |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/log_sigmoid.py |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/log_softmax.py |
98 |
50 |
0 |
49% |
| src/flag_gems/ops/logical_and.py |
20 |
2 |
0 |
90% |
| src/flag_gems/ops/logical_not.py |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/logical_or.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/logical_xor.py |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/logspace.py |
37 |
9 |
0 |
76% |
| src/flag_gems/ops/lt.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/masked_fill.py |
33 |
2 |
0 |
94% |
| src/flag_gems/ops/masked_scatter.py |
122 |
61 |
0 |
50% |
| src/flag_gems/ops/masked_select.py |
105 |
54 |
0 |
49% |
| src/flag_gems/ops/max.py |
97 |
40 |
0 |
59% |
| src/flag_gems/ops/max_pool2d_with_indices.py |
141 |
73 |
0 |
48% |
| src/flag_gems/ops/maximum.py |
18 |
4 |
0 |
78% |
| src/flag_gems/ops/mean.py |
193 |
103 |
0 |
47% |
| src/flag_gems/ops/min.py |
96 |
40 |
0 |
58% |
| src/flag_gems/ops/minimum.py |
18 |
4 |
0 |
78% |
| src/flag_gems/ops/mm.py |
104 |
59 |
0 |
43% |
| src/flag_gems/ops/mm_streamk.py |
234 |
182 |
0 |
22% |
| src/flag_gems/ops/mse_loss.py |
61 |
21 |
0 |
66% |
| src/flag_gems/ops/mul.py |
27 |
3 |
0 |
89% |
| src/flag_gems/ops/multinomial.py |
52 |
16 |
0 |
69% |
| src/flag_gems/ops/mv.py |
39 |
17 |
0 |
56% |
| src/flag_gems/ops/nan_to_num.py |
26 |
7 |
0 |
73% |
| src/flag_gems/ops/ne.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/neg.py |
14 |
1 |
0 |
93% |
| src/flag_gems/ops/nllloss.py |
196 |
119 |
0 |
39% |
| src/flag_gems/ops/nonzero.py |
46 |
13 |
0 |
72% |
| src/flag_gems/ops/normal.py |
61 |
4 |
0 |
93% |
| src/flag_gems/ops/one_hot.py |
32 |
6 |
0 |
81% |
| src/flag_gems/ops/ones.py |
31 |
6 |
0 |
81% |
| src/flag_gems/ops/ones_like.py |
18 |
0 |
0 |
100% |
| src/flag_gems/ops/pad.py |
274 |
1 |
0 |
99% |
| src/flag_gems/ops/per_token_group_quant_fp8.py |
69 |
38 |
0 |
45% |
| src/flag_gems/ops/polar.py |
17 |
3 |
0 |
82% |
| src/flag_gems/ops/pow.py |
33 |
3 |
0 |
91% |
| src/flag_gems/ops/prod.py |
86 |
31 |
0 |
64% |
| src/flag_gems/ops/quantile.py |
153 |
85 |
0 |
44% |
| src/flag_gems/ops/rand.py |
48 |
22 |
0 |
54% |
| src/flag_gems/ops/rand_like.py |
22 |
0 |
0 |
100% |
| src/flag_gems/ops/randn.py |
77 |
47 |
0 |
39% |
| src/flag_gems/ops/randn_like.py |
22 |
0 |
0 |
100% |
| src/flag_gems/ops/randperm.py |
266 |
143 |
0 |
46% |
| src/flag_gems/ops/reciprocal.py |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/relu.py |
21 |
2 |
0 |
90% |
| src/flag_gems/ops/repeat.py |
251 |
2 |
0 |
99% |
| src/flag_gems/ops/repeat_interleave.py |
83 |
21 |
0 |
75% |
| src/flag_gems/ops/resolve_conj.py |
102 |
86 |
0 |
16% |
| src/flag_gems/ops/resolve_neg.py |
7 |
0 |
0 |
100% |
| src/flag_gems/ops/rms_norm.py |
119 |
75 |
0 |
37% |
| src/flag_gems/ops/rsqrt.py |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/scaled_softmax.py |
122 |
74 |
0 |
39% |
| src/flag_gems/ops/scatter.py |
251 |
1 |
0 |
99% |
| src/flag_gems/ops/scatter_add_.py |
255 |
46 |
0 |
82% |
| src/flag_gems/ops/select_scatter.py |
45 |
14 |
0 |
69% |
| src/flag_gems/ops/sigmoid.py |
29 |
5 |
0 |
83% |
| src/flag_gems/ops/silu.py |
32 |
8 |
0 |
75% |
| src/flag_gems/ops/sin.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/slice_scatter.py |
51 |
16 |
0 |
69% |
| src/flag_gems/ops/softmax.py |
222 |
153 |
0 |
31% |
| src/flag_gems/ops/softplus.py |
17 |
5 |
0 |
71% |
| src/flag_gems/ops/sort.py |
210 |
123 |
0 |
41% |
| src/flag_gems/ops/sqrt.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/stack.py |
86 |
34 |
0 |
60% |
| src/flag_gems/ops/std.py |
110 |
60 |
0 |
45% |
| src/flag_gems/ops/sub.py |
31 |
4 |
0 |
87% |
| src/flag_gems/ops/sum.py |
206 |
124 |
0 |
40% |
| src/flag_gems/ops/tan.py |
17 |
2 |
0 |
88% |
| src/flag_gems/ops/tanh.py |
28 |
3 |
0 |
89% |
| src/flag_gems/ops/threshold.py |
21 |
2 |
0 |
90% |
| src/flag_gems/ops/tile.py |
255 |
2 |
0 |
99% |
| src/flag_gems/ops/to.py |
47 |
6 |
0 |
87% |
| src/flag_gems/ops/topk.py |
183 |
109 |
0 |
40% |
| src/flag_gems/ops/trace.py |
55 |
23 |
0 |
58% |
| src/flag_gems/ops/triu.py |
106 |
36 |
0 |
66% |
| src/flag_gems/ops/uniform.py |
41 |
20 |
0 |
51% |
| src/flag_gems/ops/unique.py |
284 |
146 |
0 |
49% |
| src/flag_gems/ops/upsample_bicubic2d_aa.py |
178 |
141 |
0 |
21% |
| src/flag_gems/ops/upsample_nearest1d.py |
51 |
23 |
0 |
55% |
| src/flag_gems/ops/upsample_nearest2d.py |
58 |
28 |
0 |
52% |
| src/flag_gems/ops/var_mean.py |
115 |
59 |
0 |
49% |
| src/flag_gems/ops/vdot.py |
140 |
66 |
0 |
53% |
| src/flag_gems/ops/vector_norm.py |
259 |
146 |
0 |
44% |
| src/flag_gems/ops/vstack.py |
76 |
20 |
0 |
74% |
| src/flag_gems/ops/weightnorm.py |
160 |
114 |
0 |
29% |
| src/flag_gems/ops/where.py |
48 |
1 |
0 |
98% |
| src/flag_gems/ops/zeros.py |
35 |
6 |
0 |
83% |
| src/flag_gems/ops/zeros_like.py |
18 |
0 |
0 |
100% |
| src/flag_gems/patches/__init__.py |
3 |
0 |
0 |
100% |
| src/flag_gems/patches/patch_util.py |
59 |
21 |
0 |
64% |
| src/flag_gems/patches/patch_vllm_all.py |
185 |
160 |
0 |
14% |
| src/flag_gems/runtime/__init__.py |
30 |
10 |
0 |
67% |
| src/flag_gems/runtime/backend/__init__.py |
217 |
39 |
0 |
82% |
| src/flag_gems/runtime/backend/_aipu/__init__.py |
6 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_aipu/fused/__init__.py |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_aipu/heuristics_config_utils.py |
136 |
136 |
0 |
0% |
| src/flag_gems/runtime/backend/_aipu/ops/__init__.py |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_aipu/ops/cumsum.py |
258 |
258 |
0 |
0% |
| src/flag_gems/runtime/backend/_aipu/ops/multinomial.py |
52 |
52 |
0 |
0% |
| src/flag_gems/runtime/backend/_amd/__init__.py |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_amd/heuristics_config_utils.py |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_amd/ops/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_arm/__init__.py |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/heuristics_config_utils.py |
122 |
122 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/__init__.py |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/add.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/gelu.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/__init__.py |
14 |
3 |
0 |
79% |
| src/flag_gems/runtime/backend/_ascend/fused/__init__.py |
5 |
5 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/cross_entropy_loss.py |
380 |
380 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/fused_add_rms_norm.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/rotary_embedding.py |
86 |
86 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/skip_layernorm.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/heuristics_config_utils.py |
143 |
143 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/__init__.py |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/addmm.py |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/all.py |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/amax.py |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/angle.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/any.py |
106 |
106 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/arange.py |
38 |
38 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/argmax.py |
148 |
148 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/argmin.py |
101 |
101 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/bmm.py |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/cat.py |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/count_nonzero.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/cumsum.py |
321 |
321 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/diag.py |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/diag_embed.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/diagonal.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/dot.py |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/embedding.py |
110 |
110 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/exponential_.py |
86 |
86 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/fill.py |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/flip.py |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/full.py |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/full_like.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/gather.py |
81 |
81 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/groupnorm.py |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/hstack.py |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index.py |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index_add.py |
145 |
145 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index_select.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/isin.py |
132 |
132 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/linspace.py |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/log_softmax.py |
99 |
99 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/masked_fill.py |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/masked_select.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/max.py |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/mean.py |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/min.py |
101 |
101 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/mm.py |
76 |
76 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/multinomial.py |
52 |
52 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/ones.py |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/ones_like.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/outer.py |
74 |
74 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/polar.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/pow.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/randperm.py |
273 |
273 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/repeat_interleave.py |
39 |
39 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/resolve_neg.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/rms_norm.py |
125 |
125 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/select_scatter.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/slice_scatter.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/softmax.py |
213 |
213 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/sort.py |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/stack.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/threshold.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/triu.py |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/unique.py |
314 |
314 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/upsample_nearest2d.py |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/var_mean.py |
155 |
155 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/vector_norm.py |
267 |
267 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/vstack.py |
80 |
80 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/where.py |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/zeros.py |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/zeros_like.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/utils/__init__.py |
8 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_cambricon/__init__.py |
18 |
16 |
0 |
11% |
| src/flag_gems/runtime/backend/_cambricon/fused/__init__.py |
9 |
9 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/cross_entropy_loss.py |
507 |
507 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/flash_mla.py |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/fused_add_rms_norm.py |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/gelu_and_mul.py |
69 |
69 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/outer.py |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/silu_and_mul.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/skip_layernorm.py |
120 |
120 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/fused/weight_norm.py |
124 |
124 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/heuristics_config_utils.py |
110 |
110 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/__init__.py |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/abs.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/acos.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/add.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/addcdiv.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/addcmul.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/addmm.py |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/all.py |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/amax.py |
133 |
133 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/any.py |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/arange.py |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/argmax.py |
133 |
133 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/atan.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/attention.py |
397 |
397 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/avg_pool2d.py |
177 |
177 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_and.py |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_left_shift.py |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_not.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_or.py |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bitwise_right_shift.py |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/bmm.py |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cat.py |
213 |
213 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ceil.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/celu.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/clamp.py |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/contiguous.py |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/copy.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cos.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/count_nonzero.py |
113 |
113 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cummin.py |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/cumsum.py |
325 |
325 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/diag.py |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/diag_embed.py |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/diagonal.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/div.py |
182 |
182 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/dropout.py |
84 |
84 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/elu.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/embedding.py |
84 |
84 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/eq.py |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/erf.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/exp.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/exp2.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/exponential_.py |
69 |
69 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/fill.py |
69 |
69 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/flip.py |
275 |
275 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/full.py |
70 |
70 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/full_like.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/gather.py |
157 |
157 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ge.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/gelu.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/glu.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/groupnorm.py |
343 |
343 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/gt.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/hstack.py |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/index_add.py |
193 |
193 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/index_select.py |
167 |
167 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isclose.py |
40 |
40 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isfinite.py |
24 |
24 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isin.py |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isinf.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/isnan.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/kron.py |
121 |
121 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/layernorm.py |
341 |
341 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/le.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/linspace.py |
39 |
39 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/log.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/log_sigmoid.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/log_softmax.py |
556 |
556 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logical_and.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logical_not.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logical_or.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logical_xor.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/logspace.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/lt.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/masked_fill.py |
73 |
73 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/masked_select.py |
73 |
73 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/max.py |
177 |
177 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/max_pool2d_with_indices.py |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/maximum.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/mean.py |
82 |
82 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/min.py |
170 |
170 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/minimum.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/mm.py |
97 |
97 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/mul.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/multinomial.py |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/mv.py |
67 |
67 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ne.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/neg.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/nonzero.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/normal.py |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ones.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ones_like.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/pad.py |
344 |
344 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/per_token_group_quant_fp8.py |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/pow.py |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/prod.py |
97 |
97 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/quantile.py |
225 |
225 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rand.py |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rand_like.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/randn.py |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/randn_like.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/randperm.py |
329 |
329 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/reciprocal.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/relu.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/repeat.py |
303 |
303 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/repeat_interleave.py |
111 |
111 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/resolve_conj.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/resolve_neg.py |
7 |
7 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rms_norm.py |
199 |
199 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/rsqrt.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/scatter.py |
184 |
184 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/select_scatter.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sigmoid.py |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/silu.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sin.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/slice_scatter.py |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/softmax.py |
546 |
546 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/softplus.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sort.py |
207 |
207 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sqrt.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/stack.py |
104 |
104 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sub.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sum.py |
142 |
142 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/tan.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/tanh.py |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/tile.py |
306 |
306 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/to.py |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/topk.py |
174 |
174 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/triu.py |
140 |
140 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/uniform.py |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/unique.py |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/upsample_nearest2d.py |
108 |
108 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/var_mean.py |
160 |
160 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/vector_norm.py |
308 |
308 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/vstack.py |
184 |
184 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/weightnorm.py |
228 |
228 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/where.py |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/zeros.py |
40 |
40 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/zeros_like.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/__init__.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/pointwise_dynamic.py |
1003 |
1003 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/reduce_utils.py |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/__init__.py |
15 |
9 |
0 |
40% |
| src/flag_gems/runtime/backend/_enflame/fused/__init__.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/heuristics_config_utils.py |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/ops/__init__.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/__init__.py |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_hygon/fused/__init__.py |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/heuristics_config_utils.py |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/__init__.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/all.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/any.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/attention.py |
399 |
399 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/div.py |
178 |
178 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/exponential_.py |
93 |
93 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/fill.py |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/flash_api.py |
373 |
373 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/flash_kernel.py |
551 |
551 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/gelu.py |
84 |
84 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/isclose.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/isin.py |
132 |
132 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/mm.py |
94 |
94 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/pow.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/randperm.py |
266 |
266 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/silu.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/sort.py |
182 |
182 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/unique.py |
287 |
287 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/upsample_nearest2d.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/__init__.py |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_iluvatar/heuristics_config_utils.py |
131 |
131 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/__init__.py |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/div.py |
166 |
166 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/__init__.py |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/__init__.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/concat_and_cache_mla.py |
79 |
79 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/cross_entropy_loss.py |
431 |
431 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/flash_mla.py |
99 |
99 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/fused_add_rms_norm.py |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/geglu.py |
68 |
68 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/gelu_and_mul.py |
70 |
70 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/instance_norm.py |
345 |
345 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/moe_align_block_size.py |
76 |
76 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/outer.py |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/reglu.py |
83 |
83 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/reshape_and_cache.py |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/reshape_and_cache_flash.py |
39 |
39 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/rotary_embedding.py |
128 |
128 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/rwkv_ka_fusion.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/rwkv_mm_sparsity.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/silu_and_mul.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/skip_layernorm.py |
93 |
93 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/topk_softmax.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/weight_norm.py |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/heuristics_config_utils.py |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/__init__.py |
170 |
170 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/abs.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/acos.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/add.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addcdiv.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addcmul.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addmm.py |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addmv.py |
61 |
61 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addr.py |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/all.py |
160 |
160 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/amax.py |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/angle.py |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/any.py |
160 |
160 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/arange.py |
40 |
40 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/argmax.py |
137 |
137 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/argmin.py |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/atan.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/attention.py |
382 |
382 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/avg_pool2d.py |
150 |
150 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/baddbmm.py |
148 |
148 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/batch_norm.py |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_and.py |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_left_shift.py |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_not.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_or.py |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bitwise_right_shift.py |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/bmm.py |
102 |
102 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cat.py |
53 |
53 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/celu.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/clamp.py |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/contiguous.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv1d.py |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv2d.py |
216 |
216 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv3d.py |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/conv_depthwise2d.py |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/copy.py |
64 |
64 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cos.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/count_nonzero.py |
140 |
140 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cummax.py |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cummin.py |
243 |
243 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cumsum.py |
269 |
269 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/diag.py |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/diag_embed.py |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/diagonal.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/div.py |
179 |
179 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/dot.py |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/dropout.py |
97 |
97 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/elu.py |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/embedding.py |
98 |
98 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/eq.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/erf.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/exp.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/exp2.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/exponential_.py |
88 |
88 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/eye.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/eye_m.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/fill.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/flash_api.py |
364 |
364 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/flash_kernel.py |
534 |
534 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/flip.py |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/full.py |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/full_like.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/gather.py |
189 |
189 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ge.py |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/gelu.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/get_scheduler_metadata.py |
277 |
277 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/glu.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/groupnorm.py |
187 |
187 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/gt.py |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/hstack.py |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index.py |
293 |
293 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_add.py |
80 |
80 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_put.py |
252 |
252 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_select.py |
46 |
46 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isclose.py |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isfinite.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isin.py |
138 |
138 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isinf.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isnan.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/kron.py |
117 |
117 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/layernorm.py |
295 |
295 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/le.py |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/lerp.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/linspace.py |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/log.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/log_sigmoid.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/log_softmax.py |
105 |
105 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logical_and.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logical_not.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logical_or.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logical_xor.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/logspace.py |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/lt.py |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/masked_fill.py |
84 |
84 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/masked_scatter.py |
122 |
122 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/masked_select.py |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/max.py |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/max_pool2d_with_indices.py |
145 |
145 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/maximum.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mean.py |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/min.py |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/minimum.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mm.py |
108 |
108 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mse_loss.py |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mul.py |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/multinomial.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/mv.py |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/nan_to_num.py |
27 |
27 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ne.py |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/neg.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/nllloss.py |
171 |
171 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/nonzero.py |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/normal.py |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ones.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ones_like.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/pad.py |
285 |
285 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/per_token_group_quant_fp8.py |
67 |
67 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/polar.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/pow.py |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/prod.py |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/quantile.py |
83 |
83 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rand.py |
131 |
131 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rand_like.py |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/randn.py |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/randn_like.py |
24 |
24 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/randperm.py |
269 |
269 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/reciprocal.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/relu.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/repeat.py |
251 |
251 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/repeat_interleave.py |
83 |
83 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/resolve_conj.py |
6 |
6 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/resolve_neg.py |
7 |
7 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rms_norm.py |
213 |
213 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rsqrt.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/rsub.py |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/scaled_softmax.py |
122 |
122 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/scatter.py |
239 |
239 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/scatter_add_.py |
255 |
255 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/select_scatter.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sigmoid.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/silu.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sin.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/slice_scatter.py |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/softmax.py |
180 |
180 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/softplus.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sort.py |
307 |
307 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sqrt.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/stack.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/std.py |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sub.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sum.py |
148 |
148 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/tan.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/tanh.py |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/threshold.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/tile.py |
255 |
255 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/to.py |
67 |
67 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/topk.py |
187 |
187 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/trace.py |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/triu.py |
68 |
68 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/uniform.py |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/unique.py |
524 |
524 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/upsample_bicubic2d_aa.py |
200 |
200 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/upsample_nearest1d.py |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/upsample_nearest2d.py |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/var_mean.py |
120 |
120 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/vdot.py |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/vector_norm.py |
263 |
263 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/vstack.py |
77 |
77 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/weightnorm.py |
171 |
171 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/where.py |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/zeros.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/zeros_like.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/__init__.py |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_metax/fused/__init__.py |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/fused/flash_mla.py |
104 |
104 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/heuristics_config_utils.py |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/__init__.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/addmm.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/amax.py |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/arange.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/bmm.py |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/exponential_.py |
176 |
176 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/full.py |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/full_like.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/groupnorm.py |
147 |
147 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/index.py |
281 |
281 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/index_put.py |
233 |
233 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/index_select.py |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/isin.py |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/log_softmax.py |
98 |
98 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/masked_fill.py |
65 |
65 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/min.py |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/mm.py |
116 |
116 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/nonzero.py |
126 |
126 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/ones.py |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/ones_like.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/outer.py |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/polar.py |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/prod.py |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/repeat_interleave.py |
117 |
117 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/resolve_conj.py |
7 |
7 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/sigmoid.py |
61 |
61 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/tanh.py |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/unique.py |
284 |
284 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/upsample_nearest2d.py |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/zeros.py |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/zeros_like.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/__init__.py |
4 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_mthreads/fused/__init__.py |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/fused/cross_entropy_loss.py |
376 |
376 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/heuristics_config_utils.py |
231 |
231 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/__init__.py |
38 |
38 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/addmm.py |
120 |
120 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/all.py |
140 |
140 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/any.py |
135 |
135 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/arange.py |
82 |
82 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/argmin.py |
121 |
121 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/batch_norm.py |
283 |
283 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/bmm.py |
137 |
137 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/celu.py |
82 |
82 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/conv2d.py |
161 |
161 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/dropout.py |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/gather.py |
71 |
71 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/gelu.py |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/index_put.py |
244 |
244 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/isin.py |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/log.py |
57 |
57 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/max.py |
150 |
150 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/min.py |
123 |
123 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/mm.py |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/ones.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/ones_like.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/prod.py |
188 |
188 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/rand.py |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/rand_like.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/randn.py |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/randn_like.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/randperm.py |
266 |
266 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/resolve_conj.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/sort.py |
210 |
210 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/tanh.py |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/unique.py |
284 |
284 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/utils.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/zeros.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/zeros_like.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/__init__.py |
5 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/ampere/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/ampere/ops/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/fused/__init__.py |
2 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/fused/fused_add_rms_norm.py |
37 |
25 |
0 |
32% |
| src/flag_gems/runtime/backend/_nvidia/heuristics_config_utils.py |
235 |
41 |
0 |
83% |
| src/flag_gems/runtime/backend/_nvidia/hopper/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/hopper/grouped_topk.py |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/heuristics_config_utils.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/__init__.py |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/mm.py |
216 |
216 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/ops/__init__.py |
3 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/ops/add.py |
20 |
14 |
0 |
30% |
| src/flag_gems/runtime/backend/_nvidia/ops/gelu.py |
60 |
32 |
0 |
47% |
| src/flag_gems/runtime/backend/_sunrise/__init__.py |
5 |
4 |
0 |
20% |
| src/flag_gems/runtime/backend/_sunrise/fused/__init__.py |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/heuristics_config_utils.py |
174 |
174 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/__init__.py |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/__init__.py |
28 |
23 |
0 |
18% |
| src/flag_gems/runtime/backend/_tsingmicro/fused/__init__.py |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/fused/cross_entropy_loss.py |
506 |
506 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/heuristics_config_utils.py |
215 |
215 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/__init__.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/argmax.py |
132 |
132 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/cat.py |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/count_nonzero.py |
113 |
113 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/hstack.py |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/isin.py |
126 |
126 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/kron.py |
120 |
120 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/masked_select.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/mm.py |
94 |
94 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/rms_norm.py |
175 |
175 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/stack.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/unique.py |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/backend_utils.py |
30 |
5 |
0 |
83% |
| src/flag_gems/runtime/backend/device.py |
83 |
8 |
0 |
90% |
| src/flag_gems/runtime/common.py |
21 |
0 |
0 |
100% |
| src/flag_gems/runtime/configloader.py |
123 |
15 |
0 |
88% |
| src/flag_gems/runtime/error.py |
8 |
4 |
0 |
50% |
| src/flag_gems/runtime/register.py |
79 |
24 |
0 |
70% |
| src/flag_gems/testing/__init__.py |
29 |
3 |
0 |
90% |
| src/flag_gems/utils/__init__.py |
6 |
0 |
0 |
100% |
| src/flag_gems/utils/code_cache.py |
26 |
3 |
0 |
88% |
| src/flag_gems/utils/code_utils.py |
96 |
38 |
0 |
60% |
| src/flag_gems/utils/codegen_config_utils.py |
58 |
22 |
0 |
62% |
| src/flag_gems/utils/device_info.py |
59 |
18 |
0 |
69% |
| src/flag_gems/utils/libentry.py |
379 |
72 |
0 |
81% |
| src/flag_gems/utils/limits.py |
29 |
23 |
0 |
21% |
| src/flag_gems/utils/models/__init__.py |
3 |
0 |
0 |
100% |
| src/flag_gems/utils/models/model.py |
37 |
9 |
0 |
76% |
| src/flag_gems/utils/models/session.py |
12 |
4 |
0 |
67% |
| src/flag_gems/utils/models/sql.py |
102 |
46 |
0 |
55% |
| src/flag_gems/utils/pointwise_dynamic.py |
788 |
33 |
0 |
96% |
| src/flag_gems/utils/random_utils.py |
60 |
29 |
0 |
52% |
| src/flag_gems/utils/shape_utils.py |
245 |
73 |
0 |
70% |
| src/flag_gems/utils/tensor_wrapper.py |
65 |
20 |
0 |
69% |
| src/flag_gems/utils/triton_driver_helper.py |
6 |
3 |
0 |
50% |
| src/flag_gems/utils/triton_lang_extension.py |
56 |
29 |
0 |
48% |
| src/flag_gems/utils/triton_lang_helper.py |
26 |
9 |
0 |
65% |
| src/flag_gems/utils/type_utils.py |
8 |
0 |
0 |
100% |