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

Update _common.py

Browse Source
This commit is contained in:
powderluv
2022-08-13 13:49:01 -07:00
committed by GitHub
parent 7975087ee2
commit db6e2207ed
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/_common.py
View File

@@ -47,7 +47,7 @@ IREE_DEVICE_MAP = {
}
IREE_TARGET_MAP = {
"cpu": "cpu",
"cpu": "llvm-cpu",
"gpu": "cuda",
"cuda": "cuda",
"vulkan": "vulkan",