| /petsc/src/mat/impls/aij/seq/ |
| H A D | matptap.c | 193 PetscInt *ai = a->i, *aj = a->j, *apj, *apjdense, *pi = p->i, *pj = p->j, *pJ = p->j, *pjj; in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy() local
202 PetscCall(PetscMalloc1(cn, &apj)); in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
234 apj[apnzj++] = pjj[k]; in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
244 PetscCall(PetscSortInt(apnzj, apj)); in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
256 if (cjj[k] == apj[nextap]) caj[k] += (*pA) * apa[apj[nextap++]]; in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
264 apa[apj[j]] = 0.; in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
265 apjdense[apj[j]] = 0; in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
274 PetscCall(PetscFree(apj)); in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
|
| H A D | aij.h | 97 PetscInt *api, *apj; /* symbolic structure of A*P */ member
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpimatmatmult.c | 87 PetscCall(PetscFree(ptap->apj)); in MatProductCtxDestroy_MPIAIJ_MatMatMult()
103 PetscInt *api, *apj, *apJ, i, k; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable() local
140 apj = ptap->apj; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
153 apJ = PetscSafePointerPlusOffset(apj, api[i]); in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
202 PetscInt *lnk, i, pnz, row, *api, *apj, *Jptr, apnz, nspacedouble = 0, j, nzi; in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() local
293 PetscCall(PetscMalloc1(api[am], &ptap->apj)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
294 apj = ptap->apj; in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
295 PetscCall(PetscFreeSpaceContiguous(&free_space, ptap->apj)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
310 PetscCall(MatSetValues_MPIAIJ_CopyFromCSRFormat_Symbolic(C, apj, api)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
668 PetscInt *api, *apj, *apJ, i, j, k, row; in MatMatMultNumeric_MPIAIJ_MPIAIJ() local
[all …]
|
| H A D | mpiaij.h | 18 …PetscInt *api, *apj; /* symbolic i and j arrays of the local product A_lo… member
188 #define AProw_scalable(i, ad, ao, p_loc, p_oth, api, apj, apa) \ argument
192 _apJ = apj + api[i]; \
|
| H A D | mpiptap.c | 62 PetscCall(PetscFree(ptap->apj)); in MatProductCtxDestroy_MPIAIJ_PtAP()
103 PetscInt i, rstart, rend, cm, ncols, row, *api, *apj, am = A->rmap->n, apnz, nout; in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable() local
139 apj = ap->j; in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable()
140 PetscCall(ISLocalToGlobalMappingApply(ptap->ltog, api[AP_loc->rmap->n], apj, apj)); in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable()
146 AProw_scalable(i, ad, ao, p_loc, p_oth, api, apj, apa); in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable()
148 …all(ISGlobalToLocalMappingApply(ptap->ltog, IS_GTOLM_DROP, api[AP_loc->rmap->n], apj, &nout, apj)); in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable()
236 …PetscInt *api, *apj, *Jptr, apnz, *prmap = p->garray, con, j, Crmax, *aj, *ai, *pi,… in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() local
343 PetscCall(PetscCalloc2(api[am], &apj, api[am], &apv)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
344 PetscCall(PetscFreeSpaceContiguous(&free_space, apj)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
348 PetscCall(MatCreateSeqAIJWithArrays(PETSC_COMM_SELF, am, pN, api, apj, apv, &ptap->AP_loc)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
[all …]
|
| /petsc/src/mat/impls/maij/ |
| H A D | maij.c | 594 …PetscInt i, j, k, pshift, poffset, anzi, pnzi, apnzj, nextap, pnzj, prow, crow, *apj, *apj… in MatPtAPNumeric_SeqAIJ_SeqMAIJ() local
600 PetscCall(PetscCalloc3(cn, &apa, cn, &apj, cn, &apjdense)); in MatPtAPNumeric_SeqAIJ_SeqMAIJ()
621 apj[apnzj++] = poffset; in MatPtAPNumeric_SeqAIJ_SeqMAIJ()
631 PetscCall(PetscSortInt(apnzj, apj)); in MatPtAPNumeric_SeqAIJ_SeqMAIJ()
648 if (cjj[k] == apj[nextap]) caj[k] += (*pA) * apa[apj[nextap++]]; in MatPtAPNumeric_SeqAIJ_SeqMAIJ()
656 apa[apj[j]] = 0.; in MatPtAPNumeric_SeqAIJ_SeqMAIJ()
657 apjdense[apj[j]] = 0; in MatPtAPNumeric_SeqAIJ_SeqMAIJ()
664 PetscCall(PetscFree3(apa, apj, apjdense)); in MatPtAPNumeric_SeqAIJ_SeqMAIJ()
|