---
id: pr-deepgemm-83
repo: deepseek-ai/DeepGEMM
pr: 83
title: Use 1D TMA store instead of 3D
author: LyricZhao
date: '2025-04-11'
url: https://github.com/deepseek-ai/DeepGEMM/pull/83
source_category: upstream-code
architectures:
- sm100
tags:
- TMA
- arch_pipeline
- compiler_runtime
- gemm_quant
- moe_routing
techniques:
- async-copy
- fine-grained-quantization
- pipeline-stages
hardware_features:
- fp8
- tma
kernel_types:
- compiler
- gemm
- moe
- runtime
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: 2e7e58011b03d394743fb91a8aa97e4ae5250aea
inclusion_reason: 'Extract the architecture assumption first: SM target, TMA/WGMMA/tcgen path, cluster use, and pipeline stages.'
changed_paths:
- deep_gemm/include/deep_gemm/fp8_gemm.cuh
- deep_gemm/include/deep_gemm/tma_utils.cuh
- deep_gemm/jit_kernels/gemm.py
- deep_gemm/jit_kernels/m_grouped_gemm.py
- indexing/main.cu
artifact_dir: evidence/pull-bundles/deepgemm/gh-83
head_ref_oid: b0d64817a77d93fe07ec22aee9cd877b3ce538cc
---
# Use 1D TMA store instead of 3D

- Repository: `deepseek-ai/DeepGEMM`
- PR: [83](https://github.com/deepseek-ai/DeepGEMM/pull/83)
- Decision: `Extract the architecture assumption first: SM target, TMA/WGMMA/tcgen path, cluster use, and pipeline stages.`
- Reason: Extract the architecture assumption first: SM target, TMA/WGMMA/tcgen path, cluster use, and pipeline stages.

## Changed Paths

- `deep_gemm/include/deep_gemm/fp8_gemm.cuh`
- `deep_gemm/include/deep_gemm/tma_utils.cuh`
- `deep_gemm/jit_kernels/gemm.py`
- `deep_gemm/jit_kernels/m_grouped_gemm.py`
- `indexing/main.cu`

## Evidence Bundle

`evidence/pull-bundles/deepgemm/gh-83`
