Skip to content

Commit cd821ea

Browse files
authored
[CI] fix kv_cache_type argument (#18594)
Signed-off-by: Andy Xie <andy.xning@gmail.com>
1 parent 7ab056c commit cd821ea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/plugins_tests/test_platform_plugins.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,5 +29,5 @@ def test_oot_attention_backend(monkeypatch: pytest.MonkeyPatch):
2929
# ignore the backend env variable if it is set
3030
with monkeypatch.context() as m:
3131
m.setenv(STR_BACKEND_ENV_VAR, STR_INVALID_VAL)
32-
backend = get_attn_backend(16, torch.float16, torch.float16, 16, False)
32+
backend = get_attn_backend(16, torch.float16, "auto", 16, False)
3333
assert backend.get_name() == "Dummy_Backend"

0 commit comments

Comments
 (0)