Skip to content

Commit 10630ab

Browse files
Add config stanza for NVIDIA RTX 2080. (huggingface#658)
Just happened to have this card on my Windows machine and verified that the SD demo works on it. ``` Average step time: 144.26142692565918ms/it Clip Inference Avg time (ms) = (205.001 + 44.000) / 2 = 124.501 VAE Inference time (ms): 281.001 Total image generation time: 7.856997728347778sec ``` I'd love to add an API upstream to derive compiler tuning flags from a host device.
1 parent 2bc6de6 commit 10630ab

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

shark/iree_utils/vulkan_utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,11 @@ def get_vulkan_triple_flag(extra_args=[]):
5858
print("Found Apple M2 Device. Using m1-moltenvk-macos")
5959
return "-iree-vulkan-target-triple=m1-moltenvk-macos"
6060
# Nvidia Targets
61+
elif all(x in vulkan_device for x in ("RTX", "2080")):
62+
print(
63+
f"Found {vulkan_device} Device. Using turing-rtx2080-{system_os}"
64+
)
65+
return f"-iree-vulkan-target-triple=turing-rtx2080-{system_os}"
6166
elif all(x in vulkan_device for x in ("A100", "SXM4")):
6267
print(
6368
f"Found {vulkan_device} Device. Using ampere-rtx3080-{system_os}"

0 commit comments

Comments
 (0)