Michael Goin f708bd4904

[CI] Add E2E Blackwell Quantized MoE Test (#25723) ...

Signed-off-by: mgoin <mgoin64@gmail.com>

2025-09-26 12:23:00 -07:00

..

configs

[Harware][AMD][Model] Triton MoE tuning configs for GLM-4.5 for MI300X (#25703)

2025-09-26 01:18:20 -07:00

__init__.py

[Bugfix][CPU] Add placeholder to avoid import errors when using fused_moe ops on platforms without triton (#25137)

2025-09-19 07:41:12 +00:00

batched_deep_gemm_moe.py

[Bugfix] Fix chunked a2_scales in modular kernels (#25264)

2025-09-19 19:42:01 +00:00

batched_triton_or_deep_gemm_moe.py

[Bugfix] Fix chunked a2_scales in modular kernels (#25264)

2025-09-19 19:42:01 +00:00

config.py

Handle triton kernel import exception (#25319)

2025-09-23 00:56:00 -07:00

cpu_fused_moe.py

[fix]: add Arm 4bit fused moe support (#23809)

2025-09-24 01:32:22 +00:00

cutlass_moe.py

[Bugfix] Fix chunked a2_scales in modular kernels (#25264)

2025-09-19 19:42:01 +00:00

deep_gemm_moe.py

[Bugfix] Fix chunked a2_scales in modular kernels (#25264)

2025-09-19 19:42:01 +00:00

deep_gemm_utils.py

[Bugfix] DeepGemm utils : Fix hardcoded type-cast (#21517)

2025-07-24 20:17:29 -07:00

deepep_ht_prepare_finalize.py

[Core/DBO][2/N] Dual-Batch Overlap add DeepEP High Throughput support and Prefill support (#24845)

2025-09-23 16:02:10 +00:00

deepep_ll_prepare_finalize.py

[Core/DBO][2/N] Dual-Batch Overlap add DeepEP High Throughput support and Prefill support (#24845)

2025-09-23 16:02:10 +00:00

flashinfer_cutlass_moe.py

feat: BF16 FlashInfer Fused Cutlass MOE for Hopper and Blackwell Expert Parallel (#25503)

2025-09-24 18:50:04 -04:00

flashinfer_cutlass_prepare_finalize.py

feat: BF16 FlashInfer Fused Cutlass MOE for Hopper and Blackwell Expert Parallel (#25503)

2025-09-24 18:50:04 -04:00

flashinfer_trtllm_moe.py

[CI] Add E2E Blackwell Quantized MoE Test (#25723)

2025-09-26 12:23:00 -07:00

fused_batched_moe.py

[Bugfix] Fix chunked a2_scales in modular kernels (#25264)

2025-09-19 19:42:01 +00:00

fused_marlin_moe.py

Remove redundant mutates_args and dispatch_key for direct_register_custom_op (#25512)

2025-09-23 22:48:40 +00:00

fused_moe.py

[Model] Add LongCat-Flash (#23991)

2025-09-24 21:53:40 -07:00

gpt_oss_triton_kernels_moe.py

[Bugfix] Fix for the import error from #24588 (#25481)

2025-09-23 21:31:08 +00:00

layer.py

[Bugfix] Fix Shared Expert/Zero expert code in FusedMoE.process_chunk (#25698)

2025-09-26 01:25:28 -07:00

modular_kernel.py

feat: BF16 FlashInfer Fused Cutlass MOE for Hopper and Blackwell Expert Parallel (#25503)

2025-09-24 18:50:04 -04:00

moe_align_block_size.py

[Refactor] Remove moe_align_block_size_triton (#21335)

2025-07-26 07:08:29 -07:00

moe_pallas.py

[Doc]: fix typos in Python comments (#24026)

2025-09-01 09:38:20 +00:00

moe_permute_unpermute.py

[Kernel] CUTLASS MoE FP8: Integrate cuda moe permute/unpermute (#23045)

2025-08-20 10:35:26 -04:00

moe_torch_iterative.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

pplx_prepare_finalize.py

[Kernels] Overlap shared experts with combine instead of dispatch (#24254)

2025-09-18 12:10:21 +08:00

prepare_finalize.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

rocm_aiter_fused_moe.py

Remove redundant mutates_args and dispatch_key for direct_register_custom_op (#25512)

2025-09-23 22:48:40 +00:00

routing_simulator.py

[Docs] Fix warnings in mkdocs build (continued) (#24092)

2025-09-10 06:23:28 -07:00

topk_weight_and_reduce.py

[Misc] ModularKernel : Perform WeightAndReduce inside TritonExperts & DeepGemmExperts (#20725)

2025-07-14 19:47:16 +00:00

triton_deep_gemm_moe.py

[Bugfix] Fix chunked a2_scales in modular kernels (#25264)

2025-09-19 19:42:01 +00:00

trtllm_moe.py

[Bugfix] Fix chunked a2_scales in modular kernels (#25264)

2025-09-19 19:42:01 +00:00

utils.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00