Replace random_shuffle with shuffle

k2-fsa · Dec 20, 2023 · a9b19b4 · a9b19b4
1 parent 6936124
commit a9b19b4
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/.github/workflows/run-tests-cpu.yml b/.github/workflows/run-tests-cpu.yml
@@ -47,7 +47,7 @@ concurrency:
 jobs:
   run-tests-cpu:
     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:

diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml
@@ -47,7 +47,6 @@ concurrency:
 jobs:
   run-tests:
     runs-on: ${{ matrix.os }}
-    name: ${{ matrix.build_type }}
     strategy:
       fail-fast: false
       matrix:

diff --git a/.github/workflows/ubuntu-cuda-wheels.yml b/.github/workflows/ubuntu-cuda-wheels.yml
@@ -25,11 +25,11 @@ 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 --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)
+          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}"
 

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);