SIGN IN SIGN UP

[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.
J
Jeff Huang committed
c57d503c651c9a67d17cab9b5b9c7d8eda2f1439
Parent: f5cee1b
Committed by GitHub <noreply@github.com> on 3/5/2026, 1:08:01 AM