Fix vulkan target env extensions.

nod-ai · Sep 11, 2023 · 399f882 · 399f882
1 parent e9e8f0e
commit 399f882
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/shark/iree_utils/vulkan_target_env_utils.py b/shark/iree_utils/vulkan_target_env_utils.py
@@ -57,7 +57,8 @@ def get_version(triple):
 @functools.cache
 def get_extensions(triple):
     def make_ext_list(ext_list):
-        return ", ".join(ext_list)
+        res = ", ".join(ext_list)
+        return f"[{res}]"
 
     arch, product, os = triple
     if arch == "m1":