Build wheels for torch 2.1.2 (#1265)

* support torch 2.1.2 * Replace random_shuffle with shuffle
k2-fsa · Dec 21, 2023 · 415fe1f · 415fe1f
1 parent 7711d16
commit 415fe1f
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 8 deletions.
diff --git a/.github/workflows/run-tests-cpu.yml b/.github/workflows/run-tests-cpu.yml
@@ -46,14 +46,13 @@ concurrency:
 
 jobs:
   run-tests-cpu:
-    if: github.event.label.name == 'ready' || github.event.label.name == 'cpp-test' || github.event_name == 'push'
     runs-on: ${{ matrix.os }}
-    name: ${{ matrix.python-version }} ${{ matrix.build_type }}
+    name: ${{ matrix.os }} ${{ matrix.torch }} ${{ matrix.python-version }} ${{ matrix.build_type }}
     strategy:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest]
-        torch: ["2.1.1"]
+        torch: ["2.1.2"]
         python-version: ["3.8", "3.9", "3.10", "3.11"]
         build_type: ["Release", "Debug"]
 

diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml
@@ -46,15 +46,13 @@ concurrency:
 
 jobs:
   run-tests:
-    if: github.event.label.name == 'ready' || github.event_name == 'push'
     runs-on: ${{ matrix.os }}
-    name: ${{ matrix.build_type }}
     strategy:
       fail-fast: false
       matrix:
         os: [ubuntu-latest]
         cuda: ["12.1"]
-        torch: ["2.1.1"]
+        torch: ["2.1.2"]
         python-version: ["3.11"]
 
     steps:

diff --git a/.github/workflows/ubuntu-cuda-wheels.yml b/.github/workflows/ubuntu-cuda-wheels.yml
@@ -25,8 +25,12 @@ jobs:
         id: set-matrix
         run: |
           # outputting for debugging purposes
+          # python ./scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch
+          # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch)
+
           python ./scripts/github_actions/generate_build_matrix.py --enable-cuda
           MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda)
+
           echo "::set-output name=matrix::${MATRIX}"
 
   build-manylinux-wheels:

diff --git a/k2/csrc/array_ops_test.cu b/k2/csrc/array_ops_test.cu
@@ -1573,7 +1573,9 @@ TEST(OpsTest, InvertPermutationTest) {
       int32_t len = RandInt(0, 10);
       std::vector<int32_t> permutation(len);
       std::iota(permutation.begin(), permutation.end(), 0);
-      std::random_shuffle(permutation.begin(), permutation.end());
+      std::random_device rd;
+      std::mt19937 g(rd());
+      std::shuffle(permutation.begin(), permutation.end(), g);
       Array1<int32_t> permutation_array(c, permutation);
       Array1<int32_t> permutation_array_inv =
           InvertPermutation(permutation_array);

diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py
@@ -172,10 +172,16 @@ def generate_build_matrix(
             if not for_windows
             else ["11.8.0", "12.1.0"],
         },
+        "2.1.2": {
+            "python-version": ["3.8", "3.9", "3.10", "3.11"],
+            "cuda": ["11.8", "12.1"]  # default 12.1
+            if not for_windows
+            else ["11.8.0", "12.1.0"],
+        },
         # https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
     }
     if test_only_latest_torch:
-        latest = "2.1.1"
+        latest = "2.1.2"
         matrix = {latest: matrix[latest]}
 
     if for_windows or for_macos: