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][utils] Add rdna3 detection (#565)

Browse Source
This commit is contained in:
Stanley Winata
2022-12-06 14:56:06 +07:00
committed by GitHub
parent 27b34f3929
commit c4444ff695
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
shark/iree_utils/vulkan_utils.py
View File

@@ -52,6 +52,9 @@ def get_vulkan_triple_flag(extra_args=[]):
elif all(x in vulkan_device for x in ("RTX", "4090")):
print(f"Found {vulkan_device} Device. Using ampere-rtx3090-linux")
return "-iree-vulkan-target-triple=ampere-rtx3090-linux"
elif all(x in vulkan_device for x in ("AMD", "7900")):
print(f"Found {vulkan_device} Device. Using rdna3-7900-linux")
return "-iree-vulkan-target-triple=rdna3-7900-linux"
elif "AMD" in vulkan_device:
print("Found AMD device. Using rdna2-unknown-linux")
return "-iree-vulkan-target-triple=rdna2-unknown-linux"