Home
last modified time | relevance | path

Searched refs:apa (Results 1 – 6 of 6) sorted by relevance

/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.h21 PetscScalar *apa; /* tmp array for store a row of A*P used in MatMatMult() */ member
188 #define AProw_scalable(i, ad, ao, p_loc, p_oth, api, apj, apa) \ argument
209 apa[_k] += _valtmp * _pa[_nextp++]; \
231 apa[_k] += _valtmp * _pa[_nextp++]; \
239 #define AProw_nonscalable(i, ad, ao, p_loc, p_oth, apa) \ argument
256 for (_k = 0; _k < _pnz; _k++) apa[_pj[_k]] += _valtmp * _pa[_k]; \
273 for (_k = 0; _k < _pnz; _k++) apa[_pj[_k]] += _valtmp * _pa[_k]; \
H A Dmpiptap.c66 if (ptap->apa) PetscCall(PetscFree(ptap->apa)); in MatProductCtxDestroy_MPIAIJ_PtAP()
104 PetscScalar *apa; in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable() local
144 apa = ap->a + api[i]; in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable()
145 PetscCall(PetscArrayzero(apa, apnz)); in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable()
146 AProw_scalable(i, ad, ao, p_loc, p_oth, api, apj, apa); in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable()
1845 PetscCall(PetscCalloc1(pN, &ptap->apa)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1866 PetscScalar *apa; in MatPtAPNumeric_MPIAIJ_MPIAIJ() local
1902 PetscCall(MatSeqAIJGetArrayWrite(AP_loc, &apa)); in MatPtAPNumeric_MPIAIJ_MPIAIJ()
1909 …AProw_nonscalable(i, ad, ao, p_loc, p_oth, ptap->apa); // Directly access the value arrays from th… in MatPtAPNumeric_MPIAIJ_MPIAIJ()
1913 apa[j + ap->i[i]] = ptap->apa[col]; in MatPtAPNumeric_MPIAIJ_MPIAIJ()
[all …]
H A Dmpimatmatmult.c88 PetscCall(PetscFree(ptap->apa)); in MatProductCtxDestroy_MPIAIJ_MatMatMult()
100 PetscScalar *apa, *ca; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable() local
137 apa = ptap->apa; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
150 AProw_nonscalable(i, ad, ao, p_loc, p_oth, apa); in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
162 ca[k0] = apa[apJ[k]]; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
163 apa[apJ[k++]] = 0.0; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
169 ca[k1] = apa[apJ[k]]; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
170 apa[apJ[k++]] = 0.0; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
176 ca[k0] = apa[apJ[k]]; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
177 apa[apJ[k++]] = 0.0; in MatMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
[all …]
/petsc/src/mat/impls/aij/seq/
H A Dmatptap.c197 MatScalar *aa, *apa, *pa, *pA, *paj, *ca, *caj; in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy() local
201 PetscCall(PetscCalloc2(cn, &apa, cn, &apjdense)); in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
236 apa[pjj[k]] += (*aa) * paj[k]; 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()
273 PetscCall(PetscFree2(apa, apjdense)); in MatPtAPNumeric_SeqAIJ_SeqAIJ_SparseAxpy()
H A Daij.h98 PetscScalar *apa; /* temporary array for storing one row of A*P */ member
/petsc/src/mat/impls/maij/
H A Dmaij.c596 MatScalar *ca = c->a, *caj, *apa; in MatPtAPNumeric_SeqAIJ_SeqMAIJ() local
600 PetscCall(PetscCalloc3(cn, &apa, cn, &apj, cn, &apjdense)); in MatPtAPNumeric_SeqAIJ_SeqMAIJ()
623 apa[poffset] += (*aa) * paj[k]; 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()
664 PetscCall(PetscFree3(apa, apj, apjdense)); in MatPtAPNumeric_SeqAIJ_SeqMAIJ()