Skip to content

Commit 9dd28f3

Browse files
authored
typo: Use of decode_maybe_q_rope_offset
Following up #855 and #847, this PR fixes a minor typo.
1 parent ea1d0cb commit 9dd28f3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

include/flashinfer/attention/decode.cuh

+1-1
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,7 @@ __global__ void BatchDecodeWithPagedKVCacheKernel(const __grid_constant__ Params
442442
if constexpr (POS_ENCODING_MODE == PosEncodingMode::kRoPELlama) {
443443
const IdType* q_rope_offset = nullptr;
444444
if constexpr (has_decode_maybe_q_rope_offset_v<Params>) {
445-
q_rope_offset = params.maybe_q_rope_offset;
445+
q_rope_offset = params.decode_maybe_q_rope_offset;
446446
}
447447
int32_t q_rope_offset_val = q_rope_offset == nullptr ? (kv_len - 1) : q_rope_offset[batch_idx];
448448
const float rope_rcp_scale = params.rope_rcp_scale;

0 commit comments

Comments
 (0)