saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79532
Approved by: https://github.com/albanD
This commit is contained in:
Kulin Seth 2022-06-30 01:50:17 +00:00 committed by PyTorch MergeBot
parent f667aaed1d
commit c71886e048
5 changed files with 1288 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.github/workflows/_mac-test-arm64.yml vendored
View file

@ -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

1289
test/test_mps.py
View file

File diff suppressed because it is too large Load diff

2
third_party/ideep vendored

@ -1 +1 @@
Subproject commit 8a114a51c116b55c4ceb689b98746786bd00c29b
Subproject commit 02b17c5748c9349dcc586c359af800c684d9b1ab

2
third_party/kineto vendored

@ -1 +1 @@
Subproject commit 0703c78999061b8329dfab7ec5046fc5764a5573
Subproject commit 6f97e31e0ce40edf9aa3d526558be78ba724d298

2
third_party/onnx vendored

@ -1 +1 @@
Subproject commit f7ee1ac60d06abe8e26c9b6bbe1e3db5286b614b
Subproject commit 96046b8ccfb8e6fa82f6b2b34b3d56add2e8849c