Merge branch 'main' into ean-pytest-bench

nod-ai · Aug 14, 2023 · 29eac5b · 29eac5b
2 parents c8479e4 + 0139dd5
commit 29eac5b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/shark/iree_utils/compile_utils.py b/shark/iree_utils/compile_utils.py
@@ -95,6 +95,7 @@ def get_iree_frontend_args(frontend):
 def get_iree_common_args():
     return [
         "--iree-stream-resource-index-bits=64",
+        "--iree-stream-resource-max-allocation-size=4294967295",
         "--iree-vm-target-index-bits=64",
         "--iree-vm-bytecode-module-strip-source-map=true",
         "--iree-util-zero-fill-elided-attrs",