| /petsc/src/mat/utils/ |
| H A D | compressedrow.c | 32 PetscInt *cpi = NULL, *ridx = NULL, nz, i, row; in MatCheckCompressedRow() local
53 PetscCall(PetscMalloc2(nrows + 1, &cpi, nrows, &ridx)); in MatCheckCompressedRow()
60 ridx[row++] = i; /* compressed row local index */ in MatCheckCompressedRow()
62 while (row < nrows) ridx[row++] = -1; // pad array in MatCheckCompressedRow()
65 compressedrow->rindex = ridx; in MatCheckCompressedRow()
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | baij2.c | 243 const PetscInt *idx, *ii, *ridx = NULL; in MatMult_SeqBAIJ_1() local
253 ridx = a->compressedrow.rindex; in MatMult_SeqBAIJ_1()
270 z[ridx[i]] = sum; in MatMult_SeqBAIJ_1()
288 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_2() local
300 ridx = a->compressedrow.rindex; in MatMult_SeqBAIJ_2()
323 if (usecprow) z = zarray + 2 * ridx[i]; in MatMult_SeqBAIJ_2()
340 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_3() local
356 ridx = a->compressedrow.rindex; in MatMult_SeqBAIJ_3()
383 if (usecprow) z = zarray + 3 * ridx[i]; in MatMult_SeqBAIJ_3()
401 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_4() local
[all …]
|
| H A D | baij.c | 1363 PetscInt ridx, cidx, lastcol = -1; in matsetvalues4_() local
1378 ridx = row % 4; in matsetvalues4_()
1392 bap = ap + 16 * i + 4 * cidx + ridx; in matsetvalues4_()
1404 ap[16 * i + 4 * cidx + ridx] = value; in matsetvalues4_()
2059 PetscInt brow, bcol, ridx, cidx, bs = A->rmap->bs, bs2 = a->bs2; in MatGetValues_SeqBAIJ() local
2083 ridx = row % bs; in MatGetValues_SeqBAIJ()
2094 *v++ = ap[bs2 * i + bs * cidx + ridx]; in MatGetValues_SeqBAIJ()
2449 PetscInt ridx, cidx, bs2 = a->bs2; in MatSetValues_SeqBAIJ() local
2470 ridx = row % bs; in MatSetValues_SeqBAIJ()
2490 bap = PetscSafePointerPlusOffset(ap, bs2 * i + bs * cidx + ridx); in MatSetValues_SeqBAIJ()
[all …]
|
| /petsc/src/mat/tests/ |
| H A D | ex139.c | 27 … m, n, top_bs, row_bs, col_bs, nlocblocks, *idx, nrowblocks, ncolblocks, *ridx, *cidx, *icol, *iro… in main() local
75 PetscCall(PetscMalloc2(nrowblocks, &ridx, ncolblocks, &cidx)); in main()
76 for (i = 0; i < nrowblocks; i++) ridx[i] = i + ((i > nrowblocks / 2) ^ !rstart); in main()
78 PetscCall(ISCreateBlock(PETSC_COMM_SELF, row_bs, nrowblocks, ridx, PETSC_COPY_VALUES, &is0)); in main()
80 PetscCall(PetscFree2(ridx, cidx)); in main()
|
| H A D | ex141.c | 15 PetscInt ridx[2], cidx[2]; in main() local
77 ridx[i] = block * bs + i; in main()
80 PetscCall(MatSetValues(C, bs, ridx, bs, cidx, value, INSERT_VALUES)); in main()
|
| /petsc/src/mat/impls/elemental/ |
| H A D | matelem.cxx | 110 PetscInt i, j, rrank, ridx, crank, cidx, erow, ecol, numQueues = 0; in MatSetValues_Elemental() local
119 P2RO(A, 0, rows[i], &rrank, &ridx); in MatSetValues_Elemental()
120 RO2E(A, 0, rrank, ridx, &erow); in MatSetValues_Elemental()
121 …PetscCheck(rrank >= 0 && ridx >= 0 && erow >= 0, PetscObjectComm((PetscObject)A), PETSC_ERR_PLIB, … in MatSetValues_Elemental()
151 P2RO(A, 0, rows[i], &rrank, &ridx); in MatSetValues_Elemental()
152 RO2E(A, 0, rrank, ridx, &erow); in MatSetValues_Elemental()
171 P2RO(A, 0, rows[i], &rrank, &ridx); in MatSetValues_Elemental()
172 RO2E(A, 0, rrank, ridx, &erow); in MatSetValues_Elemental()
173 …PetscCheck(rrank >= 0 && ridx >= 0 && erow >= 0, PetscObjectComm((PetscObject)A), PETSC_ERR_PLIB, … in MatSetValues_Elemental()
203 P2RO(A, 0, rows[i], &rrank, &ridx); in MatSetValues_Elemental()
[all …]
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | sbaij.c | 515 PetscInt brow, bcol, ridx, cidx, bs = A->rmap->bs, bs2 = a->bs2; in MatGetValues_SeqSBAIJ() local
539 ridx = row % bs; in MatGetValues_SeqSBAIJ()
550 *v++ = ap[bs2 * i + bs * cidx + ridx]; in MatGetValues_SeqSBAIJ()
743 PetscInt ridx, cidx, bs2 = a->bs2; in MatSetValues_SeqSBAIJ() local
769 ridx = row % bs; in MatSetValues_SeqSBAIJ()
771 if ((brow == bcol && ridx <= cidx) || (brow < bcol)) { in MatSetValues_SeqSBAIJ()
789 bap = ap + bs2 * i + bs * cidx + ridx; in MatSetValues_SeqSBAIJ()
793 if (brow == bcol && ridx < cidx) { in MatSetValues_SeqSBAIJ()
794 bap = ap + bs2 * i + bs * ridx + cidx; in MatSetValues_SeqSBAIJ()
813 ap[bs2 * i + bs * cidx + ridx] = value; in MatSetValues_SeqSBAIJ()
[all …]
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 368 const PetscInt *ridx = A->rmap->mapping->indices, *cidx = A->cmap->mapping->indices; in MatSeqAIJSetValuesLocalFast() local
372 row = ridx[im[0]]; in MatSeqAIJSetValuesLocalFast()
1406 const PetscInt *idx, *ii, *ridx = NULL; in MatMultTransposeAdd_SeqAIJ() local
1423 ridx = cprow.rindex; in MatMultTransposeAdd_SeqAIJ()
1432 alpha = x[ridx[i]]; in MatMultTransposeAdd_SeqAIJ()
1463 const PetscInt *ii, *ridx = NULL; in MatMult_SeqAIJ() local
1483 ridx = a->compressedrow.rindex; in MatMult_SeqAIJ()
1492 y[ridx[i]] = sum; in MatMult_SeqAIJ()
1524 const PetscInt *aj, *ii, *ridx = NULL; in MatMultMax_SeqAIJ() local
1540 ridx = a->compressedrow.rindex; in MatMultMax_SeqAIJ()
[all …]
|
| H A D | inode.c | 4129 const PetscInt *ridx, *cidx; in MatInodeAdjustForInodes_SeqAIJ_Inode() local
4142 PetscCall(ISGetIndices(ris, &ridx)); in MatInodeAdjustForInodes_SeqAIJ_Inode()
4150 indx = ridx[i]; in MatInodeAdjustForInodes_SeqAIJ_Inode()
4172 PetscCall(ISRestoreIndices(ris, &ridx)); in MatInodeAdjustForInodes_SeqAIJ_Inode()
|
| /petsc/src/mat/impls/blockmat/seq/ |
| H A D | blockmat.c | 220 PetscInt ridx, cidx; in MatSetValues_BlockMat() local
242 ridx = row % bs; in MatSetValues_BlockMat()
276 PetscCall(MatSetValues(ap[i], 1, &ridx, 1, &cidx, &value, is)); in MatSetValues_BlockMat()
|
| /petsc/src/mat/impls/sbaij/mpi/ |
| H A D | mpisbaij.c | 209 ridx = row % bs; \
221 bap = ap + bs2 * _i + bs * cidx + ridx; \
236 ap[bs2 * _i + bs * cidx + ridx] = value; \
249 ridx = row % bs; \
261 bap = ap + bs2 * _i + bs * cidx + ridx; \
276 ap[bs2 * _i + bs * cidx + ridx] = value; \
306 PetscInt low, high, t, ridx, cidx, bs2 = a->bs2; in MatSetValues_MPISBAIJ() local
|
| /petsc/src/mat/graphops/coarsen/impls/hem/ |
| H A D | hem.c | 477 PetscInt *ridx = matB->compressedrow.rindex, lid = ridx[ix]; in MatCoarsenApply_HEM_private() local
478 if (ridx[ix] >= 0) lid_cprowID[lid] = ix; in MatCoarsenApply_HEM_private()
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mpibaij.c | 168 ridx = row % bs; \
180 bap = ap + bs2 * _i + bs * cidx + ridx; \
195 ap[bs2 * _i + bs * cidx + ridx] = value; \
208 ridx = row % bs; \
220 bap = ap + bs2 * _i + bs * cidx + ridx; \
235 ap[bs2 * _i + bs * cidx + ridx] = value; \
262 PetscInt low, high, t, ridx, cidx, bs2 = a->bs2; in MatSetValues_MPIBAIJ() local
|
| /petsc/src/mat/impls/is/ |
| H A D | matis.c | 667 const PetscInt *ridx, *cidx; in MatConvert_XAIJ_IS() local
672 PetscCall(ISLocalToGlobalMappingGetBlockIndices(rl2g, &ridx)); in MatConvert_XAIJ_IS()
678 PetscCall(ISCreateBlock(comm, rbs, nr / rbs, ridx, PETSC_USE_POINTER, &rows)); in MatConvert_XAIJ_IS()
688 PetscCall(ISLocalToGlobalMappingRestoreBlockIndices(rl2g, &ridx)); in MatConvert_XAIJ_IS()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 2017 PetscInt m = A->rmap->n, *ii, *ridx, tmp; in MatSeqAIJHIPSPARSECopyToGPU() local
2046 ridx = a->compressedrow.rindex; in MatSeqAIJHIPSPARSECopyToGPU()
2050 ridx = NULL; in MatSeqAIJHIPSPARSECopyToGPU()
2125 matstruct->cprowIndices->assign(ridx, ridx + m); in MatSeqAIJHIPSPARSECopyToGPU()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 2428 PetscInt m = A->rmap->n, *ii, *ridx, tmp; in MatSeqAIJCUSPARSECopyToGPU() local
2459 ridx = a->compressedrow.rindex; in MatSeqAIJCUSPARSECopyToGPU()
2463 ridx = NULL; in MatSeqAIJCUSPARSECopyToGPU()
2549 matstruct->cprowIndices->assign(ridx, ridx + m); in MatSeqAIJCUSPARSECopyToGPU()
|
| /petsc/src/mat/impls/aij/mpi/mumps/ |
| H A D | mumps.c | 1603 const PetscInt *ridx = rows_idx[r]; in MatConvertToTriples_nest_xaij() local
1640 ridx = cols_idx[c]; in MatConvertToTriples_nest_xaij()
1659 PetscCall(PetscMUMPSIntCast(ridx[mumps->irn[k] - rst] + shift, &jcns[cumnnz + k])); in MatConvertToTriples_nest_xaij()
1664 PetscCall(PetscMUMPSIntCast(ridx[mumps->irn[k] - rst] + shift, &irns[cumnnz + k])); in MatConvertToTriples_nest_xaij()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 946 const PetscInt *aj, *ii, *ridx; in MatZeroRowsColumns_MPIAIJ() local
1002 ridx = aij->compressedrow.rindex; in MatZeroRowsColumns_MPIAIJ()
1010 if (b) bb[*ridx] -= *aa * xx[*aj]; in MatZeroRowsColumns_MPIAIJ()
1016 ridx++; in MatZeroRowsColumns_MPIAIJ()
|