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:17:37 -06:00
committed by GitHub
parent 94805aa3c6
commit 1c094d96eb
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

@@ -139,7 +139,11 @@ def get_vulkan_target_triple(device_name):
triple = f"rdna3-780m-{system_os}"
elif all(x in device_name for x in ("AMD", "PRO", "W7900")):
triple = f"rdna3-w7900-{system_os}"
elif any(x in device_name for x in ("AMD", "Radeon")):
elif "7600" in 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"):
triple = f"rdna2-unknown-{system_os}"
# Intel Targets
elif any(x in device_name for x in ("A770", "A750")):