fixes to trans_mode_kk_to_onemkl callsites

kokkos · Dec 7, 2023 · bf41c0f · bf41c0f
1 parent e66697d
commit bf41c0f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/blas/tpls/KokkosBlas3_gemm_tpl_spec_decl.hpp b/blas/tpls/KokkosBlas3_gemm_tpl_spec_decl.hpp
@@ -564,8 +564,8 @@ TPL_SCALAR_TYPE is the type MKL accents for SCALAR_TYPE
       const int64_t cst = is_lr ? C.stride(0) : C.stride(1);                     \
       const int64_t ldc = cst == 0 ? 1 : cst;                                    \
                                                                              \
-      oneapi::mkl::transpose transa = trans_mode_kk_to_onemkl(transA);           \
-      oneapi::mkl::transpose transb = trans_mode_kk_to_onemkl(transB);           \
+      oneapi::mkl::transpose transa = trans_mode_kk_to_onemkl(transA[0]);           \
+      oneapi::mkl::transpose transb = trans_mode_kk_to_onemkl(transB[0]);           \
       oneapi::mkl::blas::compute_mode mode = oneapi::mkl::blas::compute_mode::standard; \
  \
       if constexpr (!is_lr) { \