m1

pytorch · Oct 9, 2023 · 4cbf222 · 4cbf222
1 parent fad0f7a
commit 4cbf222
Showing 1 changed file with 27 additions and 1 deletion.
diff --git a/.github/workflows/test-macos.yml b/.github/workflows/test-macos.yml
@@ -19,7 +19,7 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  tests:
+  tests-intel:
     strategy:
       matrix:
         python_version: ["3.8", "3.11"]
@@ -43,3 +43,29 @@ jobs:
         bash .circleci/unittest/linux/scripts/install.sh
         bash .circleci/unittest/linux/scripts/run_test.sh
         bash .circleci/unittest/linux/scripts/post_process.sh
+
+  tests-silicon:
+    strategy:
+      matrix:
+        python_version: ["3.8"]
+      fail-fast: false
+    uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
+    runs-on: macos-m1-12
+    with:
+      repository: pytorch/rl
+      timeout: 120
+      script: |
+        # Set env vars from matrix
+        export PYTHON_VERSION=${{ matrix.python_version }}
+        export CU_VERSION="cpu"
+        export SYSTEM_VERSION_COMPAT=0
+        export TORCH_VERSION=nightly
+
+        echo "PYTHON_VERSION: $PYTHON_VERSION"
+        echo "CU_VERSION: $CU_VERSION"
+
+        ## setup_env.sh
+        bash .circleci/unittest/linux/scripts/setup_env.sh
+        bash .circleci/unittest/linux/scripts/install.sh
+        bash .circleci/unittest/linux/scripts/run_test.sh
+        bash .circleci/unittest/linux/scripts/post_process.sh