Skip to content

Commit

Permalink
GEMM: fix "batched" in names
Browse files Browse the repository at this point in the history
  • Loading branch information
Mikołaj Zuzek committed Sep 28, 2022
1 parent 38c9989 commit de3d26c
Show file tree
Hide file tree
Showing 4 changed files with 470 additions and 486 deletions.
6 changes: 3 additions & 3 deletions blas/impl/KokkosBlas2_team_gemv_spec.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ struct TeamVectorGemv<MemberType, Trans::NoTranspose, Algo::Gemv::Unblocked> {
const MemberType& member, const ScalarType alpha, const AViewType& A,
const xViewType& x, const ScalarType beta, const yViewType& y) {
static_assert(AViewType::Rank == 2,
"Batched TeamVectorGemv requires rank-2 A matrix");
"KokkosBlas::TeamVectorGemv requires rank-2 A matrix");
return Impl::TeamVectorGemvInternal<Algo::Gemv::Unblocked>::invoke(
member, A.extent(0), A.extent(1), alpha, A.data(), A.stride_0(),
A.stride_1(), x.data(), x.stride_0(), beta, y.data(), y.stride_0());
Expand All @@ -213,7 +213,7 @@ struct TeamVectorGemv<MemberType, Trans::Transpose, Algo::Gemv::Unblocked> {
const MemberType& member, const ScalarType alpha, const AViewType& A,
const xViewType& x, const ScalarType beta, const yViewType& y) {
static_assert(AViewType::Rank == 2,
"Batched TeamVectorGemv requires rank-2 A matrix");
"KokkosBlas::TeamVectorGemv requires rank-2 A matrix");
return Impl::TeamVectorGemvInternal<Algo::Gemv::Unblocked>::invoke(
member, A.extent(1), A.extent(0), alpha, A.data(), A.stride_1(),
A.stride_0(), x.data(), x.stride_0(), beta, y.data(), y.stride_0());
Expand All @@ -232,7 +232,7 @@ struct TeamVectorGemv<MemberType, Trans::ConjTranspose, Algo::Gemv::Unblocked> {
const MemberType& member, const ScalarType alpha, const AViewType& A,
const xViewType& x, const ScalarType beta, const yViewType& y) {
static_assert(AViewType::Rank == 2,
"Batched TeamVectorGemv requires rank-2 A matrix");
"KokkosBlas::TeamVectorGemv requires rank-2 A matrix");
return Impl::TeamVectorGemvInternal<Algo::Gemv::Unblocked>::invoke(
member, Impl::OpConj{}, A.extent(1), A.extent(0), alpha, A.data(),
A.stride_1(), A.stride_0(), x.data(), x.stride_0(), beta, y.data(),
Expand Down
Loading

0 comments on commit de3d26c

Please sign in to comment.