add test also for row permutations (cpu and gpu)

eth-cscs · May 27, 2024 · 5cd926d · 5cd926d
1 parent 8ff6a8e
commit 5cd926d
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/test/unit/permutations/test_permutations_distributed.cpp b/test/unit/permutations/test_permutations_distributed.cpp
@@ -273,6 +273,18 @@ TYPED_TEST(PermutationsDistTestMC, ColumnsLocal) {
   }
 }
 
+TYPED_TEST(PermutationsDistTestMC, RowsLocal) {
+  using T = TypeParam;
+  constexpr auto CPU = Device::CPU;
+
+  for (auto& comm_grid : this->commGrids()) {
+    for (const auto& [n, nb, i_begin, i_end] : params2) {
+      testDistPermutationsAsLocal<T, CPU, Coord::Row>(comm_grid, n, nb, i_begin, i_end);
+      pika::wait();
+    }
+  }
+}
+
 #ifdef DLAF_WITH_GPU
 TYPED_TEST(PermutationsDistTestGPU, ColumnsLocal) {
   using T = TypeParam;
@@ -285,4 +297,16 @@ TYPED_TEST(PermutationsDistTestGPU, ColumnsLocal) {
     }
   }
 }
+
+TYPED_TEST(PermutationsDistTestGPU, RowsLocal) {
+  using T = TypeParam;
+  constexpr auto GPU = Device::GPU;
+
+  for (auto& comm_grid : this->commGrids()) {
+    for (const auto& [n, nb, i_begin, i_end] : params2) {
+      testDistPermutationsAsLocal<T, GPU, Coord::Row>(comm_grid, n, nb, i_begin, i_end);
+      pika::wait();
+    }
+  }
+}
 #endif