| /petsc/src/mat/impls/aij/seq/ |
| H A D | matptap.c | 68 PetscInt *pti, *ptj, *ptJ, *ai = a->i, *aj = a->j, *ajj, *pi = p->i, *pj = p->j, *pjj; in MatPtAPSymbolic_SeqAIJ_SeqAIJ_SparseAxpy() local
105 ajj = aj + ai[arow]; in MatPtAPSymbolic_SeqAIJ_SeqAIJ_SparseAxpy()
107 if (!ptadenserow[ajj[k]]) { in MatPtAPSymbolic_SeqAIJ_SeqAIJ_SparseAxpy()
108 ptadenserow[ajj[k]] = -1; in MatPtAPSymbolic_SeqAIJ_SeqAIJ_SparseAxpy()
109 ptasparserow[ptanzi++] = ajj[k]; in MatPtAPSymbolic_SeqAIJ_SeqAIJ_SparseAxpy()
|
| /petsc/src/mat/impls/aij/mpi/mkl_cpardiso/ |
| H A D | mkl_cpardiso.c | 145 const PetscInt *ai, *aj, *bi, *bj, *garray, m = A->rmap->n, *ajj, *bjj; in MatConvertToTriples_mpiaij_mpiaij_MKL_CPARDISO() local
184 ajj = aj + ai[i]; /* ptr to the beginning of this row */ in MatConvertToTriples_mpiaij_mpiaij_MKL_CPARDISO()
188 colA_start = rstart + ajj[0]; /* the smallest global col index of A */ in MatConvertToTriples_mpiaij_mpiaij_MKL_CPARDISO()
200 col[nz] = rstart + ajj[j]; in MatConvertToTriples_mpiaij_mpiaij_MKL_CPARDISO()
216 …t *ai, *aj, *bi, *bj, *garray, bs = A->rmap->bs, bs2 = bs * bs, m = A->rmap->n / bs, *ajj, *bjj; in MatConvertToTriples_mpibaij_mpibaij_MKL_CPARDISO() local
255 ajj = aj + ai[i]; /* ptr to the beginning of this row */ in MatConvertToTriples_mpibaij_mpibaij_MKL_CPARDISO()
259 colA_start = rstart + (countA > 0 ? ajj[0] : 0); /* the smallest global col index of A */ in MatConvertToTriples_mpibaij_mpibaij_MKL_CPARDISO()
272 for (j = 0; j < countA; j++) col[nz++] = rstart + ajj[j] + 1; in MatConvertToTriples_mpibaij_mpibaij_MKL_CPARDISO()
289 …t *ai, *aj, *bi, *bj, *garray, bs = A->rmap->bs, bs2 = bs * bs, m = A->rmap->n / bs, *ajj, *bjj; in MatConvertToTriples_mpisbaij_mpisbaij_MKL_CPARDISO() local
327 ajj = aj + ai[i]; /* ptr to the beginning of this row */ in MatConvertToTriples_mpisbaij_mpisbaij_MKL_CPARDISO()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/mumps/ |
| H A D | mumps.c | 790 const PetscInt *ai, *aj, *ajj, M = A->rmap->n; in MatConvertToTriples_seqaij_seqaij() local
804 ajj = aj + ai[i]; in MatConvertToTriples_seqaij_seqaij()
807 PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[k])); in MatConvertToTriples_seqaij_seqaij()
847 const PetscInt *ai, *aj, *ajj, bs2 = aa->bs2; in MatConvertToTriples_seqbaij_seqaij() local
860 ajj = aj + ai[i]; in MatConvertToTriples_seqbaij_seqaij()
866 PetscCall(PetscMUMPSIntCast(bs * ajj[k] + j + shift, &col[idx])); in MatConvertToTriples_seqbaij_seqaij()
883 const PetscInt *ai, *aj, *ajj; in MatConvertToTriples_seqsbaij_seqsbaij() local
924 ajj = aj + ai[i]; in MatConvertToTriples_seqsbaij_seqsbaij()
928 if (ajj[j] > i || k >= m) { in MatConvertToTriples_seqsbaij_seqsbaij()
931 PetscCall(PetscMUMPSIntCast(ajj[j] * bs + k + shift, &col[nz])); in MatConvertToTriples_seqsbaij_seqsbaij()
[all …]
|
| /petsc/src/mat/impls/maij/ |
| H A D | maij.c | 679 const PetscInt *pi = p->i, *pj = p->j, *pjj, *ai = a->i, *aj = a->j, *ajj; in MatPtAPSymbolic_SeqAIJ_SeqMAIJ() local
714 ajj = aj + ai[arow]; in MatPtAPSymbolic_SeqAIJ_SeqMAIJ()
716 if (!ptadenserow[ajj[k]]) { in MatPtAPSymbolic_SeqAIJ_SeqMAIJ()
717 ptadenserow[ajj[k]] = -1; in MatPtAPSymbolic_SeqAIJ_SeqMAIJ()
718 ptasparserow[ptanzi++] = ajj[k]; in MatPtAPSymbolic_SeqAIJ_SeqMAIJ()
|
| /petsc/src/mat/impls/nest/ |
| H A D | matnest.c | 1790 PetscInt **aii, **ajj; in MatConvert_Nest_SeqAIJ_fast() local
1807 …PetscCall(PetscCalloc4(nest->nr * nest->nc, &aii, nest->nr * nest->nc, &ajj, nest->nr * nest->nc, … in MatConvert_Nest_SeqAIJ_fast()
1839 ajj[i * nest->nc + j] = njj; in MatConvert_Nest_SeqAIJ_fast()
1884 PetscInt *njj = ajj[i * nest->nc + j]; in MatConvert_Nest_SeqAIJ_fast()
1918 …0, PETSC_FALSE, PETSC_FALSE, &nnr, (const PetscInt **)&aii[k], (const PetscInt **)&ajj[k], &done)); in MatConvert_Nest_SeqAIJ_fast()
1925 PetscCall(PetscFree4(aii, ajj, avv, trans)); in MatConvert_Nest_SeqAIJ_fast()
|