---
id: pr-tensorrt-llm-12074
repo: NVIDIA/TensorRT-LLM
pr: 12074
title: '[TRTLLM-11289][feat] Integrate CuteDSL''s bf16 dense GEMMs'
author: peaceh-nv
date: '2026-04-30'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/12074
source_category: upstream-code
architectures:
- sm100
tags:
- MLA
- arch_pipeline
- attention kernel
- attention_kv
- blackwell
- compiler_runtime
- cute dsl
- gemm_quant
- scheduler_autotune
- sm100
techniques:
- autotuning
- fine-grained-quantization
- persistent-kernels
- pipeline-stages
- tile-scheduling
hardware_features:
- blackwell
kernel_types:
- attention
- compiler
- gemm
- mla
- runtime
- scheduler
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: e9c4b18afdd8439738b5816adb557afafa34fa14
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:
- tensorrt_llm/_torch/compilation/utils.py
- tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py
- tensorrt_llm/_torch/cute_dsl_kernels/blackwell/dense_gemm_persistent.py
- tensorrt_llm/_torch/model_config.py
- tensorrt_llm/_torch/models/modeling_deepseekv3.py
- tensorrt_llm/_torch/modules/attention.py
- tensorrt_llm/_torch/modules/linear.py
- tensorrt_llm/_torch/pyexecutor/model_loader.py
- tensorrt_llm/llmapi/llm_args.py
- tests/integration/defs/accuracy/test_llm_api_pytorch.py
- tests/integration/test_lists/qa/llm_function_core.txt
- tests/scripts/cute_dsl_kernels/run_dense_bf16_gemm_persistent.py
- tests/unittest/api_stability/references/llm.yaml
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-12074
head_ref_oid: f96c1b6cfd7e72073dd0da3e95608f9833a49f15
---
# [TRTLLM-11289][feat] Integrate CuteDSL's bf16 dense GEMMs

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [12074](https://github.com/NVIDIA/TensorRT-LLM/pull/12074)
- 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

- `tensorrt_llm/_torch/compilation/utils.py`
- `tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py`
- `tensorrt_llm/_torch/cute_dsl_kernels/blackwell/dense_gemm_persistent.py`
- `tensorrt_llm/_torch/model_config.py`
- `tensorrt_llm/_torch/models/modeling_deepseekv3.py`
- `tensorrt_llm/_torch/modules/attention.py`
- `tensorrt_llm/_torch/modules/linear.py`
- `tensorrt_llm/_torch/pyexecutor/model_loader.py`
- `tensorrt_llm/llmapi/llm_args.py`
- `tests/integration/defs/accuracy/test_llm_api_pytorch.py`
- `tests/integration/test_lists/qa/llm_function_core.txt`
- `tests/scripts/cute_dsl_kernels/run_dense_bf16_gemm_persistent.py`
- `tests/unittest/api_stability/references/llm.yaml`

## Evidence Bundle

`evidence/pull-bundles/tensorrt-llm/gh-12074`
