✨ Introduce Apple MPS backend support and testing

Author: jathu

.ci/scripts/test_model.sh

@@ -246,6 +246,15 @@ 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 [ -z "$EXPORTED_MODEL" ]; then
+    echo "[error] failed to export model: no .pte file found"
+    exit 1
+  fi
+
+  echo "Testing exported model with mps_executor_runner..."
+  ./examples/apple/mps/scripts/build_mps_executor_runner.sh
+  ./cmake-out/examples/apple/mps/mps_executor_runner --model_path "${EXPORTED_MODEL}"
 }
 
 if [[ "${BACKEND}" == "portable" ]]; then

.ci/scripts/wheel/test_macos.py

@@ -11,13 +11,17 @@
 if __name__ == "__main__":
     test_base.run_tests(
         model_tests=[
+            # test_base.ModelTest(
+            #     model=Model.Mv3,
+            #     backend=Backend.XnnpackQuantizationDelegation,
+            # ),
+            # test_base.ModelTest(
+            #     model=Model.Mv3,
+            #     backend=Backend.CoreMlExportAndTest,
+            # ),
             test_base.ModelTest(
                 model=Model.Mv3,
-                backend=Backend.XnnpackQuantizationDelegation,
-            ),
-            test_base.ModelTest(
-                model=Model.Mv3,
-                backend=Backend.CoreMlExportAndTest,
+                backend=Backend.AppleMPS,
             ),
         ]
     )

examples/models/__init__.py

@@ -46,6 +46,7 @@ class Backend(str, Enum):
     XnnpackQuantizationDelegation = "xnnpack-quantization-delegation"
     CoreMlExportOnly = "coreml"
     CoreMlExportAndTest = "coreml-test"  # AOT export + test with runner
+    AppleMPS = "mps"
 
     def __str__(self) -> str:
         return self.value