Home
last modified time | relevance | path

Searched refs:transA (Results 1 – 3 of 3) sorted by relevance

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx805 bool transA, transB; /* use bool, since KK needs this type */ in MatProductNumeric_SeqAIJKokkos_SeqAIJKokkos() local
826 transA = false; in MatProductNumeric_SeqAIJKokkos_SeqAIJKokkos()
830 transA = true; in MatProductNumeric_SeqAIJKokkos_SeqAIJKokkos()
834 transA = false; in MatProductNumeric_SeqAIJKokkos_SeqAIJKokkos()
855 if (transA) { in MatProductNumeric_SeqAIJKokkos_SeqAIJKokkos()
857 transA = false; in MatProductNumeric_SeqAIJKokkos_SeqAIJKokkos()
865 …PetscCallCXX(KokkosSparse::spgemm_numeric(pdata->kh, csrmatA, transA, csrmatB, transB, ckok->csrma… in MatProductNumeric_SeqAIJKokkos_SeqAIJKokkos()
891 bool transA, transB; in MatProductSymbolic_SeqAIJKokkos_SeqAIJKokkos() local
923 transA = false; in MatProductSymbolic_SeqAIJKokkos_SeqAIJKokkos()
928 transA = true; in MatProductSymbolic_SeqAIJKokkos_SeqAIJKokkos()
[all …]
/petsc/src/mat/impls/aij/seq/aijmkl/
H A Daijmkl.c679 …MultSymbolic_SeqAIJMKL_SeqAIJMKL_Private(Mat A, const sparse_operation_t transA, Mat B, const spar… in MatMatMultSymbolic_SeqAIJMKL_SeqAIJMKL_Private() argument
690 if (transA == SPARSE_OPERATION_NON_TRANSPOSE) nrows = A->rmap->N; in MatMatMultSymbolic_SeqAIJMKL_SeqAIJMKL_Private()
704 …PetscCallExternal(mkl_sparse_sp2m, transA, descr_type_gen, csrA, transB, descr_type_gen, csrB, SPA… in MatMatMultSymbolic_SeqAIJMKL_SeqAIJMKL_Private()
713 PetscErrorCode MatMatMultNumeric_SeqAIJMKL_SeqAIJMKL_Private(Mat A, const sparse_operation_t transA… in MatMatMultNumeric_SeqAIJMKL_SeqAIJMKL_Private() argument
731 …PetscCallExternal(mkl_sparse_sp2m, transA, descr_type_gen, csrA, transB, descr_type_gen, csrB, SPA… in MatMatMultNumeric_SeqAIJMKL_SeqAIJMKL_Private()
/petsc/src/mat/impls/h2opus/cuda/
H A Dmath2opus.cu333 static PetscErrorCode MatMultNKernel_H2OPUS(Mat A, PetscBool transA, Mat B, Mat C) in MatMultNKernel_H2OPUS() argument
381 …PetscCheck(!transA || A->symmetric, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MatMultTransp… in MatMultNKernel_H2OPUS()
387 …hgemv(transA ? H2Opus_Trans : H2Opus_NoTrans, h2opus->s, *h2opus->hmatrix, uxx, blda, 0.0, uyy, cl… in MatMultNKernel_H2OPUS()
421 …PetscCheck(!transA || A->symmetric, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MatMultTransp… in MatMultNKernel_H2OPUS()
427 …hgemv(transA ? H2Opus_Trans : H2Opus_NoTrans, h2opus->s, *h2opus->hmatrix_gpu, uxx, blda, 0.0, uyy… in MatMultNKernel_H2OPUS()