test fix dynamic embedding wheels (pytorch#1428)

Summary: Pull Request resolved: pytorch#1428 Differential Revision: D50034865
henrylhtsang · Oct 7, 2023 · df99fad · df99fad
1 parent 8beba7d
commit df99fad
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/.github/workflows/build_dynamic_embedding_wheels.yml b/.github/workflows/build_dynamic_embedding_wheels.yml
@@ -14,20 +14,24 @@ on:
 
 jobs:
   build_wheels:
-    name: Wheels on ${{ matrix.os }}/${{ matrix.pyver }}/cu${{ matrix.cuver }}
+    name: build_dynamic_embedding_wheels_${{ matrix.python-version }}_cu${{ matrix.cuver }}
     runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false
       matrix:
-        os: [ ubuntu-latest ]
-        pyver: [ cp38, cp39, cp310 ]
-        cuver: [ "11.6", "11.3"]
+        os: [linux.2xlarge]
+        python-version: ["3.8"]
+        cuver: ["11.6"]
 
     steps:
       - uses: actions/checkout@v3
         with:
           submodules: recursive
 
+      - name: install pipx
+        run: |
+          pip3 install --user pipx
+
       - uses: pypa/[email protected]
         with:
           package-dir: contrib/dynamic_embedding