danisereb 0a6a3a1290 Add support for ModelOpt MXFP8 MoE models (#35986) ...

Signed-off-by: Daniel Serebrenik <daserebrenik@nvidia.com>

2026-03-08 13:00:05 -07:00

..

modular_kernel_tools

[Hardware] Replace torch.cuda.synchronize() api with torch.accelerator.synchronize (#36085)

2026-03-05 10:36:39 +00:00

__init__.py

…

parallel_utils.py

…

test_batched_deepgemm.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_batched_moe.py

…

test_block_fp8.py

[Hardware] Replace torch.cuda.synchronize() api with torch.accelerator.synchronize (#36085)

2026-03-05 10:36:39 +00:00

test_block_int8.py

…

test_count_expert_num_tokens.py

…

test_cpu_fused_moe.py

…

test_cutedsl_moe.py

…

test_cutlass_moe.py

[Hardware] Replace torch.cuda.synchronize() api with torch.accelerator.synchronize (#36085)

2026-03-05 10:36:39 +00:00

test_cutlass_mxfp8_grouped_mm.py

[Kernel] Integrate SM100 MXFP8 blockscaled grouped MM and quant kernels (#34448)

2026-03-01 23:31:19 -08:00

test_deepep_deepgemm_moe.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_deepep_moe.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_deepgemm.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_flashinfer_moe.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_flashinfer.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_fused_topk.py

…

test_gpt_oss_triton_kernels.py

[Core] Move test utility to test file (#35672)

2026-03-02 10:56:03 -05:00

test_grouped_topk.py

[Kernel] Optimize grouped topk kernel (#34206)

2026-02-20 01:34:45 -08:00

test_marlin_vs_trtllm_mxint4.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_modular_kernel_combinations.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

test_modular_oai_triton_moe.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_moe_align_block_size.py

[Bugfix] Fix expert_ids padding values in moe_align_block_size kernel (#35161)

2026-02-24 17:14:24 -08:00

test_moe_permute_unpermute.py

…

test_moe.py

[Hardware] Replace torch.cuda.synchronize() api with torch.accelerator.synchronize (#36085)

2026-03-05 10:36:39 +00:00

test_nvfp4_moe.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

test_ocp_mx_moe.py

Add support for ModelOpt MXFP8 MoE models (#35986)

2026-03-08 13:00:05 -07:00

test_rocm_aiter_topk.py

…

test_routing_simulator.py

…

test_routing.py

…

test_shared_fused_moe_routed_transform.py

…

test_silu_mul_fp8_quant_deep_gemm.py

…

test_silu_mul_per_token_group_quant_fp8_colmajor.py

…

test_triton_moe_no_act_mul.py

…

test_triton_moe_ptpc_fp8.py

…

test_unquantized_backend_selection.py

…

utils.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00