Home
last modified time | relevance | path

Searched refs:cuSpMV (Results 1 – 2 of 2) sorted by relevance

/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu3659 if (!matstruct->cuSpMV[opA].initialized) { /* built on demand */ in MatMultAddKernel_SeqAIJCUSPARSE()
3660 …PetscCallCUSPARSE(cusparseCreateDnVec(&matstruct->cuSpMV[opA].vecXDescr, nx, xptr, cusparse_scalar… in MatMultAddKernel_SeqAIJCUSPARSE()
3661 …PetscCallCUSPARSE(cusparseCreateDnVec(&matstruct->cuSpMV[opA].vecYDescr, ny, dptr, cusparse_scalar… in MatMultAddKernel_SeqAIJCUSPARSE()
3663 …, matstruct->cuSpMV[opA].vecXDescr, beta, matstruct->cuSpMV[opA].vecYDescr, cusparse_scalartype, c… in MatMultAddKernel_SeqAIJCUSPARSE()
3664 …PetscCallCUDA(cudaMalloc(&matstruct->cuSpMV[opA].spmvBuffer, matstruct->cuSpMV[opA].spmvBufferSize… in MatMultAddKernel_SeqAIJCUSPARSE()
3667 …, matstruct->cuSpMV[opA].vecXDescr, beta, matstruct->cuSpMV[opA].vecYDescr, cusparse_scalartype, c… in MatMultAddKernel_SeqAIJCUSPARSE()
3669 matstruct->cuSpMV[opA].initialized = PETSC_TRUE; in MatMultAddKernel_SeqAIJCUSPARSE()
3672 PetscCallCUSPARSE(cusparseDnVecSetValues(matstruct->cuSpMV[opA].vecXDescr, xptr)); in MatMultAddKernel_SeqAIJCUSPARSE()
3673 PetscCallCUSPARSE(cusparseDnVecSetValues(matstruct->cuSpMV[opA].vecYDescr, dptr)); in MatMultAddKernel_SeqAIJCUSPARSE()
3676 …, matstruct->cuSpMV[opA].vecXDescr, beta, matstruct->cuSpMV[opA].vecYDescr, cusparse_scalartype, c… in MatMultAddKernel_SeqAIJCUSPARSE()
[all …]
H A Dcusparsematimpl.h292 …Mat_CusparseSpMV cuSpMV[3]; /* different Mat_CusparseSpMV structs for non-transpose, transpose, co… member
296 cuSpMV[i].initialized = PETSC_FALSE; in Mat_SeqAIJCUSPARSEMultStruct()