---
id: pr-quack-33
repo: Dao-AILab/quack
pr: 33
title: '[TORCH_COMPILE] add torch compile support'
author: mayank31398
date: '2025-09-07'
url: https://github.com/Dao-AILab/quack/pull/33
source_category: upstream-code
architectures:
- sm100
tags:
- cross-entropy
- gemm
techniques: []
hardware_features: []
kernel_types:
- gemm
languages: []
captured_at: '2026-05-16'
status: merged
merge_sha: a1dee85ccfaa7b69d21fa38257fbdf1ca2bb6ce6
inclusion_reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.
changed_paths:
- quack/cross_entropy.py
- quack/rmsnorm.py
- quack/softmax.py
- quack/topk.py
- tests/test_cross_entropy.py
- tests/test_rmsnorm.py
- tests/test_softmax.py
- tests/test_topk.py
artifact_dir: evidence/pull-bundles/quack/gh-33
head_ref_oid: 04ca6560ac694fb08416efbce7faa3e60530effc
---
# [TORCH_COMPILE] add torch compile support

- Repository: `Dao-AILab/quack`
- PR: [33](https://github.com/Dao-AILab/quack/pull/33)
- 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

- `quack/cross_entropy.py`
- `quack/rmsnorm.py`
- `quack/softmax.py`
- `quack/topk.py`
- `tests/test_cross_entropy.py`
- `tests/test_rmsnorm.py`
- `tests/test_softmax.py`
- `tests/test_topk.py`

## Evidence Bundle

`evidence/pull-bundles/quack/gh-33`
