Update conftest.py

ivy-llc · Aug 16, 2023 · f450d12 · f450d12
1 parent 8ab26fc
commit f450d12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/conftest.py b/conftest.py
@@ -13,7 +13,7 @@
 
 @pytest.fixture(autouse=True)
 def run_around_tests(dev_str, f, compile_graph, fw):
-    if "gpu" in device and fw == "numpy":
+    if "gpu" in dev_str and fw == "numpy":
         # Numpy does not support GPU
         pytest.skip()
     with ivy.utils.backend.ContextManager(fw):