diff --git a/.github/workflows/test-models.yml b/.github/workflows/test-models.yml index 07f5c91985..703bbc0071 100644 --- 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 index 2ce02118a6..e86546c478 100644 --- 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,