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

In get_iree_runtime_config get the specific device instead of the default

Browse Source
This commit is contained in:
Boian Petkantchin
2022-12-11 10:19:38 -08:00
committed by Boian Petkantchin
parent aaf60bdee6
commit bc17c29b2e
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

@@ -367,5 +367,5 @@ def get_results(compiled_vm, input, config, frontend="torch"):
def get_iree_runtime_config(device):
device = iree_device_map(device)
config = ireert.Config(driver_name=device.split("://")[0])
config = ireert.Config(device=ireert.get_device(device))
return config