BLAS 3::trsm

KokkosBlas::trsm()

Header File: KokkosBlas3_trsm.hpp

Usage: KokkosBlas::trsm(side, uplo, trans, diag, alpha, A, B);

Triangular linear system solve with multiple right-hand-sides:

op(A)*X = alpha*B if side == "L" or "l"

X*op(A) = alpha*B if side == "R" or "r"

MultiVector Interface only

template<class AViewType,
         class BViewType>
void
trsm (const char side[],
      const char uplo[],
      const char trans[],
      const char diag[],
      typename BViewType::const_value_type& alpha,
      const AViewType& A,
      const BViewType& B)

Parameters:

• AViewType: 2-D Kokkos::View
• BViewType: 2-D Kokkos::View

Arguments:

• transA [in] "N" for non-transpose, "T" for transpose, "C" for conjugate transpose. All characters after the first are ignored. This works just like the BLAS routines.
• transB [in] "N" for non-transpose, "T" for transpose, "C" for conjugate transpose. All characters after the first are ignored. This works just like the BLAS routines.
• alpha [in] Input coefficient used for multiplication with B
• A [in] Input matrix, as a 2-D Kokkos::View
• A [in] Input matrix, as a 2-D Kokkos::View If side == "L" or "l", matrix A is a M-by-M triangular matrix; otherwise, matrix A is a N-by-N triangular matrix
• B [in] Input matrix, as a 2-D Kokkos::View
• beta [in] Input coefficient of C
• C [in/out] Output vector, as a nonconst 2-D Kokkos::View

Requirements:

• For a given mode, the dimensions of the matrices must align as necessary for matrix multiplication

Example

#include<Kokkos_Core.hpp>
#include<KokkosBlas3_gemm.hpp>

int main(int argc, char* argv[]) {
   Kokkos::initialize();

   int M = atoi(argv[1]);
   int N = atoi(argv[2]);

   Kokkos::View<double**> A("A",M,N);
   Kokkos::View<double**> B("B",N,M);
   Kokkos::View<double**> C("C",M,M);
   
   Kokkos::deep_copy(A,1.0);
   Kokkos::deep_copy(B,2.0);

   const double alpha = double(1.0);
   const double beta = double(0.0);
   
   KokkosBlas::gemm("N","N",alpha,A,B,beta,C);

   Kokkos::finalize();
}