Skip to content

Commit

Permalink
fix: op type of serial invoke
Browse files Browse the repository at this point in the history
Signed-off-by: Yuuichi Asahi <[email protected]>
  • Loading branch information
Yuuichi Asahi committed Dec 19, 2024
1 parent e509c56 commit fbb30cc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion batched/dense/impl/KokkosBatched_Gemm_Team_Internal.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -122,7 +122,7 @@ KOKKOS_INLINE_FUNCTION int TeamGemmInternal<Algo::Gemm::Blocked>::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);
});
};
Expand Down

0 comments on commit fbb30cc

Please sign in to comment.