TJian 612c2edb4f [FEAT] [ROCm]: Add AITER CK 2 Stages MoE support (#17110) ...

Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
Co-authored-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>

2025-05-14 03:03:11 -07:00

..

test_cutlass_moe.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_moe_permute_unpermute.py

permute/unpermute kernel for moe optimization (#14568)

2025-05-02 11:31:55 -07:00

test_moe.py

[FEAT] [ROCm]: Add AITER CK 2 Stages MoE support (#17110)

2025-05-14 03:03:11 -07:00

test_nvfp4_moe.py

[Hardware/NVIDIA/Kernel] Enable nvidia/DeepSeek-R1-FP4 Model (#16362)

2025-05-09 16:24:41 -07:00

test_rocm_aiter_topk.py

[FEAT] [ROCm] [V1]: Add AITER biased group topk for DeepSeekV3 (#17955)

2025-05-13 22:03:47 -07:00

test_triton_moe_ptpc_fp8.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00