unit_test/blas: Check cublas gemv for unsupported transpose op

- Related to #974.
kokkos · May 13, 2021 · d419296 · d419296
1 parent 15018cb
commit d419296
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/unit_test/blas/Test_Blas2_gemv.hpp b/unit_test/blas/Test_Blas2_gemv.hpp
@@ -12,6 +12,7 @@ namespace Test {
     typedef typename ViewTypeA::value_type ScalarA;
     typedef typename ViewTypeX::value_type ScalarX;
     typedef typename ViewTypeY::value_type ScalarY;
+    using LayoutAType = typename ViewTypeA::array_layout;
     typedef Kokkos::ArithTraits<ScalarY> KAT_Y;
 
     typedef multivector_layout_adapter<ViewTypeA> vfA_type;
@@ -78,7 +79,16 @@ namespace Test {
     Kokkos::deep_copy(expected, h_org_y);
     vanillaGEMV(mode[0], alpha, h_A, h_x, beta, expected);
 
-    KokkosBlas::gemv(mode, alpha, A, x, beta, y);
+    // Cublas does not support row-major (LayoutRight) + conjugate transpose
+    // We throw a runtime error in the wrapper for cublasGemv if the user attempts
+    // this, therefore we must test this code path via the try-catch below.
+    try {
+      KokkosBlas::gemv(mode, alpha, A, x, beta, y);
+    } catch (const std::runtime_error &error) {
+      if ((mode[0] == 'c' || mode[0] == 'C') && std::is_same<LayoutAType, Kokkos::LayoutRight>::value)
+	return; // Pass since we caught the runtime error
+      FAIL();
+    }
     Kokkos::deep_copy(h_y, y);
     int numErrors = 0;
     for (int i = 0; i < ldy; i++) {