Merge branch 'main' into dependabot/pip/dataset/gradio-4.19.2

nod-ai · Mar 28, 2024 · 348ed78 · 348ed78
2 parents 9ad78eb + 58f194a
commit 348ed78
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/shark/iree_utils/_common.py b/shark/iree_utils/_common.py
@@ -91,7 +91,7 @@ def iree_target_map(device):
     "cpu-task": "llvm-cpu",
     "cpu-sync": "llvm-cpu",
     "cuda": "cuda",
-    "vulkan": "vulkan",
+    "vulkan": "vulkan-spirv",
     "metal": "metal",
     "rocm": "rocm",
     "intel-gpu": "opencl-spirv",
@@ -122,9 +122,7 @@ def check_device_drivers(device):
         )
         return True
     except RuntimeError as re:
-        print(
-            f"[ERR] Failed to get driver for {device} with error:\n{repr(re)}"
-        )
+        print(f"[ERR] Failed to get driver for {device} with error:\n{repr(re)}")
         return True
 
     # Unknown device. We assume drivers are installed.