From 4797bb89f5c5dbc7e40168e7dcb3c5cfa0e88235 Mon Sep 17 00:00:00 2001 From: Ean Garvey <87458719+monorimet@users.noreply.github.com> Date: Wed, 18 Oct 2023 12:59:23 -0700 Subject: [PATCH] Stringify path for ireec.compile_file (#1901) * Stringify path for ireec.compile_file * Update test-models.yml --- .github/workflows/test-models.yml | 3 ++- shark/iree_utils/compile_utils.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test-models.yml b/.github/workflows/test-models.yml index 07f5c919..703bbc00 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 2ce02118..e86546c4 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,