Skip to content
This repository was archived by the owner on Oct 11, 2024. It is now read-only.

Commit 9769731

Browse files
pcmoritzalexm-redhat
authored andcommitted
[Minor] Fix test_cache.py CI test failure (vllm-project#2684)
1 parent 4e1c8d2 commit 9769731

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/kernels/test_cache.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -200,12 +200,12 @@ def test_swap_blocks(
200200

201201
# Create the KV caches on the first device.
202202
src_key_caches, src_value_caches = kv_cache_factory(
203-
num_blocks, block_size, 1, num_heads, head_size, dtype, seed,
203+
num_blocks, block_size, 1, num_heads, head_size, dtype, None, seed,
204204
src_device)
205205

206206
# Create the KV caches on the second device.
207207
dist_key_caches, dist_value_caches = kv_cache_factory(
208-
num_blocks, block_size, 1, num_heads, head_size, dtype, seed,
208+
num_blocks, block_size, 1, num_heads, head_size, dtype, None, seed,
209209
dst_device)
210210

211211
src_key_caches_clone = src_key_caches[0].clone()

0 commit comments

Comments
 (0)