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 vulkan_utils.py

Browse Source
This commit is contained in:
Ean Garvey
2024-01-09 12:39:11 -06:00
committed by GitHub
parent 80906155f8
commit fd3fdeef07
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
shark/iree_utils/vulkan_utils.py
View File

@@ -143,8 +143,12 @@ def get_vulkan_target_triple(device_name):
triple = f"rdna3-7600-{system_os}"
elif "7700" in device_name:
triple = f"rdna3-7700-{system_os}"
elif any(x in device_name for x in "AMD", "Radeon"):
elif any(x in device_name for x in "AMD", "Radeon") and any(x in device_name for x in "6700", "6750"):
triple = f"rdna2-unknown-{system_os}"
elif any(x in device_name for x in "AMD", "Radeon") and "7" in device_name:
triple = f"rdna3-unknown-{system_os}"
elif any(x in device_name for x in "AMD", "Radeon"):
triple = f"rdna3-unknown-{system_os}"
# Intel Targets
elif any(x in device_name for x in ("A770", "A750")):
triple = f"arc-770-{system_os}"