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

Add 7800xt to target triples explicitly. (#1928)

Browse Source
This commit is contained in:
Ean Garvey
2023-11-01 17:11:45 -05:00
committed by GitHub
parent 71846344a2
commit 5001db3415
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
shark/iree_utils/vulkan_utils.py
View File

@@ -68,6 +68,8 @@ def get_vulkan_target_triple(device_name):
Returns:
str or None: target triple or None if no match found for given name
"""
# TODO: Replace this with a dict or something smarter.
system_os = get_os_name()
# Apple Targets
if all(x in device_name for x in ("Apple", "M1")):
@@ -117,6 +119,8 @@ def get_vulkan_target_triple(device_name):
# Amd Targets
# Linux: Radeon RX 7900 XTX
# Windows: AMD Radeon RX 7900 XTX
elif all(x in device_name for x in ("RX", "7800")):
triple = f"rdna3-7800-{system_os}"
elif all(x in device_name for x in ("RX", "7900")):
triple = f"rdna3-7900-{system_os}"
elif all(x in device_name for x in ("Radeon", "780M")):