shixianc 7d94577138

Add torch golden impl for moe_align_block_size kernel test (#20653) ...

Signed-off-by: Shixian Cui <shixian@amazon.com>
Co-authored-by: Shixian Cui <shixian@amazon.com>

2025-07-19 02:32:36 -07:00

..

modular_kernel_tools

[Kernel] DeepGemm MoE : Integrate triton permute / unpermute kernels (#20903)

2025-07-17 08:10:37 +00:00

__init__.py

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

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

parallel_utils.py

[Misc] Add unit tests for MoE ModularKernel combinations + Profiling utility (#20449)

2025-07-11 07:51:46 -07:00

test_batched_moe.py

[Refactor] Change the way of import triton (#20774)

2025-07-12 19:39:55 -07:00

test_block_fp8.py

[Feature] Integrate SM100 DeepGEMM support (#20087)

2025-07-10 20:18:05 -07:00

test_block_int8.py

[Kernels] MoE refactor (#19636)

2025-07-02 06:08:27 -07:00

test_count_expert_num_tokens.py

[Bugfix] Fused MoE Modular Kernel chunking loop (#20392)

2025-07-10 20:31:10 +00:00

test_cutlass_grouped_gemm.py

[CI Bugfix] Fix pre-commit failures on main (#20502)

2025-07-04 14:17:30 -07:00

test_cutlass_moe.py

[Perf] Add swap_ab to SM90 FP8 non-block CUTLASS moe grouped gemm (#20911)

2025-07-18 04:34:43 +00:00

test_deepep_deepgemm_moe.py

[Feature] Integrate SM100 DeepGEMM support (#20087)

2025-07-10 20:18:05 -07:00

test_deepep_moe.py

[Misc] Add SPDX-FileCopyrightText (#20428)

2025-07-04 07:40:42 +00:00

test_deepgemm.py

[CI] update typos config for CI pre-commit and fix some spells (#20919)

2025-07-15 21:12:40 -07:00

test_modular_kernel_combinations.py

[Misc] Add unit tests for MoE ModularKernel combinations + Profiling utility (#20449)

2025-07-11 07:51:46 -07:00

test_moe_align_block_size.py

Add torch golden impl for moe_align_block_size kernel test (#20653)

2025-07-19 02:32:36 -07:00

test_moe_permute_unpermute.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_moe.py

[Feature][Quantization] MXFP4 support for MOE models (#17888)

2025-07-09 13:19:02 -07:00

test_mxfp4_moe.py

[Feature][Quantization] MXFP4 support for MOE models (#17888)

2025-07-09 13:19:02 -07:00

test_nvfp4_moe.py

[Kernels] MoE refactor (#19636)

2025-07-02 06:08:27 -07:00

test_pplx_cutlass_moe.py

[Performance] Performance improvements in non-blockwise fp8 CUTLASS MoE (#20762)

2025-07-17 09:56:44 -04:00

test_pplx_moe.py

[Misc] MoE ModularKernel : Introduce TopKWeightAndReduce (#20648)

2025-07-10 14:40:38 -07:00

test_rocm_aiter_topk.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_silu_mul_fp8_quant_deep_gemm.py

[EP+DP] Optimize the little operations in the DeepGEMM + DeepEP low latency case (#19885)

2025-06-23 11:07:47 -07:00

test_triton_moe_ptpc_fp8.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

utils.py

[Kernel] Enable fp8 support for pplx and BatchedTritonExperts. (#18864)

2025-07-03 14:55:40 -07:00