amirkl94 03ee48111d

Feature: Support Relu2 in FusedMoE fp8 cutlass path (#27261)

2025-11-16 13:39:44 -05:00

..

modular_kernel_tools

[Kernels] Split up fused_moe/layer.py, isolate more modular kernel code (#28064)

2025-11-11 07:29:02 -07:00

__init__.py

…

parallel_utils.py

…

test_batched_deepgemm.py

…

test_batched_moe.py

…

test_block_fp8.py

…

test_block_int8.py

…

test_count_expert_num_tokens.py

…

test_cutlass_grouped_gemm.py

…

test_cutlass_moe.py

…

test_deepep_deepgemm_moe.py

…

test_deepep_moe.py

…

test_deepgemm.py

…

test_flashinfer_moe.py

…

test_flashinfer.py

Feature: Support Relu2 in FusedMoE fp8 cutlass path (#27261)

2025-11-16 13:39:44 -05:00

test_gpt_oss_triton_kernels.py

…

test_grouped_topk.py

…

test_modular_kernel_combinations.py

…

test_moe_align_block_size.py

…

test_moe_permute_unpermute.py

…

test_moe.py

…

test_nvfp4_moe.py

…

test_ocp_mx_moe.py

…

test_pplx_cutlass_moe.py

…

test_pplx_moe.py

…

test_rocm_aiter_topk.py

…

test_silu_mul_fp8_quant_deep_gemm.py

…

test_triton_moe_ptpc_fp8.py

…

utils.py

…