We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7ab056c commit cd821eaCopy full SHA for cd821ea
tests/plugins_tests/test_platform_plugins.py
@@ -29,5 +29,5 @@ def test_oot_attention_backend(monkeypatch: pytest.MonkeyPatch):
29
# ignore the backend env variable if it is set
30
with monkeypatch.context() as m:
31
m.setenv(STR_BACKEND_ENV_VAR, STR_INVALID_VAL)
32
- backend = get_attn_backend(16, torch.float16, torch.float16, 16, False)
+ backend = get_attn_backend(16, torch.float16, "auto", 16, False)
33
assert backend.get_name() == "Dummy_Backend"
0 commit comments