Skip to content

Commit c4444ff

Browse files
authored
[vulkan][utils] Add rdna3 detection (huggingface#565)
1 parent 27b34f3 commit c4444ff

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

shark/iree_utils/vulkan_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,9 @@ def get_vulkan_triple_flag(extra_args=[]):
5252
elif all(x in vulkan_device for x in ("RTX", "4090")):
5353
print(f"Found {vulkan_device} Device. Using ampere-rtx3090-linux")
5454
return "-iree-vulkan-target-triple=ampere-rtx3090-linux"
55+
elif all(x in vulkan_device for x in ("AMD", "7900")):
56+
print(f"Found {vulkan_device} Device. Using rdna3-7900-linux")
57+
return "-iree-vulkan-target-triple=rdna3-7900-linux"
5558
elif "AMD" in vulkan_device:
5659
print("Found AMD device. Using rdna2-unknown-linux")
5760
return "-iree-vulkan-target-triple=rdna2-unknown-linux"

0 commit comments

Comments
 (0)