Add window_size_left param ipex rocm

This commit is contained in:
Mohit Sharma 2025-03-14 07:47:45 +00:00
parent eaf18c1ccb
commit b30cdabf68
2 changed files with 2 additions and 0 deletions

View File

@ -78,6 +78,7 @@ def paged_attention(
*,
kv_scales: KVScales,
softcap: Optional[float] = None,
window_size_left: Optional[int] = -1,
):
if softcap is not None:
raise NotImplementedError("softcap is not available in IPEX")

View File

@ -59,6 +59,7 @@ def paged_attention(
*,
kv_scales: KVScales,
softcap: Optional[float] = None,
window_size_left: Optional[int] = -1,
):
# Adapted from: https://github.com/vllm-project/vllm/blob/f8a1e39fae05ca610be8d5a78be9d40f5274e5fc/vllm/model_executor/layers/attention.py
# Copyright 2023 The vLLM team. All rights