Stringify path for ireec.compile_file (#1901)

* Stringify path for ireec.compile_file * Update test-models.yml
nod-ai · Oct 18, 2023 · 4797bb8 · 4797bb8
1 parent 205e576
commit 4797bb8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/test-models.yml b/.github/workflows/test-models.yml
@@ -137,7 +137,8 @@ jobs:
         source shark.venv/bin/activate
         echo $PATH
         pip list | grep -E "torch|iree"
-        pytest --ci --ci_sha=${SHORT_SHA} --local_tank_cache="/Volumes/builder/anush/shark_cache" --tank_url="gs://shark_tank/nightly/" -k metal
+      # disabled due to a low-visibility memory issue with pytest on macos.
+      # pytest --ci --ci_sha=${SHORT_SHA} --local_tank_cache="/Volumes/builder/anush/shark_cache" --tank_url="gs://shark_tank/nightly/" -k metal
 
     - name: Validate Vulkan Models (a100)
       if: matrix.suite == 'vulkan' && matrix.os == 'a100'

diff --git a/shark/iree_utils/compile_utils.py b/shark/iree_utils/compile_utils.py
@@ -324,7 +324,7 @@ def compile_module_to_flatbuffer(
     else:
         assert os.path.isfile(module)
         flatbuffer_blob = ireec.compile_file(
-            module,
+            str(module),
             input_type=input_type,
             target_backends=[iree_target_map(device)],
             extra_args=args,