Skip to content

Commit

Permalink
Fix _IREE_TARGET_MAP (#2103)
Browse files Browse the repository at this point in the history
- Change target passed to iree for vulkan from 'vulkan'
to 'vulkan-spriv', as 'vulkan' is not a valid value for
--iree-hal-target-backends with the current iree compiler.
  • Loading branch information
one-lithe-rune authored Mar 18, 2024
1 parent c5cf005 commit 58f194a
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions shark/iree_utils/_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit 58f194a

Please sign in to comment.