Merge pull request #1204 from atalman/fix_arm64_build

Fix arm64 builds by building using source
pytorch · Oct 3, 2023 · b724a2a · b724a2a
2 parents 7c7597b + 47144d9
commit b724a2a
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/.github/workflows/_build_test_upload.yml b/.github/workflows/_build_test_upload.yml
@@ -170,8 +170,10 @@ jobs:
           fi
           if ${{ startsWith( matrix.os, 'macos' ) }}; then
             conda activate wheel_build_env
+            conda run -n wheel_build_env packaging/build_wheel.sh
+          else
+            packaging/build_wheel.sh
           fi
-          packaging/build_wheel.sh
       - name: Validate TorchData Wheel
         shell: bash -l {0}
         env:
@@ -213,8 +215,10 @@ jobs:
           fi
           if ${{ startsWith( matrix.os, 'macos' ) }}; then
             conda activate wheel_build_env
+            conda run -n wheel_build_env pip3 install dist/torchdata*.whl
+          else
+            pip3 install dist/torchdata*.whl
           fi
-          pip3 install dist/torchdata*.whl
       - name: Run Smoke Tests
         shell: bash -l {0}
         env:
@@ -228,11 +232,13 @@ jobs:
           fi
           if ${{ startsWith( matrix.os, 'macos' ) }}; then
             conda activate wheel_build_env
-          fi
-          if ${{ matrix.python-version == 'pure' }}; then
-            python test/smoke_test/smoke_test.py --no-s3
+            conda run -n wheel_build_env python test/smoke_test/smoke_test.py
           else
-            python test/smoke_test/smoke_test.py
+            if ${{ matrix.python-version == 'pure' }}; then
+              python test/smoke_test/smoke_test.py --no-s3
+            else
+              python test/smoke_test/smoke_test.py
+            fi
           fi
       - name: Upload Wheels to Github
         if: always()