Fix building pre-compiled wheels on Linux and Windows (#444)

k2-fsa · Jul 26, 2023 · eb0cb9d · eb0cb9d
1 parent 1a5d9f3
commit eb0cb9d
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 2 deletions.
diff --git a/.github/workflows/macos-cpu-wheels.yml b/.github/workflows/macos-cpu-wheels.yml
@@ -13,7 +13,7 @@ concurrency:
 jobs:
   generate_build_matrix:
     # see https://github.com/pytorch/pytorch/pull/50633
-    runs-on: ubuntu-latest
+    runs-on: macos-latest
     outputs:
       matrix: ${{ steps.set-matrix.outputs.matrix }}
     steps:

diff --git a/.github/workflows/ubuntu-cpu-wheels.yml b/.github/workflows/ubuntu-cpu-wheels.yml
@@ -42,6 +42,14 @@ jobs:
         with:
           fetch-depth: 0
 
+      - name: Setup Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Display Python version
+        run: python -c "import sys; print(sys.version)"
+
       - name: Install patchelf
         shell: bash
         run: |

diff --git a/.github/workflows/ubuntu-cuda-wheels.yml b/.github/workflows/ubuntu-cuda-wheels.yml
@@ -42,6 +42,14 @@ jobs:
         with:
           fetch-depth: 0
 
+      - name: Setup Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Display Python version
+        run: python -c "import sys; print(sys.version)"
+
       - name: Install patchelf
         shell: bash
         run: |

diff --git a/.github/workflows/windows-x64-cpu-wheels.yml b/.github/workflows/windows-x64-cpu-wheels.yml
@@ -2,6 +2,8 @@ name: build-wheels-cpu-win64
 
 on:
   push:
+    branches:
+      - release
     tags:
       - '*'
   workflow_dispatch:
@@ -47,7 +49,7 @@ jobs:
       - name: Build wheels
         uses: pypa/[email protected]
         env:
-          CIBW_BEFORE_BUILD: pip install torch==${{ matrix.torch}}+cpu cmake numpy k2==1.24.3.dev20230720+cpu.torch${{ matrix.torch }} kaldifeat==1.24.dev20230724+cpu.torch${{ matrix.torch }} -f https://csukuangfj.github.io/kaldifeat/cpu.html -f https://k2-fsa.github.io/k2/cpu.html -f https://download.pytorch.org/whl/torch_stable.html
+          CIBW_BEFORE_BUILD: pip install torch==${{ matrix.torch}}+cpu cmake numpy k2==1.24.3.dev20230726+cpu.torch${{ matrix.torch }} kaldifeat==1.25.0.dev20230726+cpu.torch${{ matrix.torch }} -f https://csukuangfj.github.io/kaldifeat/cpu.html -f https://k2-fsa.github.io/k2/cpu.html -f https://download.pytorch.org/whl/torch_stable.html
           CIBW_BUILD: ${{ matrix.python-version }}-win_amd64
           CIBW_REPAIR_WHEEL_COMMAND_WINDOWS: ""
           CIBW_BUILD_VERBOSITY: 3