---
id: pr-flashinfer-2581
repo: flashinfer-ai/flashinfer
pr: 2581
title: Implement `cutlass_fused_moe` mxfp8
author: zianglih
date: '2026-03-07'
url: https://github.com/flashinfer-ai/flashinfer/pull/2581
source_category: upstream-code
architectures:
- sm100
tags:
- fp8
- gemm
- kernel-fusion
- moe
- tma
techniques:
- kernel-fusion
hardware_features:
- fp8
- tma
kernel_types:
- gemm
- moe
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: 62d1b0c24a975b54d2b76f0d1769cd0a564b7eff
inclusion_reason: kernel file changes
changed_paths:
- csrc/fused_moe/cutlass_backend/cutlass_fused_moe_instantiation.cu
- csrc/fused_moe/cutlass_backend/cutlass_fused_moe_kernels.cuh
- csrc/fused_moe/cutlass_backend/flashinfer_cutlass_fused_moe_binding.cu
- csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/include/moe_gemm_kernels.h
- csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/include/moe_kernels.h
- csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/launchers/moe_gemm_tma_ws_launcher.inl
- csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/moe_gemm_kernels_fp8_fp8.cu
- csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/moe_gemm_template_dispatch.h
- csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/moe_gemm_template_dispatch_tma_ws.h
- flashinfer/jit/gemm/cutlass/generate_kernels.py
- tests/moe/test_trtllm_cutlass_fused_moe.py
head_ref_oid: 6ab67b45c922c4919d67314d1412172ad289215c
artifact_dir: evidence/pull-bundles/flashinfer/gh-2581
---
# Implement `cutlass_fused_moe` mxfp8

- Repository: `flashinfer-ai/flashinfer`
- PR: [2581](https://github.com/flashinfer-ai/flashinfer/pull/2581)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `csrc/fused_moe/cutlass_backend/cutlass_fused_moe_instantiation.cu`
- `csrc/fused_moe/cutlass_backend/cutlass_fused_moe_kernels.cuh`
- `csrc/fused_moe/cutlass_backend/flashinfer_cutlass_fused_moe_binding.cu`
- `csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/include/moe_gemm_kernels.h`
- `csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/include/moe_kernels.h`
- `csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/launchers/moe_gemm_tma_ws_launcher.inl`
- `csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/moe_gemm_kernels_fp8_fp8.cu`
- `csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/moe_gemm_template_dispatch.h`
- `csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/moe_gemm_template_dispatch_tma_ws.h`
- `flashinfer/jit/gemm/cutlass/generate_kernels.py`
- `tests/moe/test_trtllm_cutlass_fused_moe.py`

## Evidence Bundle

`evidence/pull-bundles/flashinfer/gh-2581`
