Update vulkan-target-triple for Radeon devices. (huggingface#446)

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>

Author: Shukla-Gaurav

shark/iree_utils/vulkan_utils.py

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