We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 27b34f3 commit c4444ffCopy full SHA for c4444ff
shark/iree_utils/vulkan_utils.py
@@ -52,6 +52,9 @@ def get_vulkan_triple_flag(extra_args=[]):
52
elif all(x in vulkan_device for x in ("RTX", "4090")):
53
print(f"Found {vulkan_device} Device. Using ampere-rtx3090-linux")
54
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"
58
elif "AMD" in vulkan_device:
59
print("Found AMD device. Using rdna2-unknown-linux")
60
return "-iree-vulkan-target-triple=rdna2-unknown-linux"
0 commit comments