From fbb30ccd373ac00f77902fd24c7d9dc1b0484644 Mon Sep 17 00:00:00 2001 From: Yuuichi Asahi Date: Thu, 19 Dec 2024 16:12:29 +0900 Subject: [PATCH] fix: op type of serial invoke Signed-off-by: Yuuichi Asahi --- batched/dense/impl/KokkosBatched_Gemm_Team_Internal.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/batched/dense/impl/KokkosBatched_Gemm_Team_Internal.hpp b/batched/dense/impl/KokkosBatched_Gemm_Team_Internal.hpp index 70c1ce3a03..ff4882b548 100644 --- a/batched/dense/impl/KokkosBatched_Gemm_Team_Internal.hpp +++ b/batched/dense/impl/KokkosBatched_Gemm_Team_Internal.hpp @@ -122,7 +122,7 @@ KOKKOS_INLINE_FUNCTION int TeamGemmInternal::invoke( i = ij / nq * mb; j = ij % nq * nb; } - inner.serial_invoke(KokkosBlas::Impl::OpID(), KokkosBlas::Impl::OpConj(), alpha, AA + i * as0, BB + j * bs1, + inner.serial_invoke(KokkosBlas::Impl::OpID(), KokkosBlas::Impl::OpID(), alpha, AA + i * as0, BB + j * bs1, (i + mb) > ib ? mp : mb, (j + nb) > jb ? np : nb, pb, CC + i * cs0 + j * cs1); }); };