File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -185,7 +185,7 @@ test_model_with_coreml() {
185185
186186 " ${PYTHON_EXECUTABLE} " -m examples.apple.coreml.scripts.export --model_name=" ${MODEL_NAME} " --compute_precision " ${DTYPE} "
187187 EXPORTED_MODEL=$( find " ." -type f -name " ${MODEL_NAME} *.pte" -print -quit)
188- # TODO:
188+
189189 if [ -n " $EXPORTED_MODEL " ]; then
190190 EXPORTED_MODEL_WITH_DTYPE=" ${EXPORTED_MODEL% .pte} _${DTYPE} .pte"
191191 mv " $EXPORTED_MODEL " " $EXPORTED_MODEL_WITH_DTYPE "
@@ -200,6 +200,11 @@ test_model_with_coreml() {
200200test_model_with_mps () {
201201 " ${PYTHON_EXECUTABLE} " -m examples.apple.mps.scripts.mps_example --model_name=" ${MODEL_NAME} " --use_fp16
202202 EXPORTED_MODEL=$( find " ." -type f -name " ${MODEL_NAME} *.pte" -print -quit)
203+
204+ if [ -n " $EXPORTED_MODEL " ]; then
205+ bash ./examples/apple/mps/scripts/build_mps_executor_runner.sh
206+ ./${CMAKE_OUTPUT_DIR} /examples/apple/mps/mps_executor_runner --model_path " ${EXPORTED_MODEL} " --bundled_program
207+ fi
203208}
204209
205210if [[ " ${BACKEND} " == " portable" ]]; then
You can’t perform that action at this time.
0 commit comments