composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00

Files

Jeff Huang 6e558658ea [rocm-libraries] ROCm/rocm-libraries#4999 (commit 45f6624)

[CK] Fix 32-bit overflow in batch prefill kernel for >4GB KV
 cache (#4999)

Use SRD rebasing for page_block_size >= kN0: move SRD base pointer to
page start via 48-bit arithmetic, encode only within-page offset in
voffset. Original code path preserved for ps1/ps16 via constexpr-if.

## Motivation

<!-- Explain the purpose of this PR and the goals it aims to achieve.
-->

## Technical Details

<!-- Explain the changes along with any relevant GitHub links. -->

## Test Plan

<!-- Explain any relevant testing done to verify this PR. -->

## Test Result

<!-- Briefly summarize test outcomes. -->

## Submission Checklist

- [ ] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

2026-03-05 01:09:12 +00:00

fmha_batch_prefill_kernel.hpp

[rocm-libraries] ROCm/rocm-libraries#4999 (commit 45f6624)

2026-03-05 01:09:12 +00:00

fmha_bwd_kernel.hpp

[rocm-libraries] ROCm/rocm-libraries#4577 (commit a36922c)

2026-03-04 01:21:07 +00:00

fmha_fwd_appendkv_kernel.hpp

chore(copyright): update copyright header for include directory (#3293 )

2025-11-26 11:00:05 -07:00

fmha_fwd_appendkv_tile_partitioner.hpp

chore(copyright): update copyright header for include directory (#3293 )