diff --git a/.github/workflows/run-readme-pr-mps.yml b/.github/workflows/run-readme-pr-mps.yml index 3e90265f5..ac7e3b6e4 100644 --- a/.github/workflows/run-readme-pr-mps.yml +++ b/.github/workflows/run-readme-pr-mps.yml @@ -10,7 +10,7 @@ jobs: uses: pytorch/test-infra/.github/workflows/macos_job.yml@main with: runner: macos-m1-14 - timeout-minutes: 50 + timeout: 120 script: | conda create -y -n test-readme-mps-macos python=3.10.11 llvm-openmp conda activate test-readme-mps-macos @@ -63,7 +63,7 @@ jobs: test-gguf-mps-macos: uses: pytorch/test-infra/.github/workflows/macos_job.yml@main with: - runner: macos-m1-14 # neeps MPS, was macos-m1-stable + runner: macos-m1-14 # needs MPS, was macos-m1-stable script: | set -x conda create -y -n test-quantization-mps-macos python=3.10.11 @@ -90,7 +90,7 @@ jobs: test-advanced-mps-macos: uses: pytorch/test-infra/.github/workflows/macos_job.yml@main with: - runner: macos-m1-14 # neeps MPS, was macos-m1-stable + runner: macos-m1-14 # needs MPS, was macos-m1-stable script: | set -x conda create -y -n test-quantization-mps-macos python=3.10.11 @@ -113,3 +113,84 @@ jobs: echo "tests complete" echo "*******************************************" echo "::endgroup::" + + test-evaluation-mps-macos: + uses: pytorch/test-infra/.github/workflows/macos_job.yml@main + with: + runner: macos-m1-14 # needs MPS, was macos-m1-stable + script: | + set -x + conda create -y -n test-evaluation-mps-macos python=3.10.11 + conda activate test-evaluation-mps-macos + # NS: Remove previous installation of torch first + # as this script does not isntall anything into conda env + # but rather system dep + pip3 uninstall -y torch || true + set -eou pipefail + + echo "::group::Print machine info" + uname -a + sysctl machdep.cpu.brand_string + sysctl machdep.cpu.core_count + echo "::endgroup::" + + .ci/scripts/run-docs evaluation + + echo "::group::Completion" + echo "tests complete" + echo "*******************************************" + echo "::endgroup::" + + test-multimodal-mps-macos: + uses: pytorch/test-infra/.github/workflows/macos_job.yml@main + with: + runner: macos-m1-14 # needs MPS, was macos-m1-stable + script: | + set -x + conda create -y -n test-multimodal-mps-macos python=3.10.11 + conda activate test-multimodal-mps-macos + # NS: Remove previous installation of torch first + # as this script does not isntall anything into conda env + # but rather system dep + pip3 uninstall -y torch || true + set -eou pipefail + + echo "::group::Print machine info" + uname -a + sysctl machdep.cpu.brand_string + sysctl machdep.cpu.core_count + echo "::endgroup::" + + .ci/scripts/run-docs multimodal + + echo "::group::Completion" + echo "tests complete" + echo "*******************************************" + echo "::endgroup::" + + test-native-mps-macos: + uses: pytorch/test-infra/.github/workflows/macos_job.yml@main + with: + runner: macos-m1-14 # needs MPS, was macos-m1-stable + script: | + set -x + conda create -y -n test-native-mps-macos python=3.10.11 + conda activate test-native-mps-macos + # NS: Remove previous installation of torch first + # as this script does not isntall anything into conda env + # but rather system dep + pip3 uninstall -y torch || true + set -eou pipefail + + echo "::group::Print machine info" + uname -a + sysctl machdep.cpu.brand_string + sysctl machdep.cpu.core_count + echo "::endgroup::" + + .ci/scripts/run-docs native + + echo "::group::Completion" + echo "tests complete" + echo "*******************************************" + echo "::endgroup::"