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

Added Adreno target triples to vulkan_utils (#1543)

Browse Source
This commit is contained in:
Surya Jasper
2023-06-15 16:42:59 -07:00
committed by GitHub
parent 38570a9bbb
commit b203779462
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
shark/iree_utils/vulkan_target_env_utils.py
View File

@@ -136,7 +136,7 @@ def get_vendor(triple):
return "Intel"
if arch in ["turing", "ampere", "pascal"]:
return "NVIDIA"
if arch == "ardeno":
if arch == "adreno":
return "Qualcomm"
if arch == "cpu":
if product == "swiftshader":

5
shark/iree_utils/vulkan_utils.py
View File

@@ -114,6 +114,11 @@ def get_vulkan_target_triple(device_name):
# Intel Targets
elif any(x in device_name for x in ("A770", "A750")):
triple = f"arc-770-{system_os}"
# Adreno Targets
elif all(x in device_name for x in ("Adreno", "740")):
triple = f"adreno-a740-{system_os}"
else:
triple = None
return triple