[MPS] Add test consistency from OpInfo based tests from PR 78504 (#79532

) Pull Request resolved: #79532 Approved by: https://github.com/albanD, https://github.com/malfet
pytorch · Jul 4, 2022 · 76cff18 · 76cff18
1 parent b4ed13e
commit 76cff18
Show file tree

Hide file tree

Showing 2 changed files with 633 additions and 16 deletions.
diff --git a/.github/workflows/_mac-test-arm64.yml b/.github/workflows/_mac-test-arm64.yml
@@ -40,7 +40,7 @@ jobs:
           # shellcheck disable=SC1090
           . ~/miniconda3/etc/profile.d/conda.sh
           set -ex
-          conda create -yp "${ENV_NAME}" "python=${PY_VERS}" numpy expecttest
+          conda create -yp "${ENV_NAME}" "python=${PY_VERS}" numpy expecttest pyyaml
           # As wheels are cross-compiled they are reported as x86_64 ones
           ORIG_WHLNAME=$(ls -1 dist/*.whl); ARM_WHLNAME=${ORIG_WHLNAME/x86_64/arm64}; mv ${ORIG_WHLNAME} ${ARM_WHLNAME}
           conda run -p "${ENV_NAME}" python3 -mpip install dist/*.whl