---
id: pr-flashinfer-2944
repo: flashinfer-ai/flashinfer
pr: 2944
title: 'feat: Add CuTe DSL grouped-gemm + combine fusion support'
author: nvcastet
date: '2026-05-14'
url: https://github.com/flashinfer-ai/flashinfer/pull/2944
source_category: upstream-code
architectures:
- sm100
tags:
- TMA
- attention_kv
- benchmark_test
- blackwell
- compiler_runtime
- cuda kernel
- cute dsl
- cutlass
- gemm_quant
- moe_routing
- sm100
- topk
techniques:
- async-copy
- epilogue-fusion
- fine-grained-quantization
- kernel-fusion
hardware_features:
- blackwell
- tma
kernel_types:
- attention
- compiler
- gemm
- moe
languages:
- cuda-cpp
- cute-dsl
- ptx
captured_at: '2026-05-16'
status: merged
merge_sha: 89bfb6882edbf21c8dd4fee7fa719f6206e9de36
inclusion_reason: 'Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.'
changed_paths:
- flashinfer/gemm/kernels/grouped_gemm_masked_blackwell.py
- scripts/task_test_multi_gpu_comm_kernels.sh
- tests/gemm/test_multi_gpu_cute_dsl_blockscaled_gemm_fusion.py
artifact_dir: evidence/pull-bundles/flashinfer/gh-2944
head_ref_oid: 3da3b65d967dce7a7440b11ae443e55fc13a0f93
---
# feat: Add CuTe DSL grouped-gemm + combine fusion support

- Repository: `flashinfer-ai/flashinfer`
- PR: [2944](https://github.com/flashinfer-ai/flashinfer/pull/2944)
- Decision: `Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.`
- Reason: Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.

## Changed Paths

- `flashinfer/gemm/kernels/grouped_gemm_masked_blackwell.py`
- `scripts/task_test_multi_gpu_comm_kernels.sh`
- `tests/gemm/test_multi_gpu_cute_dsl_blockscaled_gemm_fusion.py`

## Evidence Bundle

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