Add target triple support for TITAN RTX (nod-ai#756)

zjgarvey · Jan 4, 2023 · 017dcab · 017dcab
1 parent e60b456
commit 017dcab
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/shark/iree_utils/vulkan_utils.py b/shark/iree_utils/vulkan_utils.py
@@ -78,6 +78,8 @@ def get_vulkan_target_triple(device_name):
         triple = f"turing-rtx6000-{system_os}"
     elif all(x in device_name for x in ("RTX", "8000")):
         triple = f"turing-rtx8000-{system_os}"
+    elif all(x in device_name for x in ("TITAN", "RTX")):
+        triple = f"turing-titanrtx-{system_os}"
     elif all(x in device_name for x in ("GTX", "1060")):
         triple = f"pascal-gtx1060-{system_os}"
     elif all(x in device_name for x in ("GTX", "1070")):