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

Fix cases where full URI is given

Browse Source
This commit is contained in:
Ean Garvey
2023-11-17 19:44:49 +00:00
parent 82128672fd
commit d78e86bd48
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
shark/iree_utils/compile_utils.py
View File

@@ -81,7 +81,7 @@ def clean_device_info(raw_device):
)
if "://" in device:
device, device_id = device.split("://")
device_id = int(device_id) # using device index in webui
device_id = int(device_id[0]) # using device index in webui
if device not in ["rocm", "vulkan"]:
device_id = None