github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-25 03:00:12 -04:00
Code Issues Packages Projects Releases Wiki Activity

Don't require vulkan installation to query devices. (#1953)

Browse Source
This commit is contained in:
Ean Garvey
2023-11-09 14:46:44 -06:00
committed by GitHub
parent f41ad87ef6
commit 31005bcf73
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
shark/iree_utils/vulkan_utils.py
View File

@@ -27,8 +27,12 @@ from shark.parser import shark_args
def get_all_vulkan_devices():
from iree.runtime import get_driver
driver = get_driver("vulkan")
device_list_src = driver.query_available_devices()
try:
driver = get_driver("vulkan")
device_list_src = driver.query_available_devices()
except:
device_list_src = {}
return [d["name"] for d in device_list_src]