Skip to content

Commit 00e38ab

Browse files
authored
Add 4080 support
1 parent e3e4ea5 commit 00e38ab

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

shark/iree_utils/vulkan_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,8 @@ def get_vulkan_target_triple(device_name):
7171
triple = f"ampere-rtx3090-{system_os}"
7272
elif all(x in device_name for x in ("RTX", "4090")):
7373
triple = f"ampere-rtx3090-{system_os}"
74+
elif all(x in device_name for x in ("RTX", "4080")):
75+
triple = f"ampere-rtx3090-{system_os}"
7476
elif all(x in device_name for x in ("RTX", "4000")):
7577
triple = f"turing-rtx4000-{system_os}"
7678
elif all(x in device_name for x in ("RTX", "5000")):

0 commit comments

Comments
 (0)