Flydsl if else by xudoyuan · Pull Request #2740 · ROCm/aiter

xudoyuan · 2026-04-14T10:29:16Z

Motivation

flydsl test case normalization

Technical Details

Test Plan

Test Result

Submission Checklist

Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

github-actions · 2026-04-14T10:29:50Z

🏷️ CI Guide

Runs automatically on every PR:

✅ Pre-checks (submodule verification, code formatting)
✅ Aiter op tests (gfx942 + gfx950)
✅ Triton tests (only when aiter/ops/triton/** or related paths are changed)

Extended tests (opt-in via labels):

Label	Tests
`ci:triton-355`	Run Triton tests on MI355 in addition to MI325
`ci:sglang`	SGLang integration tests
`ci:atom`	ATOM benchmark (DeepSeek-R1 + GPT-OSS)
`ci:vllm`	vLLM benchmark
`ci:all`	All of the above

Add labels via the sidebar or gh pr edit 2740 --add-label <label>

github-actions

Remaining comments which cannot be posted as a review comment to avoid GitHub Rate Limit

ruff

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1341 in 399dc5d

if const_expr(not gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1355 in 399dc5d

if const_expr(_pp_has_scale[_p]):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1374 in 399dc5d

if const_expr(not gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1387 in 399dc5d

if const_expr(_b_type == "gate"):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1460 in 399dc5d

if const_expr(not gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1465 in 399dc5d

if const_expr(not gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1477 in 399dc5d

if const_expr(not gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1482 in 399dc5d

if const_expr(not use_async_copy):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1500 in 399dc5d

if const_expr(use_async_copy):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1527 in 399dc5d

if const_expr(use_async_copy):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1548 in 399dc5d

if const_expr(k_main2_py < 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1601 in 399dc5d

if const_expr(use_async_copy):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1611 in 399dc5d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1627 in 399dc5d

if const_expr(tile_m == 32):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1631 in 399dc5d

elif const_expr(tile_m == 64):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1645 in 399dc5d

if const_expr(k_main2_py > 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1712 in 399dc5d

if const_expr(odd_k_tiles):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1728 in 399dc5d

if const_expr(use_async_copy):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1748 in 399dc5d

if const_expr(not use_async_copy):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1790 in 399dc5d

if const_expr(not _is_splitk):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1801 in 399dc5d

if const_expr(epilogue_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1817 in 399dc5d

if const_expr(lds_out is None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1834 in 399dc5d

if const_expr(_apply_weight):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1836 in 399dc5d

if const_expr(tw_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1848 in 399dc5d

if const_expr(_apply_weight):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1850 in 399dc5d

if const_expr(_need_quant):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1966 in 399dc5d

if const_expr(_need_sort):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 1974 in 399dc5d

if const_expr(_need_quant and not _is_splitk):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2020 in 399dc5d

if const_expr(_pack_bytes == 1):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2030 in 399dc5d

elif const_expr(_pack_bytes == 2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2050 in 399dc5d

if const_expr(_need_sort):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2081 in 399dc5d

elif const_expr(_is_splitk):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2118 in 399dc5d

if const_expr(gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2147 in 399dc5d

elif const_expr(_is_splitk):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2301 in 399dc5d

if const_expr(gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2719 in 399dc5d

if const_expr(not bool(accumulate)):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2750 in 399dc5d

if const_expr(not is_f16_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2751 in 399dc5d

if const_expr(is_f4_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2768 in 399dc5d

if const_expr(not is_f16_b):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2817 in 399dc5d

if const_expr(_persistent):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2846 in 399dc5d

if const_expr(_persistent):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2869 in 399dc5d

if const_expr(_persistent):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2896 in 399dc5d

if const_expr(pack_M < _scale_pack_m):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2912 in 399dc5d

if const_expr(is_f16_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2913 in 399dc5d

if const_expr(bytes_per_thread_x % 16 != 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2919 in 399dc5d

if const_expr(bytes_per_thread_x % 16 == 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2921 in 399dc5d

elif const_expr(bytes_per_thread_x % 8 == 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2923 in 399dc5d

elif const_expr(bytes_per_thread_x % 4 == 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2973 in 399dc5d

if const_expr(x_load_bytes == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3035 in 399dc5d

if const_expr(x_load_bytes == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3037 in 399dc5d

elif const_expr(x_load_bytes == 8):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3065 in 399dc5d

if const_expr(pack_N < _scale_pack_n):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3187 in 399dc5d

if const_expr(k_shift_bits > 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3246 in 399dc5d

if const_expr(x_load_bytes == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3261 in 399dc5d

elif const_expr(x_load_bytes == 8):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3328 in 399dc5d

if const_expr(b_hi_loader is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3339 in 399dc5d

if const_expr(prefetch_epilogue):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3340 in 399dc5d

if const_expr(enable_bias):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3351 in 399dc5d

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3389 in 399dc5d

if const_expr(b_hi_loader is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3407 in 399dc5d

if const_expr(_m_scale_shift_i32 is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3418 in 399dc5d

if const_expr(_n_scale_shift_i32 is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3430 in 399dc5d

const_expr((a0_prefetch is not None)

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3436 in 399dc5d

const_expr((a1_prefetch is not None)

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3446 in 399dc5d

if const_expr(is_f8_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3507 in 399dc5d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3510 in 399dc5d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3512 in 399dc5d

if const_expr(num_acc_n < 4):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3515 in 399dc5d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3519 in 399dc5d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3525 in 399dc5d

if const_expr(dswr_tail > sche_iters):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3534 in 399dc5d

if const_expr(sche_i >= dswr_start - 1):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3540 in 399dc5d

if const_expr(pack_K >= _scale_pack_k):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3565 in 399dc5d

if const_expr(_b_split_enabled):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3605 in 399dc5d

if const_expr(k_main2_py < 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3620 in 399dc5d

if const_expr(k_main2_py > 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3705 in 399dc5d

if const_expr(odd_k_tiles):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3783 in 399dc5d

if const_expr(epilogue_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3801 in 399dc5d

if const_expr(lds_out is None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3830 in 399dc5d

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3832 in 399dc5d

if const_expr(tw_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3845 in 399dc5d

if const_expr(is_int8):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3847 in 399dc5d

if const_expr(enable_bias):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3850 in 399dc5d

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3874 in 399dc5d

if const_expr(accumulate):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3894 in 399dc5d

if const_expr(not bool(accumulate)):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3956 in 399dc5d

if const_expr(_persistent):

github-actions

Remaining comments which cannot be posted as a review comment to avoid GitHub Rate Limit

ruff

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2050 in 64bfaa9

if const_expr(_need_sort):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2081 in 64bfaa9

elif const_expr(_is_splitk):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2118 in 64bfaa9

if const_expr(gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2147 in 64bfaa9

elif const_expr(_is_splitk):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2301 in 64bfaa9

if const_expr(gate_only):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2719 in 64bfaa9

if const_expr(not bool(accumulate)):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2750 in 64bfaa9

if const_expr(not is_f16_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2751 in 64bfaa9

if const_expr(is_f4_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2768 in 64bfaa9

if const_expr(not is_f16_b):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2817 in 64bfaa9

if const_expr(_persistent):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2846 in 64bfaa9

if const_expr(_persistent):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2869 in 64bfaa9

if const_expr(_persistent):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2896 in 64bfaa9

if const_expr(pack_M < _scale_pack_m):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2912 in 64bfaa9

if const_expr(is_f16_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2913 in 64bfaa9

if const_expr(bytes_per_thread_x % 16 != 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2919 in 64bfaa9

if const_expr(bytes_per_thread_x % 16 == 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2921 in 64bfaa9

elif const_expr(bytes_per_thread_x % 8 == 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2923 in 64bfaa9

elif const_expr(bytes_per_thread_x % 4 == 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 2973 in 64bfaa9

if const_expr(x_load_bytes == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3035 in 64bfaa9

if const_expr(x_load_bytes == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3037 in 64bfaa9

elif const_expr(x_load_bytes == 8):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3065 in 64bfaa9

if const_expr(pack_N < _scale_pack_n):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3187 in 64bfaa9

if const_expr(k_shift_bits > 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3246 in 64bfaa9

if const_expr(x_load_bytes == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3261 in 64bfaa9

elif const_expr(x_load_bytes == 8):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3328 in 64bfaa9

if const_expr(b_hi_loader is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3339 in 64bfaa9

if const_expr(prefetch_epilogue):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3340 in 64bfaa9

if const_expr(enable_bias):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3351 in 64bfaa9

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3389 in 64bfaa9

if const_expr(b_hi_loader is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3407 in 64bfaa9

if const_expr(_m_scale_shift_i32 is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3418 in 64bfaa9

if const_expr(_n_scale_shift_i32 is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3429 in 64bfaa9

if const_expr(

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3435 in 64bfaa9

elif const_expr(

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3446 in 64bfaa9

if const_expr(is_f8_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3507 in 64bfaa9

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3510 in 64bfaa9

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3512 in 64bfaa9

if const_expr(num_acc_n < 4):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3515 in 64bfaa9

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3519 in 64bfaa9

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3525 in 64bfaa9

if const_expr(dswr_tail > sche_iters):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3534 in 64bfaa9

if const_expr(sche_i >= dswr_start - 1):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3540 in 64bfaa9

if const_expr(pack_K >= _scale_pack_k):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3565 in 64bfaa9

if const_expr(_b_split_enabled):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3605 in 64bfaa9

if const_expr(k_main2_py < 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3620 in 64bfaa9

if const_expr(k_main2_py > 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3705 in 64bfaa9

if const_expr(odd_k_tiles):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3783 in 64bfaa9

if const_expr(epilogue_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3801 in 64bfaa9

if const_expr(lds_out is None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3830 in 64bfaa9

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3832 in 64bfaa9

if const_expr(tw_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3845 in 64bfaa9

if const_expr(is_int8):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3847 in 64bfaa9

if const_expr(enable_bias):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3850 in 64bfaa9

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3874 in 64bfaa9

if const_expr(accumulate):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3894 in 64bfaa9

if const_expr(not bool(accumulate)):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3956 in 64bfaa9

if const_expr(_persistent):

github-actions

Remaining comments which cannot be posted as a review comment to avoid GitHub Rate Limit

ruff

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3407 in 64bfaa9

if const_expr(_m_scale_shift_i32 is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3418 in 64bfaa9

if const_expr(_n_scale_shift_i32 is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3429 in 64bfaa9

if const_expr(

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3435 in 64bfaa9

elif const_expr(

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3446 in 64bfaa9

if const_expr(is_f8_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3507 in 64bfaa9

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3510 in 64bfaa9

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3512 in 64bfaa9

if const_expr(num_acc_n < 4):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3515 in 64bfaa9

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3519 in 64bfaa9

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3525 in 64bfaa9

if const_expr(dswr_tail > sche_iters):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3534 in 64bfaa9

if const_expr(sche_i >= dswr_start - 1):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3540 in 64bfaa9

if const_expr(pack_K >= _scale_pack_k):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3565 in 64bfaa9

if const_expr(_b_split_enabled):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3605 in 64bfaa9

if const_expr(k_main2_py < 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3620 in 64bfaa9

if const_expr(k_main2_py > 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3705 in 64bfaa9

if const_expr(odd_k_tiles):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3783 in 64bfaa9

if const_expr(epilogue_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3801 in 64bfaa9

if const_expr(lds_out is None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3830 in 64bfaa9

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3832 in 64bfaa9

if const_expr(tw_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3845 in 64bfaa9

if const_expr(is_int8):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3847 in 64bfaa9

if const_expr(enable_bias):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3850 in 64bfaa9

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3874 in 64bfaa9

if const_expr(accumulate):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3894 in 64bfaa9

if const_expr(not bool(accumulate)):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3956 in 64bfaa9

if const_expr(_persistent):

github-actions

Remaining comments which cannot be posted as a review comment to avoid GitHub Rate Limit

ruff

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3407 in e50873d

if const_expr(_m_scale_shift_i32 is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3418 in e50873d

if const_expr(_n_scale_shift_i32 is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3429 in e50873d

if const_expr(

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3435 in e50873d

elif const_expr(

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3446 in e50873d

if const_expr(is_f8_a):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3507 in e50873d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3510 in e50873d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3512 in e50873d

if const_expr(num_acc_n < 4):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3515 in e50873d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3519 in e50873d

if const_expr(tile_m == 16):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3525 in e50873d

if const_expr(dswr_tail > sche_iters):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3534 in e50873d

if const_expr(sche_i >= dswr_start - 1):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3540 in e50873d

if const_expr(pack_K >= _scale_pack_k):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3565 in e50873d

if const_expr(_b_split_enabled):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3605 in e50873d

if const_expr(k_main2_py < 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3620 in e50873d

if const_expr(k_main2_py > 0):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3705 in e50873d

if const_expr(odd_k_tiles):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3783 in e50873d

if const_expr(epilogue_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3801 in e50873d

if const_expr(lds_out is None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3830 in e50873d

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3832 in e50873d

if const_expr(tw_pf is not None):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3845 in e50873d

if const_expr(is_int8):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3847 in e50873d

if const_expr(enable_bias):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3850 in e50873d

if const_expr(doweight_stage2):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3874 in e50873d

if const_expr(accumulate):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3894 in e50873d

if const_expr(not bool(accumulate)):

⚠️ [ruff] <F821> _{reported by reviewdog 🐶}
Undefined name const_expr

aiter/aiter/ops/flydsl/kernels/mixed_moe_gemm_2stage.py

Line 3956 in e50873d

if const_expr(_persistent):

github-actions · 2026-04-14T12:36:32Z

🏷️ CI Guide

Runs automatically on every PR:

✅ Pre-checks (submodule verification, code formatting)
✅ Aiter op tests (gfx942 + gfx950)
✅ Triton tests (only when aiter/ops/triton/** or related paths are changed)

Extended tests (opt-in via labels):

Label	Tests
`ci:triton-355`	Run Triton tests on MI355 in addition to MI325
`ci:sglang`	SGLang integration tests
`ci:atom`	ATOM benchmark (DeepSeek-R1 + GPT-OSS)
`ci:vllm`	vLLM benchmark
`ci:all`	All of the above

Add labels via the sidebar or gh pr edit 2740 --add-label <label>

coderfeli

LGTM

xudoyuan added 6 commits April 14, 2026 08:19

[FLYDSL]: if const_expr

eca9e12

[FLYDSL]: acc func arg

ec9dba4

[FLYDSL]: silu_and_mul_fq.py const_expr

a21f0e2

[FLYDSL]: arg initialization

cfe4c26

[FLYDSL]: mixed_moe_gemm_2stage.py const_expr

e057885

[FLYDSL]: mixed_moe_gemm_2stage.py jit if const_expr

399dc5d

xudoyuan requested a review from a team April 14, 2026 10:29

github-actions bot reviewed Apr 14, 2026

View reviewed changes

[FLYDSL]: black files

64bfaa9

github-actions bot reviewed Apr 14, 2026

View reviewed changes

coderfeli marked this pull request as draft April 14, 2026 12:00

Merge branch 'main' into flydsl_if_else

e50873d

github-actions bot reviewed Apr 14, 2026

View reviewed changes

[FLYDSL]: import const_expr

4f23cb2

xudoyuan closed this Apr 14, 2026

xudoyuan reopened this Apr 14, 2026

xudoyuan marked this pull request as ready for review April 14, 2026 12:36

xudoyuan marked this pull request as draft April 14, 2026 12:36

xudoyuan marked this pull request as ready for review April 15, 2026 07:01

Merge branch 'main' into flydsl_if_else

77a881b

coderfeli approved these changes Apr 15, 2026

View reviewed changes

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Flydsl if else#2740

Flydsl if else#2740
xudoyuan wants to merge 10 commits intomainfrom
flydsl_if_else

xudoyuan commented Apr 14, 2026

Uh oh!

github-actions bot commented Apr 14, 2026

Uh oh!

github-actions bot left a comment

Uh oh!

github-actions bot left a comment

Uh oh!

github-actions bot left a comment

Uh oh!

github-actions bot left a comment

Uh oh!

github-actions bot commented Apr 14, 2026

Uh oh!

coderfeli left a comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants

Conversation

xudoyuan commented Apr 14, 2026

Motivation

Technical Details

Test Plan

Test Result

Submission Checklist

Uh oh!

github-actions bot commented Apr 14, 2026

🏷️ CI Guide

Uh oh!

github-actions bot left a comment

Choose a reason for hiding this comment

Uh oh!

github-actions bot left a comment

Choose a reason for hiding this comment

Uh oh!

github-actions bot left a comment

Choose a reason for hiding this comment

Uh oh!

github-actions bot left a comment

Choose a reason for hiding this comment

Uh oh!

github-actions bot commented Apr 14, 2026

🏷️ CI Guide

Uh oh!

coderfeli left a comment

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants