fix problem with GPU

eth-cscs · Jan 4, 2024 · 3ef793d · 3ef793d
1 parent 1fceaea
commit 3ef793d
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/test/unit/test_blas_tile_extensions.cpp b/test/unit/test_blas_tile_extensions.cpp
@@ -61,7 +61,7 @@ TYPED_TEST(TileOperationsExtensionsTestGPU, Scal) {
     dlaf::test::testScal<Device::GPU, Type>(m, n, extra_lda);
 
     // Test a and b non const Tiles.
-    dlaf::test::testScal<Device::CPU, Type, Type>(m, n, extra_lda);
+    dlaf::test::testScal<Device::GPU, Type, Type>(m, n, extra_lda);
   }
 }
 #endif
@@ -75,26 +75,26 @@ std::vector<std::tuple<SizeType, SizeType, SizeType, SizeType, SizeType, SizeTyp
     {6, 9, 12, 1, 1, 1}, {32, 32, 32, 0, 0, 0}, {32, 32, 32, 4, 5, 7}, {128, 128, 128, 0, 0, 0},
 };
 
-TYPED_TEST(TileOperationsExtensionsTestMC, Scal) {
+TYPED_TEST(TileOperationsExtensionsTestMC, Add) {
   using Type = TypeParam;
   for (const auto& [m, n, k, extra_lda, extra_ldb, extra_ldc] : gemm_sizes) {
     // Test a and b const Tiles.
-    dlaf::test::testScal<Device::CPU, Type>(m, n, extra_lda);
+    dlaf::test::testAdd<Device::CPU, Type>(m, n, extra_lda);
     // Test a and b non const Tiles.
-    dlaf::test::testScal<Device::CPU, Type>(m, n, extra_lda);
+    dlaf::test::testAdd<Device::CPU, Type>(m, n, extra_lda);
   }
 }
 
 #ifdef DLAF_WITH_GPU
-TYPED_TEST(TileOperationsExtensionsTestGPU, Scal) {
+TYPED_TEST(TileOperationsExtensionsTestGPU, Add) {
   using Type = TypeParam;
 
   for (const auto& [m, n, k, extra_lda, extra_ldb, extra_ldc] : gemm_sizes) {
     // Test a and b const Tiles.
-    dlaf::test::testScal<Device::GPU, Type>(m, n, extra_lda);
+    dlaf::test::testAdd<Device::GPU, Type>(m, n, extra_lda);
 
     // Test a and b non const Tiles.
-    dlaf::test::testScal<Device::CPU, Type, Type>(m, n, extra_lda);
+    dlaf::test::testAdd<Device::GPU, Type, Type>(m, n, extra_lda);
   }
 }
 #endif