Adding the conda builds for m1 (#1794) (#1795)

* Adding the conda builds for m1

* Fix alignment

* Don't use mkl on arm64

* prettier

Author: atalman

.github/workflows/build-m1-binaries.yml

@@ -96,3 +96,75 @@ jobs:
           for pkg in dist/*; do
             aws s3 cp "$pkg" "s3://pytorch/whl/${CHANNEL}/cpu/" --acl public-read
           done
+  build_conda:
+    name: "Build TorchText M1 conda packages"
+    runs-on: macos-m1
+    strategy:
+      matrix:
+        py_vers: ["3.8", "3.9", "3.10"]
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v2
+      - name: Set CHANNEL (only for tagged pushes)
+        if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') }}
+        run: |
+          # reference ends with an RC suffix
+          if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then
+            echo "CHANNEL=test" >> "$GITHUB_ENV"
+          fi
+      - name: Set CHANNEL Release (for release)
+        if:
+          ${{ (github.event_name == 'pull_request' && startsWith(github.base_ref, 'release')) || startsWith(github.ref,
+          'refs/heads/release') }}
+        run: |
+          echo "CHANNEL=test" >> "$GITHUB_ENV"
+      - name: Install conda-build and purge previous artifacts
+        shell: arch -arch arm64 bash {0}
+        run: |
+          . ~/miniconda3/etc/profile.d/conda.sh
+          conda install -yq conda-build
+          conda build purge-all
+
+      - name: Build TorchText M1 conda package
+        shell: arch -arch arm64 bash {0}
+        env:
+          ENV_NAME: conda-env-${{ github.run_id }}
+          PYTHON_VERSION: ${{ matrix.py_vers }}
+          CU_VERSION: cpu
+        run: |
+          . ~/miniconda3/etc/profile.d/conda.sh
+          set -ex
+          . packaging/pkg_helpers.bash
+
+          if [[ $CHANNEL == "test" ]]; then
+            setup_base_build_version
+            export CONDA_CHANNEL_FLAGS="-c pytorch-test"
+          else
+            setup_build_version
+          fi
+          export BUILD_TYPE="conda"
+          export NO_CUDA_PACKAGE=1
+          export SOURCE_ROOT_DIR="$PWD"
+
+          setup_conda_pytorch_constraint
+          export SOURCE_ROOT_DIR=$(pwd)
+          conda build -c defaults $CONDA_CHANNEL_FLAGS --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchtext
+          mkdir -p dist
+          cp ~/miniconda3/conda-bld/osx-arm64/*.tar.bz2 dist/
+      - name: Upload package to GitHub
+        uses: actions/upload-artifact@v3
+        with:
+          name: torchtext-py${{ matrix.py_vers }}-macos11-m1-conda
+          path: dist/
+      - name: Upload package to conda
+        if:
+          ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || startsWith(github.event.ref,
+          'refs/tags/')) }}
+        shell: arch -arch arm64 bash {0}
+        env:
+          CONDA_PYTORCHBOT_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
+        run: |
+          . ~/miniconda3/etc/profile.d/conda.sh
+          conda install -yq anaconda-client
+          set -x
+          anaconda  -t "${CONDA_PYTORCHBOT_TOKEN}" upload ~/miniconda3/conda-bld/osx-arm64/*.tar.bz2 -u "pytorch-${CHANNEL}" --label main --no-progress --force

packaging/pkg_helpers.bash

@@ -214,8 +214,11 @@ setup_conda_pytorch_constraint() {
   fi
   # TODO: Remove me later, see https://github.com/pytorch/pytorch/issues/62424 for more details
   if [[ "$(uname)" == Darwin ]]; then
-    # Use less than equal to avoid version conflict in python=3.6 environment
-    export CONDA_EXTRA_BUILD_CONSTRAINT="- mkl<=2021.2.0"
+    arch_name="$(uname -m)"
+    if [ "${arch_name}" != "arm64" ]; then
+      # Use less than equal to avoid version conflict in python=3.6 environment
+      export CONDA_EXTRA_BUILD_CONSTRAINT="- mkl<=2021.2.0"
+    fi
   fi
 }