diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index 3f95fbc0b6c..67b0931e902 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -185,7 +185,7 @@ test_model_with_coreml() { "${PYTHON_EXECUTABLE}" -m examples.apple.coreml.scripts.export --model_name="${MODEL_NAME}" --compute_precision "${DTYPE}" EXPORTED_MODEL=$(find "." -type f -name "${MODEL_NAME}*.pte" -print -quit) - # TODO: + if [ -n "$EXPORTED_MODEL" ]; then EXPORTED_MODEL_WITH_DTYPE="${EXPORTED_MODEL%.pte}_${DTYPE}.pte" mv "$EXPORTED_MODEL" "$EXPORTED_MODEL_WITH_DTYPE" @@ -200,6 +200,11 @@ test_model_with_coreml() { test_model_with_mps() { "${PYTHON_EXECUTABLE}" -m examples.apple.mps.scripts.mps_example --model_name="${MODEL_NAME}" --use_fp16 EXPORTED_MODEL=$(find "." -type f -name "${MODEL_NAME}*.pte" -print -quit) + + if [ -n "$EXPORTED_MODEL" ]; then + bash ./examples/apple/mps/scripts/build_mps_executor_runner.sh + ./${CMAKE_OUTPUT_DIR}/examples/apple/mps/mps_executor_runner --model_path "${EXPORTED_MODEL}" --bundled_program + fi } if [[ "${BACKEND}" == "portable" ]]; then