Home
last modified time | relevance | path

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

/petsc/src/mat/impls/aij/seq/seqhipsparse/ !
H A Dhipsparsematimpl.h264 …Mat_HipsparseSpMV hipSpMV[3]; /* different Mat_CusparseSpMV structs for non-transpose, trans… member
267 for (int i = 0; i < 3; i++) hipSpMV[i].initialized = PETSC_FALSE; in Mat_SeqAIJHIPSPARSEMultStruct()
H A Daijhipsparse.hip.cxx3103 if (!matstruct->hipSpMV[opA].initialized) { /* built on demand */ in MatMultAddKernel_SeqAIJHIPSPARSE()
3104 …PetscCallHIPSPARSE(hipsparseCreateDnVec(&matstruct->hipSpMV[opA].vecXDescr, nx, xptr, hipsparse_sc… in MatMultAddKernel_SeqAIJHIPSPARSE()
3105 …PetscCallHIPSPARSE(hipsparseCreateDnVec(&matstruct->hipSpMV[opA].vecYDescr, ny, dptr, hipsparse_sc… in MatMultAddKernel_SeqAIJHIPSPARSE()
3106 …tstruct->alpha_one, matstruct->matDescr, matstruct->hipSpMV[opA].vecXDescr, beta, matstruct->hipSp… in MatMultAddKernel_SeqAIJHIPSPARSE()
3107 &matstruct->hipSpMV[opA].spmvBufferSize)); in MatMultAddKernel_SeqAIJHIPSPARSE()
3108 …PetscCallHIP(hipMalloc(&matstruct->hipSpMV[opA].spmvBuffer, matstruct->hipSpMV[opA].spmvBufferSize… in MatMultAddKernel_SeqAIJHIPSPARSE()
3109 matstruct->hipSpMV[opA].initialized = PETSC_TRUE; in MatMultAddKernel_SeqAIJHIPSPARSE()
3112 PetscCallHIPSPARSE(hipsparseDnVecSetValues(matstruct->hipSpMV[opA].vecXDescr, xptr)); in MatMultAddKernel_SeqAIJHIPSPARSE()
3113 PetscCallHIPSPARSE(hipsparseDnVecSetValues(matstruct->hipSpMV[opA].vecYDescr, dptr)); in MatMultAddKernel_SeqAIJHIPSPARSE()
3116 …matstruct->hipSpMV[opA].vecXDescr, beta, matstruct->hipSpMV[opA].vecYDescr, hipsparse_scalartype, … in MatMultAddKernel_SeqAIJHIPSPARSE()
[all …]