Searched refs:hipSpMV (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | hipsparsematimpl.h | 264 …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 D | aijhipsparse.hip.cxx | 3103 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 …]
|