fixes to trans_mode_kk_to_onemkl callsites

cwpearson · cwpearson · commit bf41c0fb23f8 · 2023-12-07T10:34:43.000-07:00
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) { \

Original file line number	Diff line number	Diff line change
`@@ -564,8 +564,8 @@ TPL_SCALAR_TYPE is the type MKL accents for SCALAR_TYPE`
`564`	`564`	`const int64_t cst = is_lr ? C.stride(0) : C.stride(1); \`
`565`	`565`	`const int64_t ldc = cst == 0 ? 1 : cst; \`
`566`	`566`	`\`
`567`		`- oneapi::mkl::transpose transa = trans_mode_kk_to_onemkl(transA); \`
`568`		`- oneapi::mkl::transpose transb = trans_mode_kk_to_onemkl(transB); \`
	`567`	`+ oneapi::mkl::transpose transa = trans_mode_kk_to_onemkl(transA[0]); \`
	`568`	`+ oneapi::mkl::transpose transb = trans_mode_kk_to_onemkl(transB[0]); \`
`569`	`569`	`oneapi::mkl::blas::compute_mode mode = oneapi::mkl::blas::compute_mode::standard; \`
`570`	`570`	`\`
`571`	`571`	`if constexpr (!is_lr) { \`