Use 32 core ubuntu

facebookresearch · Sep 3, 2024 · 3220a74 · 3220a74
1 parent f0884fc
commit 3220a74
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 11 deletions.
diff --git a/.github/workflows/conda_build_cuda.yml b/.github/workflows/conda_build_cuda.yml
@@ -5,7 +5,7 @@ on:
     inputs:
       os-build:
         type: string
-        default: "ubuntu-latest"
+        default: "32-core-ubuntu"
       os-test:
         type: string
         default: "4-core-ubuntu-gpu-t4"

diff --git a/.github/workflows/package_linux.yml b/.github/workflows/package_linux.yml
@@ -100,20 +100,20 @@ jobs:
     name: "Wheel (Python 3.12, CUDA 12.1)"
     uses: ./.github/workflows/wheel_build_cuda.yml
     with:
-      python-version: "cp312-cp312"
+      python-version: "3.12"
       cu-version: "12.1"
 
   wheel-py311-cu121:
     name: "Wheel (Python 3.11, CUDA 12.1)"
     uses: ./.github/workflows/wheel_build_cuda.yml
     with:
-      python-version: "cp311-cp311"
+      python-version: "3.11"
       cu-version: "12.1"
 
   wheel-py310-cu121:
     name: "Wheel (Python 3.10, CUDA 21.1)"
     uses: ./.github/workflows/wheel_build_cuda.yml
     with:
-      python-version: "cp310-cp310"
+      python-version: "3.10"
       cu-version: "12.1"
       run-test: true
diff --git a/.github/workflows/wheel_build_cuda.yml b/.github/workflows/wheel_build_cuda.yml
@@ -5,7 +5,7 @@ on:
     inputs:
       os-build:
         type: string
-        default: "ubuntu-latest"
+        default: "32-core-ubuntu"
       os-test:
         type: string
         default: "4-core-ubuntu-gpu-t4"
@@ -40,13 +40,14 @@ jobs:
           SPDL_USE_TRACING: "1"
           SPDL_BUILD_STUB: "0"
         run: |
+          ver="${${{ inputs.python-version }}//./}"
           python3 -m pip uninstall -y cmake
-          ./packaging/build_wheel.sh ${{ inputs.python-version }}
+          ./packaging/build_wheel.sh "cp${ver}-cp${ver}"
           ./packaging/audit_wheel.sh
 
       - uses: actions/upload-artifact@v4
         with:
-          name: "spdl.wheel.manylinux.${{ inputs.python-version }}.cu${{ inputs.cu-version }}"
+          name: "spdl.wheel.manylinux.py${{ inputs.python-version }}.cu${{ inputs.cu-version }}"
           path: ./wheelhouse
           if-no-files-found: error
           retention-days: 1
@@ -68,13 +69,13 @@ jobs:
 
       - uses: actions/download-artifact@v4
         with:
-          name: "spdl.wheel.manylinux.${{ inputs.python-version }}.cu${{ inputs.cu-version }}"
+          name: "spdl.wheel.manylinux.py${{ inputs.python-version }}.cu${{ inputs.cu-version }}"
           path: ~/package
 
       - name: Unit test
         run: |
-          pip install --find-links "${HOME}/package/ spdl
-          apt update && apt install -yqq ffmpeg
-          pip install -c pytorch numpy pytest pytorch numba
+          pip install --find-links "${HOME}/package/" spdl
+          sudo apt update && sudo apt install -yqq ffmpeg
+          pip install numpy pytest torch --index-url https://download.pytorch.org/whl/cu${{ inputs.cu-version }}
           pytest -v \
               tests/spdl_unittest/cuda/