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:
Prashant Kumar
2022-10-11 20:53:41 +05:30
committed by GitHub
parent a63755bc24
commit 326827198b
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

@@ -32,7 +32,7 @@ def get_vulkan_triple_flag():
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 any(x in vulkan_device for x in ("Radeon", "RX 5")):
elif all(x in vulkan_device for x in ("Radeon", "RX 5")):
print(
"Found AMD Radeon RX 5000 series device. Using rdna1-5700xt-linux"
)