Update vulkan_utils.py

2026-01-11 14:58:11 -05:00 · 2022-10-11 20:53:41 +05:30
1 changed files with 1 additions and 1 deletions
--- a/shark/iree_utils/vulkan_utils.py
+++ b/shark/iree_utils/vulkan_utils.py
@@ -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"
        )