github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Drop RDNA1 for now

Browse Source
This commit is contained in:
powderluv
2022-10-29 14:29:09 -07:00
committed by GitHub
parent f8dc996004
commit fd89b06641
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
shark/iree_utils/vulkan_utils.py
View File

@@ -48,12 +48,7 @@ def get_vulkan_triple_flag(extra_args=[]):
elif all(x in vulkan_device for x in ("RTX", "3090")):
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 ("Radeon", "RX 5")):
print(
"Found AMD Radeon RX 5000 series device. Using rdna1-5700xt-linux"
)
return "-iree-vulkan-target-triple=rdna1-5700xt-linux"
elif any(x in vulkan_device for x in ("Radeon", "RX 6")):
elif any(x in vulkan_device for x in ("Radeon", "AMD")):
print(
"Found AMD Radeon RX 6000 series device. Using rdna2-unknown-linux"
)