| src/flag_gems/__init__.py |
80 |
7 |
0 |
91% |
| src/flag_gems/config.py |
104 |
55 |
2 |
47% |
| 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/_log_softmax_backward_data.py |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/_unsafe_view.py |
76 |
76 |
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 |
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/amin.py |
135 |
135 |
0 |
0% |
| src/flag_gems/experimental_ops/arccosh.py |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/arctanh.py |
39 |
39 |
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/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/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 |
42 |
42 |
0 |
0% |
| src/flag_gems/experimental_ops/hardswish.py |
74 |
74 |
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 |
56 |
56 |
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/log10_.py |
38 |
38 |
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/logaddexp2.py |
68 |
68 |
0 |
0% |
| src/flag_gems/experimental_ops/logical_xor_.py |
41 |
41 |
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/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/relu.py |
39 |
39 |
0 |
0% |
| src/flag_gems/experimental_ops/relu_.py |
28 |
28 |
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 |
46 |
40 |
0 |
13% |
| src/flag_gems/experimental_ops/rsqrt_.py |
38 |
38 |
0 |
0% |
| src/flag_gems/experimental_ops/sgn.py |
60 |
60 |
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/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/softplus.py |
69 |
69 |
0 |
0% |
| src/flag_gems/experimental_ops/special_xlog1py.py |
90 |
90 |
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/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/zeros_like.py |
79 |
79 |
0 |
0% |
| src/flag_gems/fused/DSA/bin_topk.py |
540 |
508 |
0 |
6% |
| src/flag_gems/fused/DSA/indexer_k_tiled.py |
58 |
50 |
0 |
14% |
| src/flag_gems/fused/DSA/sparse_mla.py |
173 |
132 |
0 |
24% |
| src/flag_gems/fused/FLA/__init__.py |
3 |
0 |
0 |
100% |
| src/flag_gems/fused/FLA/chunk.py |
49 |
16 |
0 |
67% |
| src/flag_gems/fused/FLA/chunk_delta_h.py |
163 |
134 |
0 |
18% |
| src/flag_gems/fused/FLA/chunk_fused_tail_vblock.py |
65 |
36 |
0 |
45% |
| src/flag_gems/fused/FLA/chunk_gated_delta_direct.py |
75 |
41 |
0 |
45% |
| src/flag_gems/fused/FLA/chunk_o.py |
69 |
43 |
0 |
38% |
| src/flag_gems/fused/FLA/chunk_scaled_dot_kkt.py |
46 |
46 |
0 |
0% |
| src/flag_gems/fused/FLA/cumsum.py |
90 |
90 |
0 |
0% |
| src/flag_gems/fused/FLA/fused_cumsum_kkt_solve_tril.py |
157 |
129 |
0 |
18% |
| src/flag_gems/fused/FLA/fused_recurrent.py |
256 |
215 |
0 |
16% |
| src/flag_gems/fused/FLA/index.py |
17 |
0 |
0 |
100% |
| 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 |
4 |
0 |
94% |
| src/flag_gems/fused/FLA/wy_fast.py |
50 |
27 |
0 |
46% |
| src/flag_gems/fused/__init__.py |
49 |
0 |
0 |
100% |
| src/flag_gems/fused/add_rms_norm.py |
93 |
58 |
0 |
38% |
| src/flag_gems/fused/apply_repetition_penalties.py |
37 |
16 |
0 |
57% |
| src/flag_gems/fused/bincount.py |
120 |
99 |
0 |
18% |
| src/flag_gems/fused/chunk_gated_delta_rule.py |
107 |
25 |
0 |
77% |
| src/flag_gems/fused/concat_and_cache_mla.py |
80 |
39 |
0 |
51% |
| src/flag_gems/fused/cp_gather_indexer_k_quant_cache.py |
78 |
42 |
0 |
46% |
| src/flag_gems/fused/cross_entropy_loss.py |
376 |
286 |
0 |
24% |
| src/flag_gems/fused/cutlass_scaled_mm.py |
194 |
158 |
0 |
19% |
| src/flag_gems/fused/deepseek_v4_attention_combine_topk_swa_indices.py |
44 |
23 |
0 |
48% |
| src/flag_gems/fused/deepseek_v4_attention_compute_global_topk_indices_and_lens.py |
35 |
18 |
0 |
49% |
| src/flag_gems/fused/deepseek_v4_attention_dequantize_and_gather_k_cache.py |
65 |
54 |
0 |
17% |
| src/flag_gems/fused/deepseek_v4_attention_fused_q_kv_rmsnorm.py |
43 |
20 |
0 |
53% |
| src/flag_gems/fused/flash_mla.py |
107 |
72 |
0 |
33% |
| src/flag_gems/fused/flash_mla_with_kvcache.py |
1201 |
1148 |
0 |
4% |
| src/flag_gems/fused/flashmla_sparse.py |
501 |
446 |
0 |
11% |
| src/flag_gems/fused/fused_add_rms_norm.py |
40 |
17 |
0 |
58% |
| src/flag_gems/fused/fused_deepseek_v4_qnorm_rope_kv_rope_quant_insert.py |
199 |
193 |
0 |
3% |
| src/flag_gems/fused/fused_inv_rope_fp8_quant.py |
101 |
87 |
0 |
14% |
| src/flag_gems/fused/fused_marlin_moe.py |
326 |
215 |
0 |
34% |
| src/flag_gems/fused/fused_moe.py |
921 |
546 |
0 |
41% |
| src/flag_gems/fused/geglu.py |
70 |
34 |
0 |
51% |
| src/flag_gems/fused/gelu_and_mul.py |
70 |
35 |
0 |
50% |
| src/flag_gems/fused/grouped_topk.py |
133 |
123 |
0 |
8% |
| src/flag_gems/fused/indexer_k_quant_and_cache.py |
56 |
48 |
0 |
14% |
| src/flag_gems/fused/instance_norm.py |
308 |
213 |
0 |
31% |
| src/flag_gems/fused/mhc/__init__.py |
6 |
0 |
0 |
100% |
| src/flag_gems/fused/mhc/hc_head_fused_kernel.py |
112 |
72 |
0 |
36% |
| src/flag_gems/fused/mhc/hc_split_sinkhorn.py |
312 |
260 |
0 |
17% |
| src/flag_gems/fused/mhc/mhc_bwd.py |
249 |
177 |
0 |
29% |
| src/flag_gems/fused/mhc/mhc_post.py |
93 |
60 |
0 |
35% |
| src/flag_gems/fused/mhc/mhc_pre.py |
346 |
275 |
0 |
21% |
| src/flag_gems/fused/moe_align_block_size.py |
334 |
257 |
0 |
23% |
| src/flag_gems/fused/moe_sum.py |
30 |
15 |
0 |
50% |
| src/flag_gems/fused/outer.py |
26 |
6 |
0 |
77% |
| src/flag_gems/fused/pack_seq.py |
50 |
23 |
0 |
54% |
| src/flag_gems/fused/reglu.py |
80 |
37 |
0 |
54% |
| 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 |
40 |
21 |
0 |
48% |
| src/flag_gems/fused/rwkv_mm_sparsity.py |
33 |
14 |
0 |
58% |
| src/flag_gems/fused/silu_and_mul.py |
39 |
14 |
0 |
64% |
| src/flag_gems/fused/silu_and_mul_with_clamp.py |
53 |
25 |
0 |
53% |
| src/flag_gems/fused/skip_layernorm.py |
50 |
20 |
0 |
60% |
| src/flag_gems/fused/sparse_attention.py |
53 |
47 |
0 |
11% |
| src/flag_gems/fused/swiglu.py |
74 |
35 |
0 |
53% |
| src/flag_gems/fused/top_k_per_row_decode.py |
305 |
284 |
0 |
7% |
| src/flag_gems/fused/top_k_per_row_prefill.py |
43 |
24 |
0 |
44% |
| src/flag_gems/fused/topk_softmax.py |
54 |
46 |
0 |
15% |
| src/flag_gems/fused/topk_softplus_sqrt.py |
88 |
67 |
0 |
24% |
| src/flag_gems/fused/unpack_seq.py |
41 |
17 |
0 |
59% |
| src/flag_gems/fused/weight_norm.py |
124 |
97 |
0 |
22% |
| src/flag_gems/fused_moe_mxq.py |
310 |
310 |
0 |
0% |
| src/flag_gems/logging_utils.py |
113 |
71 |
0 |
37% |
| 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 |
306 |
0 |
0 |
100% |
| src/flag_gems/ops/_euclidean_dist.py |
42 |
16 |
0 |
62% |
| src/flag_gems/ops/_functional_sym_constrain_range_for_size.py |
35 |
11 |
0 |
69% |
| src/flag_gems/ops/_is_all_true.py |
47 |
15 |
0 |
68% |
| src/flag_gems/ops/_safe_softmax.py |
56 |
15 |
0 |
73% |
| src/flag_gems/ops/_upsample_nearest_exact1d.py |
134 |
63 |
0 |
53% |
| src/flag_gems/ops/abs.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/absolute.py |
43 |
23 |
0 |
47% |
| src/flag_gems/ops/acos.py |
14 |
1 |
0 |
93% |
| src/flag_gems/ops/act_quant.py |
91 |
77 |
0 |
15% |
| src/flag_gems/ops/add.py |
68 |
15 |
0 |
78% |
| src/flag_gems/ops/addcdiv.py |
17 |
1 |
0 |
94% |
| src/flag_gems/ops/addcmul.py |
22 |
1 |
0 |
95% |
| src/flag_gems/ops/addmm.py |
85 |
43 |
0 |
49% |
| src/flag_gems/ops/addmv.py |
57 |
21 |
0 |
63% |
| src/flag_gems/ops/addr.py |
44 |
19 |
0 |
57% |
| src/flag_gems/ops/affine_grid_generator.py |
52 |
31 |
0 |
40% |
| src/flag_gems/ops/alias_copy.py |
49 |
27 |
0 |
45% |
| src/flag_gems/ops/all.py |
103 |
33 |
0 |
68% |
| src/flag_gems/ops/amax.py |
92 |
50 |
0 |
46% |
| src/flag_gems/ops/aminmax.py |
120 |
59 |
0 |
51% |
| 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 |
48 |
13 |
0 |
73% |
| src/flag_gems/ops/arcsinh.py |
54 |
17 |
0 |
69% |
| src/flag_gems/ops/arcsinh_.py |
34 |
15 |
0 |
56% |
| src/flag_gems/ops/arctanh_.py |
42 |
22 |
0 |
48% |
| src/flag_gems/ops/argmax.py |
163 |
92 |
0 |
44% |
| src/flag_gems/ops/argmin.py |
161 |
89 |
0 |
45% |
| src/flag_gems/ops/argsort.py |
7 |
3 |
0 |
57% |
| src/flag_gems/ops/as_strided_copy.py |
161 |
57 |
0 |
65% |
| src/flag_gems/ops/asinh.py |
18 |
4 |
0 |
78% |
| src/flag_gems/ops/asinh_.py |
49 |
29 |
0 |
41% |
| src/flag_gems/ops/assert_async.py |
11 |
2 |
0 |
82% |
| src/flag_gems/ops/atan.py |
18 |
1 |
0 |
94% |
| src/flag_gems/ops/atan2.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/atanh.py |
21 |
9 |
0 |
57% |
| src/flag_gems/ops/attention.py |
430 |
299 |
0 |
30% |
| src/flag_gems/ops/avg_pool2d.py |
150 |
86 |
0 |
43% |
| src/flag_gems/ops/avg_pool3d.py |
183 |
121 |
0 |
34% |
| src/flag_gems/ops/baddbmm.py |
159 |
111 |
0 |
30% |
| src/flag_gems/ops/batch_norm.py |
154 |
100 |
0 |
35% |
| src/flag_gems/ops/bernoulli_.py |
45 |
24 |
0 |
47% |
| src/flag_gems/ops/bincount.py |
55 |
13 |
0 |
76% |
| 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 |
11 |
3 |
0 |
73% |
| src/flag_gems/ops/bmm.py |
95 |
60 |
0 |
37% |
| src/flag_gems/ops/cat.py |
193 |
88 |
0 |
54% |
| src/flag_gems/ops/cauchy.py |
62 |
29 |
0 |
53% |
| 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/clamp_max.py |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/clip.py |
35 |
5 |
0 |
86% |
| src/flag_gems/ops/col2im.py |
83 |
40 |
0 |
52% |
| src/flag_gems/ops/concatenate.py |
8 |
0 |
0 |
100% |
| src/flag_gems/ops/conj_physical.py |
31 |
8 |
0 |
74% |
| 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 |
42 |
0 |
60% |
| src/flag_gems/ops/conv_depthwise2d.py |
10 |
6 |
0 |
40% |
| src/flag_gems/ops/conv_transpose1d.py |
82 |
38 |
0 |
54% |
| src/flag_gems/ops/conv_transpose2d.py |
750 |
414 |
0 |
45% |
| src/flag_gems/ops/copy.py |
65 |
19 |
0 |
71% |
| src/flag_gems/ops/copysign.py |
33 |
14 |
0 |
58% |
| src/flag_gems/ops/cos.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/cosh.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/count_nonzero.py |
95 |
36 |
0 |
62% |
| src/flag_gems/ops/ctc_loss.py |
446 |
265 |
0 |
41% |
| src/flag_gems/ops/cudnn_convolution.py |
25 |
4 |
0 |
84% |
| src/flag_gems/ops/cummax.py |
242 |
153 |
0 |
37% |
| src/flag_gems/ops/cummin.py |
242 |
153 |
0 |
37% |
| src/flag_gems/ops/cumprod.py |
240 |
126 |
0 |
48% |
| 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/diff.py |
94 |
35 |
0 |
63% |
| src/flag_gems/ops/digamma_.py |
54 |
38 |
0 |
30% |
| src/flag_gems/ops/div.py |
185 |
72 |
0 |
61% |
| 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/embedding_dense_backward.py |
66 |
31 |
0 |
53% |
| 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/expm1.py |
18 |
1 |
0 |
94% |
| src/flag_gems/ops/exponential_.py |
202 |
138 |
0 |
32% |
| 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/feature_dropout.py |
86 |
35 |
0 |
59% |
| src/flag_gems/ops/fft.py |
692 |
668 |
0 |
3% |
| src/flag_gems/ops/fill.py |
61 |
9 |
0 |
85% |
| src/flag_gems/ops/flash_api.py |
537 |
73 |
0 |
86% |
| src/flag_gems/ops/flash_kernel.py |
574 |
497 |
0 |
13% |
| src/flag_gems/ops/flip.py |
32 |
1 |
0 |
97% |
| src/flag_gems/ops/floor.py |
18 |
2 |
0 |
89% |
| src/flag_gems/ops/floor_.py |
41 |
19 |
0 |
54% |
| src/flag_gems/ops/fmin.py |
87 |
26 |
0 |
70% |
| src/flag_gems/ops/fmod.py |
34 |
10 |
0 |
71% |
| src/flag_gems/ops/fmod_.py |
41 |
20 |
0 |
51% |
| src/flag_gems/ops/fp8_matmul.py |
67 |
54 |
0 |
19% |
| src/flag_gems/ops/fp8_mqa_logits.py |
54 |
42 |
0 |
22% |
| src/flag_gems/ops/fp8_paged_mqa_logits.py |
125 |
114 |
0 |
9% |
| 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 |
132 |
3 |
0 |
98% |
| src/flag_gems/ops/gcd.py |
172 |
99 |
0 |
42% |
| 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_paged_mqa_logits_metadata.py |
40 |
34 |
0 |
15% |
| src/flag_gems/ops/get_scheduler_metadata.py |
333 |
309 |
0 |
7% |
| src/flag_gems/ops/glu.py |
34 |
7 |
0 |
79% |
| src/flag_gems/ops/greater.py |
31 |
8 |
0 |
74% |
| src/flag_gems/ops/grid_sample.py |
2179 |
1892 |
0 |
13% |
| src/flag_gems/ops/group_gemm.py |
273 |
235 |
0 |
14% |
| src/flag_gems/ops/groupnorm.py |
139 |
88 |
0 |
37% |
| src/flag_gems/ops/gt.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/hadamard_transform.py |
412 |
369 |
0 |
10% |
| src/flag_gems/ops/hardsigmoid.py |
30 |
16 |
0 |
47% |
| src/flag_gems/ops/hardswish_.py |
44 |
19 |
0 |
57% |
| src/flag_gems/ops/histc.py |
64 |
35 |
0 |
45% |
| src/flag_gems/ops/hstack.py |
106 |
34 |
0 |
68% |
| src/flag_gems/ops/hypot.py |
75 |
33 |
0 |
56% |
| src/flag_gems/ops/i0.py |
63 |
30 |
0 |
52% |
| src/flag_gems/ops/i0_.py |
47 |
25 |
0 |
47% |
| 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_copy_.py |
160 |
2 |
0 |
99% |
| src/flag_gems/ops/index_put.py |
263 |
10 |
0 |
96% |
| src/flag_gems/ops/index_reduce.py |
443 |
249 |
0 |
44% |
| 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/isneginf.py |
19 |
3 |
0 |
84% |
| 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/leaky_relu.py |
55 |
13 |
0 |
76% |
| src/flag_gems/ops/lerp.py |
35 |
3 |
0 |
91% |
| src/flag_gems/ops/lift_fresh_copy.py |
56 |
34 |
0 |
39% |
| src/flag_gems/ops/linspace.py |
34 |
10 |
0 |
71% |
| src/flag_gems/ops/log.py |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/log10.py |
18 |
1 |
0 |
94% |
| src/flag_gems/ops/log1p.py |
15 |
3 |
0 |
80% |
| src/flag_gems/ops/log1p_.py |
38 |
15 |
0 |
61% |
| src/flag_gems/ops/log_sigmoid.py |
12 |
1 |
0 |
92% |
| src/flag_gems/ops/log_softmax.py |
115 |
54 |
0 |
53% |
| src/flag_gems/ops/logaddexp.py |
20 |
5 |
0 |
75% |
| 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/logit.py |
80 |
41 |
0 |
49% |
| src/flag_gems/ops/logit_.py |
63 |
30 |
0 |
52% |
| src/flag_gems/ops/logspace.py |
37 |
9 |
0 |
76% |
| src/flag_gems/ops/logsumexp.py |
120 |
77 |
0 |
36% |
| src/flag_gems/ops/lt.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/margin_ranking_loss.py |
111 |
63 |
0 |
43% |
| 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/max_pool3d_with_indices.py |
164 |
145 |
0 |
12% |
| src/flag_gems/ops/maximum.py |
18 |
4 |
0 |
78% |
| src/flag_gems/ops/mean.py |
223 |
121 |
0 |
46% |
| src/flag_gems/ops/median.py |
799 |
435 |
0 |
46% |
| 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 |
283 |
192 |
0 |
32% |
| 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 |
36 |
6 |
0 |
83% |
| 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/nanmedian.py |
825 |
548 |
0 |
34% |
| src/flag_gems/ops/ne.py |
19 |
2 |
0 |
89% |
| src/flag_gems/ops/neg.py |
14 |
1 |
0 |
93% |
| src/flag_gems/ops/new_full.py |
15 |
10 |
0 |
33% |
| src/flag_gems/ops/nll_loss_nd.py |
138 |
121 |
0 |
12% |
| src/flag_gems/ops/nllloss.py |
196 |
119 |
0 |
39% |
| src/flag_gems/ops/nonzero.py |
46 |
13 |
0 |
72% |
| src/flag_gems/ops/nonzero_numpy.py |
7 |
0 |
0 |
100% |
| src/flag_gems/ops/normal.py |
61 |
4 |
0 |
93% |
| src/flag_gems/ops/one_hot.py |
35 |
14 |
0 |
60% |
| 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 |
279 |
0 |
0 |
100% |
| src/flag_gems/ops/per_token_group_quant_fp8.py |
523 |
462 |
0 |
12% |
| src/flag_gems/ops/pixel_shuffle.py |
46 |
18 |
0 |
61% |
| src/flag_gems/ops/pixel_unshuffle.py |
87 |
32 |
0 |
63% |
| src/flag_gems/ops/poisson.py |
74 |
42 |
0 |
43% |
| src/flag_gems/ops/polar.py |
17 |
3 |
0 |
82% |
| src/flag_gems/ops/pow.py |
33 |
3 |
0 |
91% |
| src/flag_gems/ops/prelu.py |
62 |
22 |
0 |
65% |
| src/flag_gems/ops/prod.py |
86 |
31 |
0 |
64% |
| src/flag_gems/ops/quantile.py |
153 |
85 |
0 |
44% |
| src/flag_gems/ops/rad2deg.py |
16 |
1 |
0 |
94% |
| 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/randint.py |
59 |
28 |
0 |
53% |
| src/flag_gems/ops/randint_like.py |
51 |
25 |
0 |
51% |
| 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/reflection_pad1d.py |
78 |
36 |
0 |
54% |
| src/flag_gems/ops/reflection_pad1d_backward.py |
89 |
47 |
0 |
47% |
| src/flag_gems/ops/reflection_pad2d.py |
93 |
42 |
0 |
55% |
| src/flag_gems/ops/relu.py |
21 |
2 |
0 |
90% |
| src/flag_gems/ops/relu6.py |
28 |
9 |
0 |
68% |
| src/flag_gems/ops/remainder.py |
38 |
8 |
0 |
79% |
| src/flag_gems/ops/renorm.py |
107 |
43 |
0 |
60% |
| src/flag_gems/ops/repeat.py |
251 |
2 |
0 |
99% |
| src/flag_gems/ops/repeat_interleave.py |
86 |
22 |
0 |
74% |
| src/flag_gems/ops/replication_pad1d.py |
79 |
27 |
0 |
66% |
| src/flag_gems/ops/replication_pad3d.py |
50 |
26 |
0 |
48% |
| src/flag_gems/ops/resolve_conj.py |
103 |
86 |
0 |
17% |
| src/flag_gems/ops/resolve_neg.py |
7 |
0 |
0 |
100% |
| src/flag_gems/ops/rms_norm.py |
166 |
112 |
0 |
33% |
| src/flag_gems/ops/roll.py |
279 |
95 |
0 |
66% |
| src/flag_gems/ops/rot90.py |
98 |
62 |
0 |
37% |
| src/flag_gems/ops/round.py |
115 |
50 |
0 |
57% |
| src/flag_gems/ops/rrelu_with_noise_backward.py |
41 |
18 |
0 |
56% |
| src/flag_gems/ops/rsqrt.py |
15 |
1 |
0 |
93% |
| src/flag_gems/ops/rsub.py |
18 |
2 |
0 |
89% |
| src/flag_gems/ops/scaled_mm.py |
213 |
124 |
0 |
42% |
| src/flag_gems/ops/scaled_softmax.py |
126 |
74 |
0 |
41% |
| 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/scatter_reduce.py |
386 |
255 |
0 |
34% |
| src/flag_gems/ops/searchsorted.py |
136 |
46 |
0 |
66% |
| src/flag_gems/ops/segment_reduce.py |
591 |
384 |
0 |
35% |
| src/flag_gems/ops/select_backward.py |
50 |
19 |
0 |
62% |
| src/flag_gems/ops/select_scatter.py |
45 |
14 |
0 |
69% |
| src/flag_gems/ops/selu.py |
45 |
22 |
0 |
51% |
| src/flag_gems/ops/selu_.py |
43 |
21 |
0 |
51% |
| src/flag_gems/ops/sgn_.py |
41 |
18 |
0 |
56% |
| src/flag_gems/ops/sigmoid.py |
29 |
5 |
0 |
83% |
| src/flag_gems/ops/signbit.py |
30 |
14 |
0 |
53% |
| src/flag_gems/ops/silu.py |
32 |
8 |
0 |
75% |
| src/flag_gems/ops/sin.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/sinh_.py |
40 |
15 |
0 |
62% |
| src/flag_gems/ops/slice_backward.py |
36 |
12 |
0 |
67% |
| src/flag_gems/ops/slice_scatter.py |
51 |
17 |
0 |
67% |
| src/flag_gems/ops/smooth_l1_loss.py |
167 |
68 |
0 |
59% |
| src/flag_gems/ops/soft_margin_loss.py |
126 |
78 |
0 |
38% |
| src/flag_gems/ops/softmax.py |
247 |
157 |
0 |
36% |
| src/flag_gems/ops/softplus.py |
17 |
5 |
0 |
71% |
| src/flag_gems/ops/softshrink.py |
52 |
20 |
0 |
62% |
| src/flag_gems/ops/sort.py |
211 |
124 |
0 |
41% |
| src/flag_gems/ops/special_i0e.py |
47 |
22 |
0 |
53% |
| src/flag_gems/ops/special_i1.py |
76 |
48 |
0 |
37% |
| src/flag_gems/ops/split_with_sizes_copy.py |
43 |
7 |
0 |
84% |
| src/flag_gems/ops/sqrt.py |
16 |
1 |
0 |
94% |
| src/flag_gems/ops/square.py |
21 |
2 |
0 |
90% |
| 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 |
62 |
13 |
0 |
79% |
| src/flag_gems/ops/sum.py |
242 |
110 |
0 |
55% |
| src/flag_gems/ops/svd.py |
1838 |
1452 |
0 |
21% |
| src/flag_gems/ops/t_copy.py |
70 |
33 |
0 |
53% |
| src/flag_gems/ops/tan.py |
17 |
2 |
0 |
88% |
| src/flag_gems/ops/tanh.py |
28 |
3 |
0 |
89% |
| src/flag_gems/ops/tensor_split.py |
95 |
86 |
0 |
9% |
| 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 |
52 |
7 |
0 |
87% |
| src/flag_gems/ops/topk.py |
327 |
248 |
0 |
24% |
| src/flag_gems/ops/trace.py |
55 |
23 |
0 |
58% |
| src/flag_gems/ops/tril.py |
424 |
217 |
0 |
49% |
| src/flag_gems/ops/triu.py |
106 |
36 |
0 |
66% |
| src/flag_gems/ops/unfold_backward.py |
45 |
16 |
0 |
64% |
| src/flag_gems/ops/uniform.py |
41 |
20 |
0 |
51% |
| src/flag_gems/ops/unique.py |
287 |
146 |
0 |
49% |
| src/flag_gems/ops/unique_consecutive.py |
168 |
84 |
0 |
50% |
| src/flag_gems/ops/unique_dim.py |
428 |
199 |
0 |
54% |
| src/flag_gems/ops/upsample_bicubic2d.py |
127 |
92 |
0 |
28% |
| src/flag_gems/ops/upsample_bicubic2d_aa.py |
178 |
141 |
0 |
21% |
| src/flag_gems/ops/upsample_bicubic2d_aa_backward.py |
190 |
122 |
0 |
36% |
| src/flag_gems/ops/upsample_linear1d.py |
56 |
25 |
0 |
55% |
| src/flag_gems/ops/upsample_linear1d_backward.py |
71 |
49 |
0 |
31% |
| 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/upsample_nearest3d.py |
63 |
32 |
0 |
49% |
| src/flag_gems/ops/upsample_trilinear3d.py |
107 |
69 |
0 |
36% |
| src/flag_gems/ops/var.py |
114 |
57 |
0 |
50% |
| 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/view_copy.py |
46 |
18 |
0 |
61% |
| src/flag_gems/ops/vstack.py |
76 |
20 |
0 |
74% |
| src/flag_gems/ops/w8a8_block_fp8_matmul.py |
97 |
46 |
0 |
53% |
| src/flag_gems/ops/weightnorm.py |
160 |
114 |
0 |
29% |
| src/flag_gems/ops/where.py |
48 |
5 |
0 |
90% |
| src/flag_gems/ops/zero.py |
49 |
19 |
0 |
61% |
| 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 |
61 |
50 |
0 |
18% |
| src/flag_gems/patches/patch_vllm_all.py |
200 |
170 |
0 |
15% |
| src/flag_gems/runtime/__init__.py |
35 |
16 |
0 |
54% |
| src/flag_gems/runtime/backend/__init__.py |
291 |
65 |
0 |
78% |
| src/flag_gems/runtime/backend/_aipu/__init__.py |
6 |
6 |
0 |
0% |
| 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 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_amd/heuristics_config_utils.py |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_amd/ops/__init__.py |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/__init__.py |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/__init__.py |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/fused_add_rms_norm.py |
57 |
57 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_llama_arch.py |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_5_attention.py |
46 |
46 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_5_conv1d.py |
70 |
70 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_5_gated_delta.py |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_5_rmsnorm.py |
69 |
69 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_5_rmsnorm_gated.py |
61 |
61 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_layer_norm.py |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_mlp.py |
77 |
77 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_rmsnorm.py |
85 |
85 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/fused/patch_qwen3_rope.py |
87 |
87 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/heuristics_config_utils.py |
122 |
122 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/int8/__init__.py |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/int8/quantize_live.py |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/int8/replace.py |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/int8/tle_int8_linear.py |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/__init__.py |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/addmm.py |
181 |
181 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/all.py |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/any.py |
97 |
97 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/argmax.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/attention.py |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/bmm.py |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/cumsum.py |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/div.py |
115 |
115 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/exponential_.py |
76 |
76 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/full.py |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/gather.py |
129 |
129 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/index_select.py |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/int_mm.py |
125 |
125 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/isin.py |
147 |
147 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/lt.py |
41 |
41 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/masked_fill.py |
119 |
119 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/min.py |
91 |
91 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/mm.py |
329 |
329 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/multinomial.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/pow.py |
296 |
296 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/quantile.py |
86 |
86 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/quantized_linear_dynamic.py |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/rms_norm.py |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/rope.py |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/scatter.py |
253 |
253 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/silu_and_mul.py |
24 |
24 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/sort.py |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/sub.py |
279 |
279 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/topk.py |
195 |
195 |
0 |
0% |
| src/flag_gems/runtime/backend/_arm/ops/where.py |
126 |
126 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/__init__.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/__init__.py |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/chunk.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/chunk_delta_h.py |
112 |
112 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/chunk_o.py |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/chunk_scaled_dot_kkt.py |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/cumsum.py |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/fused_qkvzba_split_reshape.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/fused_recurrent.py |
100 |
100 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/layernorm_guard.py |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/sigmoid_gating.py |
95 |
95 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/solve_tril.py |
180 |
180 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/utils.py |
13 |
13 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fla/wy_fast.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/__init__.py |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/cross_entropy_loss.py |
380 |
380 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/flash_mla.py |
95 |
95 |
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/fused_moe.py |
687 |
687 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/moe_align_block_size.py |
352 |
352 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/fused/moe_sum.py |
84 |
84 |
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/fused/sparse_attention.py |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/heuristics_config_utils.py |
136 |
136 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/__init__.py |
75 |
75 |
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 |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/argmax.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/argmin.py |
98 |
98 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/attention.py |
399 |
399 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/baddbmm.py |
147 |
147 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/batch_norm.py |
159 |
159 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/bmm.py |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/cat.py |
76 |
76 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/count_nonzero.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/cummax.py |
260 |
260 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/cummin.py |
260 |
260 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/cumsum.py |
258 |
258 |
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 |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/full_like.py |
21 |
21 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/gather.py |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/gather_ascend.py |
234 |
234 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/gather_collapsed.py |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/gather_collapsed_uintdiv.py |
255 |
255 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/gather_dispatch.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/groupnorm.py |
168 |
168 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/hadamard_transform.py |
123 |
123 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/hstack.py |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index.py |
116 |
116 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/index_add.py |
82 |
82 |
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 |
147 |
147 |
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 |
113 |
113 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/masked_fill.py |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/masked_scatter.py |
98 |
98 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/masked_select.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/matmul_bf16.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/matmul_int8.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/max.py |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/mean.py |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/min.py |
101 |
101 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/mm.py |
92 |
92 |
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 |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/randperm.py |
270 |
270 |
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/scatter.py |
73 |
73 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/scatter_add_.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/select_backward.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/select_scatter.py |
21 |
21 |
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 |
256 |
256 |
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 |
310 |
310 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/upsample_bicubic2d_aa.py |
178 |
178 |
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 |
156 |
156 |
0 |
0% |
| src/flag_gems/runtime/backend/_ascend/ops/vector_norm.py |
273 |
273 |
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 |
28 |
28 |
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 |
8 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/__init__.py |
18 |
18 |
0 |
0% |
| 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 |
140 |
140 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/abs.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/acos.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/add.py |
64 |
64 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/addcdiv.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/addcmul.py |
18 |
18 |
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 |
51 |
51 |
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 |
58 |
58 |
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 |
206 |
206 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/dropout.py |
88 |
88 |
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 |
91 |
91 |
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 |
86 |
86 |
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 |
49 |
49 |
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 |
49 |
49 |
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 |
83 |
83 |
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 |
72 |
72 |
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/nan_to_num.py |
46 |
46 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/ne.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/neg.py |
45 |
45 |
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 |
349 |
349 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/per_token_group_quant_fp8.py |
80 |
80 |
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 |
224 |
224 |
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 |
331 |
331 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/reciprocal.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/relu.py |
50 |
50 |
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 |
113 |
113 |
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 |
47 |
47 |
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 |
552 |
552 |
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 |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/stack.py |
104 |
104 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sub.py |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/sum.py |
151 |
151 |
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/threshold.py |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/tile.py |
306 |
306 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/to.py |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/ops/topk.py |
319 |
319 |
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 |
94 |
94 |
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 |
1111 |
1111 |
0 |
0% |
| src/flag_gems/runtime/backend/_cambricon/utils/reduce_utils.py |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/__init__.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/fused/__init__.py |
24 |
24 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/heuristics_config_utils.py |
113 |
113 |
0 |
0% |
| src/flag_gems/runtime/backend/_enflame/ops/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_hygon/__init__.py |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/fused/__init__.py |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/fused/sparse_attention.py |
55 |
55 |
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 |
20 |
20 |
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 |
375 |
375 |
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/hadamard_transform.py |
87 |
87 |
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/matmul_bf16.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/matmul_int8.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/mm.py |
94 |
94 |
0 |
0% |
| src/flag_gems/runtime/backend/_hygon/ops/pow.py |
51 |
51 |
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 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/fused/__init__.py |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/fused/sparse_attention.py |
66 |
66 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/heuristics_config_utils.py |
131 |
131 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/__init__.py |
5 |
5 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/div.py |
188 |
188 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/hadamard_transform.py |
87 |
87 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/matmul_bf16.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_iluvatar/ops/matmul_int8.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/__init__.py |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/__init__.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/fused/bincount.py |
63 |
63 |
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/sparse_attention.py |
42 |
42 |
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 |
182 |
182 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/abs.py |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/acos.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/add.py |
66 |
66 |
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 |
85 |
85 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addmv.py |
69 |
69 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/addr.py |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/all.py |
109 |
109 |
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/apply_repetition_penalties.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/arange.py |
58 |
58 |
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 |
172 |
172 |
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 |
18 |
18 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/count_nonzero.py |
88 |
88 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cummax.py |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cummin.py |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/cumprod.py |
240 |
240 |
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/digamma_.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/div.py |
201 |
201 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/dot.py |
61 |
61 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/dropout.py |
122 |
122 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/elu.py |
29 |
29 |
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 |
95 |
95 |
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 |
197 |
197 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/ge.py |
28 |
28 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/gelu.py |
59 |
59 |
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/hadamard_transform.py |
74 |
74 |
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_add_.py |
80 |
80 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_put.py |
282 |
282 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/index_select.py |
46 |
46 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/instance_norm.py |
343 |
343 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isclose.py |
48 |
48 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isfinite.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/isin.py |
141 |
141 |
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 |
14 |
14 |
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 |
43 |
43 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/matmul_bf16.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/matmul_int8.py |
54 |
54 |
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 |
88 |
88 |
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 |
64 |
64 |
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 |
173 |
173 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/nonzero.py |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/normal.py |
63 |
63 |
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 |
282 |
282 |
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/reflection_pad1d.py |
76 |
76 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/reflection_pad2d.py |
91 |
91 |
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/round.py |
58 |
58 |
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 |
243 |
243 |
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 |
47 |
47 |
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/soft_margin_loss.py |
98 |
98 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/softmax.py |
186 |
186 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/softplus.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/softshrink.py |
54 |
54 |
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 |
64 |
64 |
0 |
0% |
| src/flag_gems/runtime/backend/_kunlunxin/ops/sum.py |
171 |
171 |
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 |
69 |
69 |
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_linear1d.py |
55 |
55 |
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/weight_norm.py |
127 |
127 |
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/zero.py |
32 |
32 |
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 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/fused/__init__.py |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/fused/flash_mla.py |
104 |
104 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/fused/sparse_attention.py |
55 |
55 |
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 |
34 |
34 |
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/hadamard_transform.py |
87 |
87 |
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 |
232 |
232 |
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/matmul_bf16.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_metax/ops/matmul_int8.py |
54 |
54 |
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 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/fused/__init__.py |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/fused/cross_entropy_loss.py |
376 |
376 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/fused/sparse_attention.py |
67 |
67 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/heuristics_config_utils.py |
243 |
243 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/__init__.py |
45 |
45 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/addmm.py |
139 |
139 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/all.py |
141 |
141 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/amax.py |
143 |
143 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/any.py |
135 |
135 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/arange.py |
89 |
89 |
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 |
135 |
135 |
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 |
73 |
73 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/gelu.py |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/index_add.py |
74 |
74 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/index_put.py |
382 |
382 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/index_select.py |
101 |
101 |
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/log_softmax.py |
259 |
259 |
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 |
196 |
196 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/normal.py |
59 |
59 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/one_hot.py |
108 |
108 |
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/repeat.py |
176 |
176 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/repeat_interleave.py |
227 |
227 |
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 |
211 |
211 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/tanh.py |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/tile.py |
174 |
174 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/unique.py |
284 |
284 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/utils.py |
5 |
5 |
0 |
0% |
| src/flag_gems/runtime/backend/_mthreads/ops/w8a8_block_fp8_matmul.py |
141 |
141 |
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 |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/fused/fused_add_rms_norm.py |
37 |
37 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/heuristics_config_utils.py |
243 |
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 |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/fill.py |
107 |
107 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/fp8_einsum.py |
11 |
11 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/mm.py |
432 |
432 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/mul.py |
87 |
87 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/sqrt.py |
35 |
35 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/w8a8_block_fp8_bmm.py |
276 |
276 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/hopper/ops/w8a8_block_fp8_matmul.py |
178 |
178 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/ops/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/ops/add.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/ops/gelu.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/turing/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_nvidia/turing/ops/__init__.py |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_nvidia/turing/ops/_safe_softmax.py |
75 |
75 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/__init__.py |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/fused/__init__.py |
1 |
1 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/heuristics_config_utils.py |
134 |
134 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/__init__.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/addmm.py |
53 |
53 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/argmax.py |
111 |
111 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/argmin.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/bmm.py |
88 |
88 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/conv1d.py |
12 |
12 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/conv2d.py |
108 |
108 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/conv_depthwise2d.py |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/flash_attention.py |
135 |
135 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/gelu.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/layernorm.py |
166 |
166 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/mean.py |
85 |
85 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/mm.py |
102 |
102 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/mv.py |
33 |
33 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/pow.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/rsqrt.py |
14 |
14 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/sigmoid.py |
29 |
29 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/silu.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/softmax.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/thnn_conv2d.py |
9 |
9 |
0 |
0% |
| src/flag_gems/runtime/backend/_spacemit/ops/where.py |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/__init__.py |
400 |
400 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/__init__.py |
10 |
10 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/bincount.py |
228 |
228 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/flash_mla.py |
94 |
94 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/fused_add_rms_norm.py |
72 |
72 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/fused_moe.py |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/fused_recurrent.py |
261 |
261 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/hc_head_fused_kernel.py |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/reshape_and_cache_flash.py |
44 |
44 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/skip_layernorm.py |
85 |
85 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/fused/sparse_attention.py |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/heuristics_config_utils.py |
260 |
260 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/monkey_patch.py |
1104 |
1104 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/__init__.py |
92 |
92 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/_safe_softmax.py |
55 |
55 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/_upsample_nearest_exact1d.py |
133 |
133 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/abs.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/add.py |
119 |
119 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/addmm.py |
85 |
85 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/aminmax.py |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/angle.py |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/arcsinh.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/attention.py |
440 |
440 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/bitwise_and.py |
30 |
30 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/bitwise_left_shift.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/bitwise_right_shift.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/clamp.py |
152 |
152 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/conj_physical.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/conv2d.py |
181 |
181 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/cos.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/count_nonzero.py |
95 |
95 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/ctc_loss.py |
446 |
446 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/cumsum.py |
328 |
328 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/div.py |
367 |
367 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/dropout.py |
90 |
90 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/embedding.py |
103 |
103 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/eq.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/exponential_.py |
200 |
200 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/fft.py |
692 |
692 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/fill.py |
64 |
64 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/flash_api.py |
536 |
536 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/flash_kernel.py |
572 |
572 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/gather.py |
136 |
136 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/ge.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/gelu.py |
60 |
60 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/hypot.py |
2 |
2 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/i0.py |
63 |
63 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/i0_.py |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/index_add.py |
158 |
158 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/index_put.py |
265 |
265 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/index_select.py |
93 |
93 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/isin.py |
130 |
130 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/isnan.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/layernorm.py |
243 |
243 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/lift_fresh_copy.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/linspace.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/log_softmax.py |
206 |
206 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/logaddexp.py |
20 |
20 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/logical_and.py |
23 |
23 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/logical_or.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/margin_ranking_loss.py |
111 |
111 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/masked_select.py |
105 |
105 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/mean.py |
223 |
223 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/mul.py |
99 |
99 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/multinomial.py |
52 |
52 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/mv.py |
40 |
40 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/neg.py |
16 |
16 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/nonzero.py |
46 |
46 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/one_hot.py |
102 |
102 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/pad.py |
286 |
286 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/polar.py |
17 |
17 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/pow.py |
36 |
36 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/prelu.py |
62 |
62 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/quantile.py |
154 |
154 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/randperm.py |
266 |
266 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/reflection_pad2d.py |
93 |
93 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/repeat.py |
255 |
255 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/repeat_interleave.py |
86 |
86 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/resolve_neg.py |
7 |
7 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/rms_norm.py |
224 |
224 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/scatter.py |
269 |
269 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/scatter_reduce.py |
470 |
470 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/select_backward.py |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/sigmoid.py |
32 |
32 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/soft_margin_loss.py |
126 |
126 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/softmax.py |
247 |
247 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/sort.py |
242 |
242 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/special_i0e.py |
47 |
47 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/special_i1.py |
76 |
76 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/sub.py |
96 |
96 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/sum.py |
295 |
295 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/svd.py |
1842 |
1842 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/t_copy.py |
70 |
70 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/tile.py |
259 |
259 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/to.py |
58 |
58 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/topk.py |
323 |
323 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/triu.py |
108 |
108 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/unique.py |
286 |
286 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/unique_consecutive.py |
191 |
191 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/upsample_bicubic2d.py |
127 |
127 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/upsample_linear1d.py |
56 |
56 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/upsample_nearest2d.py |
52 |
52 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/vdot.py |
147 |
147 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/where.py |
51 |
51 |
0 |
0% |
| src/flag_gems/runtime/backend/_sunrise/ops/zero.py |
49 |
49 |
0 |
0% |
| src/flag_gems/runtime/backend/_thead/__init__.py |
4 |
4 |
0 |
0% |
| src/flag_gems/runtime/backend/_thead/fused/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_thead/heuristics_config_utils.py |
121 |
121 |
0 |
0% |
| src/flag_gems/runtime/backend/_thead/ops/__init__.py |
0 |
0 |
0 |
100% |
| src/flag_gems/runtime/backend/_tsingmicro/__init__.py |
31 |
31 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/fused/__init__.py |
3 |
3 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/fused/cross_entropy_loss.py |
506 |
506 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/fused/moe_align_block_size.py |
136 |
136 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/heuristics_config_utils.py |
230 |
230 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/__init__.py |
26 |
26 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/argmax.py |
132 |
132 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/attention.py |
399 |
399 |
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 |
114 |
114 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/flash_api.py |
349 |
349 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/flash_kernel.py |
521 |
521 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/hstack.py |
50 |
50 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/index.py |
281 |
281 |
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/matmul_bf16.py |
54 |
54 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/matmul_int8.py |
57 |
57 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/mm.py |
97 |
97 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/normal.py |
61 |
61 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/pow.py |
34 |
34 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/randn.py |
78 |
78 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/randn_like.py |
22 |
22 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/rms_norm.py |
119 |
119 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/rsqrt.py |
15 |
15 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/silu_and_mul.py |
37 |
37 |
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/_tsingmicro/ops/vdot.py |
81 |
81 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/zeros.py |
42 |
42 |
0 |
0% |
| src/flag_gems/runtime/backend/_tsingmicro/ops/zeros_like.py |
19 |
19 |
0 |
0% |
| src/flag_gems/runtime/backend/backend_utils.py |
56 |
23 |
0 |
59% |
| src/flag_gems/runtime/backend/device.py |
83 |
42 |
0 |
49% |
| src/flag_gems/runtime/common.py |
31 |
0 |
0 |
100% |
| src/flag_gems/runtime/configs_loader.py |
237 |
84 |
0 |
65% |
| src/flag_gems/runtime/error.py |
9 |
3 |
0 |
67% |
| src/flag_gems/runtime/flagtune.py |
91 |
36 |
0 |
60% |
| src/flag_gems/runtime/op_registrar.py |
100 |
31 |
0 |
69% |
| src/flag_gems/runtime/precision_register.py |
94 |
94 |
0 |
0% |
| src/flag_gems/testing/__init__.py |
29 |
3 |
0 |
90% |
| src/flag_gems/utils/__init__.py |
7 |
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 |
466 |
73 |
0 |
84% |
| 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 |
2 |
0 |
83% |
| src/flag_gems/utils/models/sql.py |
102 |
17 |
0 |
83% |
| src/flag_gems/utils/pointwise_dynamic.py |
1019 |
60 |
0 |
94% |
| src/flag_gems/utils/pointwise_dynamic_backup.py |
805 |
805 |
0 |
0% |
| src/flag_gems/utils/random_utils.py |
78 |
43 |
0 |
45% |
| 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 |
64 |
16 |
0 |
75% |
| src/flag_gems/utils/triton_version_utils.py |
35 |
14 |
0 |
60% |
| src/flag_gems/utils/type_utils.py |
8 |
0 |
0 |
100% |