---
id: pr-flashinfer-776
repo: flashinfer-ai/flashinfer
pr: 776
title: 'perf: refactor fa2 prefill template'
author: yzh119
date: '2025-02-04'
url: https://github.com/flashinfer-ai/flashinfer/pull/776
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- prefill
techniques: []
hardware_features: []
kernel_types:
- attention
- prefill
languages:
- cuda-cpp
captured_at: '2026-04-17'
status: merged
merge_sha: fc037728a9abaafdc0cedb1559c8844e4b59b264
inclusion_reason: kernel file changes
changed_paths:
- aot_build_utils/generate_batch_paged_prefill_inst.py
- aot_build_utils/generate_batch_ragged_prefill_inst.py
- csrc/batch_prefill_paged_kernel_inst.jinja
- csrc/batch_prefill_ragged_kernel_inst.jinja
- include/flashinfer/attention/prefill.cuh
- include/flashinfer/attention/scheduler.cuh
- include/flashinfer/mma.cuh
- include/flashinfer/utils.cuh
- tests/test_jit_example.py
head_ref_oid: 4ff0c5110789a1430b342c830d78af439cad4b82
artifact_dir: evidence/pull-bundles/flashinfer/gh-776
---
# perf: refactor fa2 prefill template

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

## Changed Paths

- `aot_build_utils/generate_batch_paged_prefill_inst.py`
- `aot_build_utils/generate_batch_ragged_prefill_inst.py`
- `csrc/batch_prefill_paged_kernel_inst.jinja`
- `csrc/batch_prefill_ragged_kernel_inst.jinja`
- `include/flashinfer/attention/prefill.cuh`
- `include/flashinfer/attention/scheduler.cuh`
- `include/flashinfer/mma.cuh`
- `include/flashinfer/utils.cuh`
- `tests/test_jit_example.py`

## Evidence Bundle

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