From 10a1de2bf0f1ec23bb46b2e4e41a3ffcdc9d94ed Mon Sep 17 00:00:00 2001
From: vmoens <vincentmoens@gmail.com>
Date: Thu, 7 Sep 2023 07:22:20 -0400
Subject: [PATCH] amend

---
 .github/workflows/wheels.yml | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
index fef883d26..765fc8946 100644
--- a/.github/workflows/wheels.yml
+++ b/.github/workflows/wheels.yml
@@ -84,6 +84,38 @@ jobs:
           name: tensordict-batch.whl
           path: dist/*.whl
 
+  build-wheel-mac-m1:
+    runs-on: osx-arm64
+    strategy:
+      matrix:
+        python_version: [["3.8", "3.8"], ["3.9", "3.9"], ["3.10", "3.10.3"], ["3.11", "3.11"]]
+    steps:
+      - name: Setup Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python_version[1] }}
+          architecture: x64
+      - name: Checkout tensordict
+        uses: actions/checkout@v2
+      - name: Install PyTorch RC
+        run: |
+          python3 -mpip install torch --extra-index-url https://download.pytorch.org/whl/cpu
+      - name: Build wheel
+        run: |
+          export CC=clang CXX=clang++
+          python3 -mpip install wheel
+          BUILD_VERSION=0.1.3 python3 setup.py bdist_wheel
+      - name: Upload wheel for the test-wheel job
+        uses: actions/upload-artifact@v2
+        with:
+          name: tensordict-mac-${{ matrix.python_version[0] }}.whl
+          path: dist/tensordict-*.whl
+      - name: Upload wheel for download
+        uses: actions/upload-artifact@v2
+        with:
+          name: tensordict-batch.whl
+          path: dist/*.whl
+
   build-wheel-windows:
     runs-on: windows-latest
     strategy: