Unverified Commit cd821ea5 authored by Ning Xie's avatar Ning Xie Committed by GitHub
Browse files

[CI] fix kv_cache_type argument (#18594)


Signed-off-by: default avatarAndy Xie <andy.xning@gmail.com>
parent 7ab056c2
......@@ -29,5 +29,5 @@ def test_oot_attention_backend(monkeypatch: pytest.MonkeyPatch):
# ignore the backend env variable if it is set
with monkeypatch.context() as m:
m.setenv(STR_BACKEND_ENV_VAR, STR_INVALID_VAL)
backend = get_attn_backend(16, torch.float16, torch.float16, 16, False)
backend = get_attn_backend(16, torch.float16, "auto", 16, False)
assert backend.get_name() == "Dummy_Backend"
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment