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

[vulkan] Add "radeon" check to the default AMD triple (#604)

Browse Source
This commit is contained in:
Stanley Winata
2022-12-11 00:05:48 +07:00
committed by GitHub
parent 2a59c4f670
commit 57c94f8f80
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/vulkan_utils.py
View File

@@ -74,7 +74,7 @@ def get_vulkan_triple_flag(extra_args=[]):
elif all(x in vulkan_device for x in ("AMD", "7900")):
print(f"Found {vulkan_device} Device. Using rdna3-7900-{system_os}")
return f"-iree-vulkan-target-triple=rdna3-7900-{system_os}"
elif "AMD" in vulkan_device:
elif any(x in vulkan_device for x in ("AMD", "Radeon")):
print(f"Found AMD device. Using rdna2-unknown-{system_os}")
return f"-iree-vulkan-target-triple=rdna2-unknown-{system_os}"
else: