---
id: pr-thunderkittens-104
repo: HazyResearch/ThunderKittens
pr: 104
title: Fix scaled matmul
author: alex-t-hu
date: '2025-03-31'
url: https://github.com/HazyResearch/ThunderKittens/pull/104
source_category: upstream-code
architectures:
- sm100
tags:
- matmul
techniques: []
hardware_features: []
kernel_types:
- gemm
languages: []
captured_at: '2026-05-16'
status: merged
merge_sha: b815090830b054f30407f5f2ce1c9a903891ee7b
inclusion_reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.
changed_paths:
- config.py
- kernels/torch_scaled/gentests.py
- kernels/torch_scaled/harness.impl
- kernels/torch_scaled/scaled_matmul
- kernels/torch_scaled/scaled_matmul.cu
- kernels/torch_scaled/visualize.py
- thunderkittens.cpp
artifact_dir: evidence/pull-bundles/thunderkittens/gh-104
head_ref_oid: b3d483c694aa744a47e96021550d84d476bea38e
---
# Fix scaled matmul

- Repository: `HazyResearch/ThunderKittens`
- PR: [104](https://github.com/HazyResearch/ThunderKittens/pull/104)
- Decision: `Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.`
- Reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.

## Changed Paths

- `config.py`
- `kernels/torch_scaled/gentests.py`
- `kernels/torch_scaled/harness.impl`
- `kernels/torch_scaled/scaled_matmul`
- `kernels/torch_scaled/scaled_matmul.cu`
- `kernels/torch_scaled/visualize.py`
- `thunderkittens.cpp`

## Evidence Bundle

`evidence/pull-bundles/thunderkittens/gh-104`
