Fix device parsing.

nod-ai · May 23, 2024 · 7be8803 · 7be8803
1 parent 4562b79
commit 7be8803
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/shark/iree_utils/compile_utils.py b/shark/iree_utils/compile_utils.py
@@ -70,8 +70,8 @@ def get_iree_device_args(device, extra_args=[]):
 def get_iree_target_triple(device):
     args = get_iree_device_args(device)
     for flag in args:
-        if "triple" in flag.split("-"):
-            triple = flag.split("=")
+        if "triple" in flag:
+            triple = flag.split("=")[-1]
             return triple
     return ""