3 people 3e267bae70 [ROCm] triton+triton_kernels 3.8 mxfp4 MoE support (gpt-oss + DeepSeek-V4) (#55934) ...

Signed-off-by: Rohan Potdar <rohan.potdar@amd.com>
Signed-off-by: Rohan138 <rohanpotdar138@gmail.com>
Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
Co-authored-by: Matt <156021403+mawong-amd@users.noreply.github.com>

2026-09-16 20:03:34 -05:00

..

modular_kernel_tools

[ROCm][CI] Enable AITER FP8/unquantized cases in modular-kernel sweep + fix MoRI per-tensor FP8 dispatch (#57074)

2026-09-16 19:53:06 -05:00

__init__.py

[Kernel] DeepEP dispatch-combine kernel integration (#18434)

2025-06-03 12:30:02 -07:00

conftest.py

[CI] Restore clean GPU state before OAI Triton MoE tests (#55453)

2026-09-08 01:15:21 -07:00

parallel_utils.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_b12x.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_batched_deepgemm.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_batched_moe.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_block_fp8.py

[Model] Support Qwen3.8-Flash-Next (#53896)

2026-08-31 05:57:56 +00:00

test_block_int8.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_count_expert_num_tokens.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_cpu_fused_moe.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_cpu_int4_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_cpu_quant_fused_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_cutedsl_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_cutlass_moe.py

[Bugfix] Handle padded routes in CUTLASS MoE permutations (#54747)

2026-09-01 19:08:17 -07:00

test_deepep_deepgemm_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_deepep_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_deepep_v2_async_finalize.py

[Kernel][MoE] DeepEP v2: async finalize to overlap shared experts with combine (#52781)

2026-09-15 00:04:22 -04:00

test_deepep_v2_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_deepgemm.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_flashinfer_b12x_moe.py

[BugFix] weights processing peak memory reduction for nvfp4 MoE layers (#46276)

2026-07-11 02:05:35 +00:00

test_flashinfer_cutedsl_layout.py

[Kernel] Support Nvfp4 Cutedsl Moe Swiglu-oai and Relu2(non-gated) Activation (#47106)

2026-08-06 17:42:45 -07:00

test_flashinfer_cutedsl_nvfp4_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_flashinfer_moe.py

Revert "[Bugfix][MoE] Tune FlashInfer experts to scheduler token limit" (#52989) (#53186)

2026-08-21 12:39:28 -06:00

test_flashinfer.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_flydsl_moe.py

[ROCm][CI] Stabilize MI355 FlyDSL MoE accuracy test (#53024)

2026-08-21 13:17:18 -07:00

test_fused_topk.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_gemma4router.py

[FEAT] [Perf] [Gemma4] Fused Gemma4 Routing Function Triton (#39083)

2026-04-19 09:57:39 +00:00

test_gpt_oss_triton_kernels.py

[ROCm] triton+triton_kernels 3.8 mxfp4 MoE support (gpt-oss + DeepSeek-V4) (#55934)

2026-09-16 20:03:34 -05:00

test_grouped_topk.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_marlin_vs_trtllm_mxint4.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_modular_kernel_combinations.py

[ROCm][CI] Enable AITER FP8/unquantized cases in modular-kernel sweep + fix MoRI per-tensor FP8 dispatch (#57074)

2026-09-16 19:53:06 -05:00

test_modular_oai_triton_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_moe_align_block_size.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_moe_fused_mul_sum.py

[Quantization][Humming] Support MXFP4 weight + block-FP8 activation for MoE (#51332)

2026-08-25 10:21:00 -04:00

test_moe_kernel_oracle.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_moe_layer.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_moe_permute_unpermute.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_moe_weight_loading_padded.py

[Bugfix][MoE] Fix fused block-scale orientation (#50727)

2026-08-11 16:05:38 -04:00

test_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_mxfp4_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_mxfp8_aiter_backend_selection.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_nvfp4_moe.py

[MoE] Remove inplace fused experts mechanism (#43727)

2026-05-27 20:00:19 -07:00

test_ocp_mx_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_profile_modular_kernel.py

[Bugfix] Fix broken profile_modular_kernel.py (#43300)

2026-06-11 12:17:23 +01:00

test_rocm_aiter_moe.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_rocm_aiter_topk.py

[MoE] Move various experts classes to fused_moe/experts/ (#41979)

2026-05-11 07:54:33 +08:00

test_routed_experts_capture_monolithic.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_routing_simulator.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_routing.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_shared_fused_moe_routed_transform.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_silu_mul_fp8_quant_deep_gemm.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_silu_mul_per_token_group_quant_fp8_colmajor.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_situ_mul_fp8_quant.py

[MoE] Generalize masked activation for padded layouts (#51217)

2026-09-01 11:34:13 -04:00

test_topk_softplus_sqrt.py

[New model][Multimodal] Add DeepSeek-V4-Flash-Vision-Exp support (#54566)

2026-09-02 11:09:09 -07:00

test_triton_moe_no_act_mul.py

[XPU][Tests] Make tests device-agnostic (#51968)

2026-08-20 02:25:41 -07:00

test_triton_moe_ptpc_fp8.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_trtllm_bf16_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_trtllm_nvfp4_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_unquantized_backend_selection.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_zen_cpu_fused_moe.py

[Model] Add DeepSeek-V4 CPU backend (#55355)

2026-09-10 14:14:48 +08:00

test_zen_cpu_int8_moe.py

[Model] Add DeepSeek-V4 CPU backend (#55355)

2026-09-10 14:14:48 +08:00

test_zero_expert_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

utils.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00