---
id: pr-flashinfer-778
repo: flashinfer-ai/flashinfer
pr: 778
title: 'feat: Separate QK/VO head dim dispatch for sm90 AOT'
author: abcdabcd987
date: '2025-02-04'
url: https://github.com/flashinfer-ai/flashinfer/pull/778
source_category: upstream-code
architectures:
- sm100
- sm90
tags:
- prefill
techniques: []
hardware_features: []
kernel_types:
- prefill
languages:
- python
captured_at: '2026-04-17'
status: merged
merge_sha: 1ebbde36b1c10f2eba84cd84fb923ef2b906065d
inclusion_reason: kernel file changes
changed_paths:
- aot_build_utils/generate.py
- aot_build_utils/generate_dispatch_inc.py
- aot_build_utils/generate_sm90.py
- csrc/aot_extension_utils.h
- csrc/batch_prefill_sm90_config.inc
- csrc/pytorch_extension_utils.h
- csrc/single_prefill_sm90_config.inc
- setup.py
head_ref_oid: 4359ad2aa56961387e7086aff85a6b1f3b901d84
artifact_dir: evidence/pull-bundles/flashinfer/gh-778
---
# feat: Separate QK/VO head dim dispatch for sm90 AOT

- Repository: `flashinfer-ai/flashinfer`
- PR: [778](https://github.com/flashinfer-ai/flashinfer/pull/778)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `aot_build_utils/generate.py`
- `aot_build_utils/generate_dispatch_inc.py`
- `aot_build_utils/generate_sm90.py`
- `csrc/aot_extension_utils.h`
- `csrc/batch_prefill_sm90_config.inc`
- `csrc/pytorch_extension_utils.h`
- `csrc/single_prefill_sm90_config.inc`
- `setup.py`

## Evidence Bundle

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