| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mmbaij.c | 12 PetscInt bs = mat->rmap->bs, *stmp; in MatSetUpMultiply_MPIBAIJ()
62 …omSizes(PetscObjectComm((PetscObject)baij->B), ec * mat->rmap->bs, ec * mat->rmap->bs, mat->rmap->… in MatSetUpMultiply_MPIBAIJ()
91 …omSizes(PetscObjectComm((PetscObject)baij->B), ec * mat->rmap->bs, ec * mat->rmap->bs, mat->rmap->… in MatSetUpMultiply_MPIBAIJ()
134 PetscInt bs2 = baij->bs2, *nz = NULL, m = A->rmap->n; in MatDisAssemble_MPIBAIJ()
163 PetscCall(MatSeqBAIJSetPreallocation(Bnew, B->rmap->bs, 0, nz)); in MatDisAssemble_MPIBAIJ()
203 PetscInt bs = inA->rmap->bs, i, n, nt, j, cstart, cend, no, *garray = ina->garray, *lindices; in MatMPIBAIJDiagonalScaleLocalSetUp()
209 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd)); in MatMPIBAIJDiagonalScaleLocalSetUp()
211 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIBAIJDiagonalScaleLocalSetUp()
212 … if (inA->rmap->mapping->indices[i] * bs >= cstart && inA->rmap->mapping->indices[i] * bs < cend) { in MatMPIBAIJDiagonalScaleLocalSetUp()
214 r_rmapd[i] = inA->rmap->mapping->indices[i] + 1; in MatMPIBAIJDiagonalScaleLocalSetUp()
[all …]
|
| H A D | mpb_baij.c | 10 PetscInt bs = mat->rmap->bs; in MatGetMultiProcBlock_MPIBAIJ()
21 PetscCall(MatSetSizes(*subMat, mat->rmap->n, mat->cmap->n, PETSC_DECIDE, PETSC_DECIDE)); in MatGetMultiProcBlock_MPIBAIJ()
22 PetscCall(MatSetBlockSizes(*subMat, mat->rmap->bs, mat->cmap->bs)); in MatGetMultiProcBlock_MPIBAIJ()
25 PetscCall(PetscLayoutSetBlockSize((*subMat)->rmap, mat->rmap->bs)); in MatGetMultiProcBlock_MPIBAIJ()
27 PetscCall(PetscLayoutSetUp((*subMat)->rmap)); in MatGetMultiProcBlock_MPIBAIJ()
54 PetscCall(PetscCalloc1(aij->B->rmap->n / bs, &nnz)); in MatGetMultiProcBlock_MPIBAIJ()
55 for (i = 0; i < aij->B->rmap->n / bs; i++) { in MatGetMultiProcBlock_MPIBAIJ()
81 for (i = 0; i < aij->B->rmap->n / bs; i++) { in MatGetMultiProcBlock_MPIBAIJ()
82 newRow = (*subMat)->rmap->range[subCommRank] + i * bs; in MatGetMultiProcBlock_MPIBAIJ()
|
| H A D | mpibaij.c | 12 …State((PetscObject)mat, "Rows=%" PetscInt_FMT ",Cols=%" PetscInt_FMT, mat->rmap->N, mat->cmap->N)); in MatDestroy_MPIBAIJ()
60 PetscInt i, *idxb = NULL, m = A->rmap->n, bs = A->cmap->bs; in MatGetRowMaxAbs_MPIBAIJ()
147 PetscInt nbs = B->nbs, i, bs = mat->rmap->bs; in MatCreateColmap_MPIBAIJ_Private()
246 PetscInt rstart_orig = mat->rmap->rstart; in MatSetValues_MPIBAIJ()
247 PetscInt rend_orig = mat->rmap->rend, cstart_orig = mat->cmap->rstart; in MatSetValues_MPIBAIJ()
248 PetscInt cend_orig = mat->cmap->rend, bs = mat->rmap->bs; in MatSetValues_MPIBAIJ()
268 …im[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" PetscInt_F… in MatSetValues_MPIBAIJ()
331 PetscInt *aj = a->j, nonew = a->nonew, bs2 = a->bs2, bs = A->rmap->bs; in MatSetValuesBlocked_SeqBAIJ_Inlined()
415 PetscInt cend = baij->cendbs, bs = mat->rmap->bs, bs2 = baij->bs2; in MatSetValuesBlocked_MPIBAIJ()
497 PetscInt rstart_orig = mat->rmap->rstart; in MatSetValues_MPIBAIJ_HT()
[all …]
|
| /petsc/src/mat/impls/dense/seq/hdf5/ |
| H A D | densehdf5.c | 54 if (mat->rmap->n >= 0 && mat->cmap->N < 0) { in MatLoad_Dense_HDF5()
59 vmap->n = (mat->rmap->n < 0 || mat->cmap->N < 0) ? -1 : mat->rmap->n * mat->cmap->N; in MatLoad_Dense_HDF5()
60 vmap->N = (mat->rmap->N < 0 || mat->cmap->N < 0) ? -1 : mat->rmap->N * mat->cmap->N; in MatLoad_Dense_HDF5()
68 mat->rmap->n = vmap->n / mat->cmap->N; in MatLoad_Dense_HDF5()
69 mat->rmap->N = vmap->N / mat->cmap->N; in MatLoad_Dense_HDF5()
70 PetscCall(PetscLayoutSetUp(mat->rmap)); in MatLoad_Dense_HDF5()
|
| /petsc/src/mat/tests/ |
| H A D | ex46.c | 10 ISLocalToGlobalMapping rmap, cmap; in main() local
19 PetscCall(ISLocalToGlobalMappingCreateIS(is, &rmap)); in main()
20 PetscCall(ISLocalToGlobalMappingSetBlockSize(rmap, 2)); in main()
24 PetscCall(MatSetLocalToGlobalMapping(A, rmap, cmap)); in main()
25 PetscCall(ISLocalToGlobalMappingDestroy(&rmap)); in main()
42 PetscCall(MatGetLocalToGlobalMapping(A, &rmap, &cmap)); in main()
43 PetscCall(ISLocalToGlobalMappingGetBlockSize(rmap, &l2grbs)); in main()
|
| H A D | ex210.c | 8 ISLocalToGlobalMapping cmap, rmap; in main() local
21 …PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD, 1, 1, indices, PETSC_COPY_VALUES, &rmap)); in main()
22 PetscCall(MatSetLocalToGlobalMapping(A, rmap, cmap)); in main()
27 PetscCall(MatSetLocalToGlobalMapping(B, rmap, rmap)); in main()
28 PetscCall(ISLocalToGlobalMappingDestroy(&rmap)); in main()
|
| H A D | ex212.c | 9 ISLocalToGlobalMapping cmap, rmap; in main() local
22 …PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD, 2, 1, indices, PETSC_COPY_VALUES, &rmap)); in main()
24 PetscCall(MatSetLocalToGlobalMapping(A, rmap, cmap)); in main()
25 PetscCall(ISLocalToGlobalMappingDestroy(&rmap)); in main()
27 …PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD, 1, 2, indices, PETSC_COPY_VALUES, &rmap)); in main()
33 PetscCall(MatSetLocalToGlobalMapping(B, rmap, cmap)); in main()
34 PetscCall(ISLocalToGlobalMappingDestroy(&rmap)); in main()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpihashmat.h | 12 const PetscInt rStart = A->rmap->rstart; in MatSetValues_MPI_Hash()
13 const PetscInt rEnd = A->rmap->rend; in MatSetValues_MPI_Hash()
17 const PetscInt bs = A->rmap->bs; in MatSetValues_MPI_Hash()
63 PetscCall(MatStashScatterBegin_Private(A, &A->stash, A->rmap->range)); in MatAssemblyBegin_MPI_Hash()
171 PetscCall(PetscLayoutSetUp(A->rmap)); in MatSetUp_MPI_Hash()
173 if (A->rmap->bs < 1) A->rmap->bs = 1; in MatSetUp_MPI_Hash()
181 a->mbs = A->rmap->n / bs; in MatSetUp_MPI_Hash()
183 a->Mbs = A->rmap->N / bs; in MatSetUp_MPI_Hash()
186 for (PetscInt i = 0; i <= a->size; i++) a->rangebs[i] = A->rmap->range[i] / bs; in MatSetUp_MPI_Hash()
187 a->rstartbs = A->rmap->rstart / bs; in MatSetUp_MPI_Hash()
[all …]
|
| H A D | mmaij.c | 29 PetscCall(PetscHMapICreateWithSize(aij->B->rmap->n, &gid1_lid1)); in MatSetUpMultiply_MPIAIJ()
30 for (i = 0; i < aij->B->rmap->n; i++) { in MatSetUpMultiply_MPIAIJ()
55 for (i = 0; i < aij->B->rmap->n; i++) { in MatSetUpMultiply_MPIAIJ()
70 for (i = 0; i < aij->B->rmap->n; i++) { in MatSetUpMultiply_MPIAIJ()
88 for (i = 0; i < aij->B->rmap->n; i++) { in MatSetUpMultiply_MPIAIJ()
151 … PetscInt i, j, m = B->rmap->n, n = A->cmap->N, col, ct = 0, *garray = aij->garray, *nz; in MatDisAssemble_MPIAIJ()
209 PetscInt i, j, n, nt, cstart, cend, no, *garray = ina->garray, *lindices, bs = inA->rmap->bs; in MatMPIAIJDiagonalScaleLocalSetUp()
215 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd)); in MatMPIAIJDiagonalScaleLocalSetUp()
217 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIAIJDiagonalScaleLocalSetUp()
218 … if (inA->rmap->mapping->indices[i] * bs >= cstart && inA->rmap->mapping->indices[i] * bs < cend) { in MatMPIAIJDiagonalScaleLocalSetUp()
[all …]
|
| H A D | mpb_aij.c | 20 PetscCall(MatSetSizes(*subMat, mat->rmap->n, mat->cmap->n, PETSC_DECIDE, PETSC_DECIDE)); in MatGetMultiProcBlock_MPIAIJ()
24 PetscCall(PetscLayoutSetUp((*subMat)->rmap)); in MatGetMultiProcBlock_MPIAIJ()
49 PetscCall(PetscCalloc1(aij->B->rmap->n, &nnz)); in MatGetMultiProcBlock_MPIAIJ()
50 for (i = 0; i < aij->B->rmap->n; i++) { in MatGetMultiProcBlock_MPIAIJ()
73 for (i = 0; i < aij->B->rmap->n; i++) { in MatGetMultiProcBlock_MPIAIJ()
74 newRow = (*subMat)->rmap->range[subCommRank] + i; in MatGetMultiProcBlock_MPIAIJ()
|
| /petsc/src/ksp/ksp/utils/schurm/ |
| H A D | schurm.c | 233 …>rmap->n == A00->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local rows of A00 %" PetscInt_FMT "… in MatSchurComplementSetSubMatrices()
234 …rmap->n == Ap00->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local rows of A00 %" PetscInt_FMT "… in MatSchurComplementSetSubMatrices()
235 …rmap->n == Ap00->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local rows of Ap00 %" PetscInt_FMT … in MatSchurComplementSetSubMatrices()
236 …rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local columns of A00 %" PetscInt_FMT " do not equal … in MatSchurComplementSetSubMatrices()
237 …rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local columns of A10 %" PetscInt_FMT " do not equal … in MatSchurComplementSetSubMatrices()
241 …rmap->n == A11->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local rows of A10 %" PetscInt_FMT " … in MatSchurComplementSetSubMatrices()
244 PetscCall(MatSetSizes(S, A10->rmap->n, A01->cmap->n, A10->rmap->N, A01->cmap->N)); in MatSchurComplementSetSubMatrices()
376 …>rmap->n == A00->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local rows of A00 %" PetscInt_FMT "… in MatSchurComplementUpdateSubMatrices()
377 …rmap->n == Ap00->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local rows of A00 %" PetscInt_FMT "… in MatSchurComplementUpdateSubMatrices()
378 …rmap->n == Ap00->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local rows of Ap00 %" PetscInt_FMT … in MatSchurComplementUpdateSubMatrices()
[all …]
|
| /petsc/src/mat/impls/sell/mpi/ |
| H A D | mmsell.c | 42 PetscCall(MatSetSizes(Bnew, B->rmap->n, N, B->rmap->n, N)); in MatDisAssemble_MPISELL()
56 totalslices = PetscCeilInt(B->rmap->n, Bsell->sliceheight); in MatDisAssemble_MPISELL()
90 totalslices = PetscCeilInt(sell->B->rmap->n, B->sliceheight); in MatSetUpMultiply_MPISELL()
95 PetscCall(PetscHMapICreateWithSize(sell->B->rmap->n, &gid1_lid1)); in MatSetUpMultiply_MPISELL()
209 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd)); in MatMPISELLDiagonalScaleLocalSetUp()
211 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPISELLDiagonalScaleLocalSetUp()
212 if (inA->rmap->mapping->indices[i] >= cstart && inA->rmap->mapping->indices[i] < cend) { in MatMPISELLDiagonalScaleLocalSetUp()
214 r_rmapd[i] = inA->rmap->mapping->indices[i] + 1; in MatMPISELLDiagonalScaleLocalSetUp()
219 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPISELLDiagonalScaleLocalSetUp()
226 no = inA->rmap->mapping->n - nt; in MatMPISELLDiagonalScaleLocalSetUp()
[all …]
|
| /petsc/src/mat/impls/aij/seq/hdf5/ |
| H A D | aijhdf5.c | 16 PetscInt bs = mat->rmap->bs; in MatLoad_AIJ_HDF5()
69 tmp = mat->rmap; in MatLoad_AIJ_HDF5()
70 mat->rmap = mat->cmap; in MatLoad_AIJ_HDF5()
75 …rmap->N < 0 || mat->rmap->N == M, PETSC_COMM_SELF, PETSC_ERR_FILE_UNEXPECTED, "Inconsistent # of r… in MatLoad_AIJ_HDF5()
79 mat->rmap->N = M; in MatLoad_AIJ_HDF5()
81 PetscCall(PetscLayoutSetUp(mat->rmap)); in MatLoad_AIJ_HDF5()
83 m = mat->rmap->n; in MatLoad_AIJ_HDF5()
99 m = mat->rmap->n; in MatLoad_AIJ_HDF5()
|
| /petsc/src/mat/graphops/partition/impls/pmetis/ |
| H A D | pmetis.c | 57 if (amat->rmap->n > 0) bs = mat->rmap->n / amat->rmap->n; in MatPartitioningApply_Parmetis_Private()
65 PetscInt *vtxdist = pmat->rmap->range; in MatPartitioningApply_Parmetis_Private()
77 for (i = 0; i < pmat->rmap->n; i++) { in MatPartitioningApply_Parmetis_Private()
82 PetscCall(PetscMalloc1(pmat->rmap->n, &locals)); in MatPartitioningApply_Parmetis_Private()
89 for (i = 0; i < pmat->rmap->n; i++) locals[i] = part_indices[i * bs]; in MatPartitioningApply_Parmetis_Private()
129 PetscCall(PetscMalloc1(pmat->rmap->n, &NDorder)); in MatPartitioningApply_Parmetis_Private()
135 for (i = 0; i < pmat->rmap->n; i++) { in MatPartitioningApply_Parmetis_Private()
171 PetscCall(PetscMalloc1(bs * pmat->rmap->n, &newlocals)); in MatPartitioningApply_Parmetis_Private()
172 for (i = 0; i < pmat->rmap->n; i++) { in MatPartitioningApply_Parmetis_Private()
176 …PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)part), bs * pmat->rmap->n, newlocals, PETSC… in MatPartitioningApply_Parmetis_Private()
[all …]
|
| /petsc/src/mat/graphops/partition/impls/scotch/ |
| H A D | scotch.c | 284 nold = mat->rmap->n; in MatPartitioningApply_PTScotch_Private()
286 if (mat->rmap->n > 0) bs = nold / mat->rmap->n; in MatPartitioningApply_PTScotch_Private()
293 PetscCall(PetscMalloc1(mat->rmap->n + 1, &locals)); in MatPartitioningApply_PTScotch_Private()
304 PetscCall(PetscMalloc1(mat->rmap->n, &NDorder)); in MatPartitioningApply_PTScotch_Private()
306 SCOTCH_ParMETIS_V3_NodeND(mat->rmap->range, adj->i, adj->j, &base, NULL, NDorder, sizes, &comm); in MatPartitioningApply_PTScotch_Private()
308 for (i = 0; i < mat->rmap->n; i++) { in MatPartitioningApply_PTScotch_Private()
346 vertlocnbr = mat->rmap->range[rank + 1] - mat->rmap->range[rank]; in MatPartitioningApply_PTScotch_Private()
352 for (p = 0; !mat->rmap->range[p + 1] && p < nparts; ++p); in MatPartitioningApply_PTScotch_Private()
353 distributed = (mat->rmap->range[p + 1] == mat->rmap->N) ? PETSC_FALSE : PETSC_TRUE; in MatPartitioningApply_PTScotch_Private()
410 PetscCall(PetscMalloc1(bs * mat->rmap->n, &newlocals)); in MatPartitioningApply_PTScotch_Private()
[all …]
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | mattransposematmult.c | 40 PetscCall(MatSetSizes(C, A->rmap->n, B->rmap->N, A->rmap->n, B->rmap->N)); in MatTMatTMultSymbolic_SeqAIJ_SeqDense()
41 dofm = B->rmap->n; in MatTMatTMultSymbolic_SeqAIJ_SeqDense()
64 PetscInt i, j, m = A->rmap->n, n = A->cmap->n, blda, clda; in MatTMatTMultNumeric_SeqAIJ_SeqDense()
106 if (blda == B->rmap->n) { in MatTMatTMultNumeric_SeqAIJ_SeqDense()
110 PetscInt bm = B->rmap->n; in MatTMatTMultNumeric_SeqAIJ_SeqDense()
120 if (blda == B->rmap->n) PetscCall(VecResetArray(ct)); in MatTMatTMultNumeric_SeqAIJ_SeqDense()
|
| /petsc/src/mat/utils/ |
| H A D | zerorows.c | 7 PetscInt *owners = A->rmap->range; in MatZeroRowsMapLocal_Private()
8 PetscInt n = A->rmap->n; in MatZeroRowsMapLocal_Private()
23 …>= 0 && A->rmap->N > idx, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row %" PetscInt_FMT " out of… in MatZeroRowsMapLocal_Private()
25 PetscCall(PetscLayoutFindOwner(A->rmap, idx, &p)); in MatZeroRowsMapLocal_Private()
|
| H A D | gcreate.c | 10 …rmap->bs <= 0 || mat->rmap->bs == rbs, PetscObjectComm((PetscObject)mat), PETSC_ERR_SUP, "Cannot c… in MatSetBlockSizes_Default()
104 PetscCall(PetscLayoutCreate(comm, &B->rmap)); in MatCreate()
233 …PetscCheck((A->rmap->n < 0 || A->rmap->N < 0) || (A->rmap->n == m && (M <= 0 || A->rmap->N == M)),… in MatSetSizes()
234 A->rmap->n, A->rmap->N); in MatSetSizes()
237 A->rmap->n = m; in MatSetSizes()
239 A->rmap->N = M > -1 ? M : A->rmap->N; in MatSetSizes()
291 PetscCall(PetscLayoutSetBlockSize(B->rmap, newbs)); in MatSetFromOptions()
326 if (flg && B->rmap->n < bind_below) PetscCall(MatBindToCPU(B, PETSC_TRUE)); in MatSetFromOptions()
362 PetscCall(PetscLayoutSetUp(A->rmap)); in MatXAIJSetPreallocation()
472 PetscCall(PetscLayoutDestroy(&A->rmap)); in MatHeaderMerge()
[all …]
|
| /petsc/src/mat/impls/htool/ |
| H A D | htool.cxx | 63 …PetscCall(MatCreateDense(PETSC_COMM_SELF, A->rmap->n, A->rmap->n, A->rmap->n, A->rmap->n, nullptr,… in MatGetDiagonalBlock_Htool()
137 …if (idx[j] + k < A->rmap->N && idx[j] + k < A->cmap->N) set.insert(idx[j] + k); /* do not insert i… in MatIncreaseOverlap_Htool()
184 it = std::lower_bound(idxr, idxr + nrow, A->rmap->rstart); in MatCreateSubMatrices_Htool()
185 if (it != idxr + nrow && *it == A->rmap->rstart) { /* rmap->rstart in IS? */ in MatCreateSubMatrices_Htool()
186 …if (std::distance(idxr, it) + A->rmap->n <= nrow) { /* long enough IS to store the local diagonal … in MatCreateSubMatrices_Htool()
187 for (PetscInt j = 0; j < A->rmap->n && flg; ++j) in MatCreateSubMatrices_Htool()
188 if (PetscUnlikely(it[j] != A->rmap->rstart + j)) flg = PETSC_FALSE; in MatCreateSubMatrices_Htool()
198 …for (PetscInt k = 0; k < A->rmap->n; ++k) PetscCall(PetscArraycpy(ptr + (m + k) * nrow + m, copy +… in MatCreateSubMatrices_Htool()
204 … PetscCall(MatCreateDense(PETSC_COMM_SELF, A->rmap->n, m, A->rmap->n, m, ptr + m, &B)); in MatCreateSubMatrices_Htool()
206 … PetscCall(MatCreateDense(PETSC_COMM_SELF, m, A->rmap->n, m, A->rmap->n, ptr + m * nrow, &BT)); in MatCreateSubMatrices_Htool()
[all …]
|
| /petsc/src/mat/impls/dense/mpi/ |
| H A D | mpidense.c | 59 PetscInt j, lda, rstart = A->rmap->rstart, rend = A->rmap->rend, rend2; in MatShift_MPIDense()
77 PetscInt lrow, rstart = A->rmap->rstart, rend = A->rmap->rend; in MatGetRow_MPIDense()
89 PetscInt lrow, rstart = A->rmap->rstart, rend = A->rmap->rend; in MatRestoreRow_MPIDense()
101 PetscInt m = A->rmap->n, rstart = A->rmap->rstart; in MatGetDiagonalBlock_MPIDense()
136 PetscInt i, j, rstart = mat->rmap->rstart, rend = mat->rmap->rend, row; in MatSetValues_MPIDense()
142 PetscCheck(idxm[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large"); in MatSetValues_MPIDense()
169 PetscInt i, j, rstart = mat->rmap->rstart, rend = mat->rmap->rend, row; in MatGetValues_MPIDense()
174 PetscCheck(idxm[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large"); in MatGetValues_MPIDense()
203 PetscCall(PetscLayoutSetUp(A->rmap)); in MatDenseSetLDA_MPIDense()
206 PetscCall(MatSetSizes(a->A, A->rmap->n, A->cmap->N, A->rmap->n, A->cmap->N)); in MatDenseSetLDA_MPIDense()
[all …]
|
| H A D | mmdense.c | 56 PetscInt N = C->cmap->N, rstart = C->rmap->rstart, count; in MatCreateSubMatrices_MPIDense_Local()
61 PetscInt is_no, jmax, **rmap, *rmap_i; in MatCreateSubMatrices_MPIDense_Local() local
74 m = C->rmap->N; in MatCreateSubMatrices_MPIDense_Local()
97 jmax = C->rmap->range[i + 1]; in MatCreateSubMatrices_MPIDense_Local()
277 …PetscCheck(!(submats[i]->rmap->n != nrow[i]) && !(submats[i]->cmap->n != ncol[i]), PETSC_COMM_SELF… in MatCreateSubMatrices_MPIDense_Local()
278 PetscCall(PetscArrayzero(mat->v, submats[i]->rmap->n * submats[i]->cmap->n)); in MatCreateSubMatrices_MPIDense_Local()
320 PetscCall(PetscMalloc1(ismax, &rmap)); in MatCreateSubMatrices_MPIDense_Local()
321 PetscCall(PetscCalloc1(ismax * C->rmap->N, &rmap[0])); in MatCreateSubMatrices_MPIDense_Local()
322 for (PetscInt i = 1; i < ismax; i++) rmap[i] = rmap[i - 1] + C->rmap->N; in MatCreateSubMatrices_MPIDense_Local()
324 rmap_i = rmap[i]; in MatCreateSubMatrices_MPIDense_Local()
[all …]
|
| /petsc/src/mat/impls/aij/seq/matlab/ |
| H A D | aijmatlab.c | 17 mat = mxCreateSparse(B->cmap->n, B->rmap->n, aij->nz, mxREAL); in MatSeqAIJToMatlab()
23 for (i = 0; i < B->rmap->n + 1; i++) ii[i] = aij->i[i]; in MatSeqAIJToMatlab()
55 if (mat->rmap->n < 0 && mat->cmap->n < 0) { in MatSeqAIJFromMatlab()
60 …cCheck(mat->rmap->n == n, PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot change size of PETSc matrix %" P… in MatSeqAIJFromMatlab()
68 PetscCall(PetscShmgetAllocateArray(mat->rmap->n + 1, sizeof(PetscInt), (void **)&aij->i)); in MatSeqAIJFromMatlab()
80 for (k = 0; k < mat->rmap->n; k++) aij->ilen[k] = aij->imax[k] = aij->i[k + 1] - aij->i[k]; in MatSeqAIJFromMatlab()
162 PetscCheck(A->cmap->N == A->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "matrix must be square"); in MatLUFactorSymbolic_Matlab()
189 PetscCheck(A->cmap->N == A->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "matrix must be square"); in MatGetFactor_seqaij_matlab()
191 PetscCall(MatSetSizes(*F, A->rmap->n, A->cmap->n, A->rmap->n, A->cmap->n)); in MatGetFactor_seqaij_matlab()
|
| /petsc/src/mat/impls/dense/seq/ |
| H A D | dense.c | 16 PetscInt j, k, n = A->rmap->n; in MatSeqDenseSymmetrize_Private()
20 …PetscCheck(A->rmap->n == A->cmap->n, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Cannot symme… in MatSeqDenseSymmetrize_Private()
41 if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(PETSC_SUCCESS); in MatSeqDenseInvertFactors_Private()
94 PetscInt m = l->lda, n = A->cmap->n, r = A->rmap->n, i, j; in MatZeroRowsColumns_SeqDense()
102 …rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row %" PetscInt_FMT " requested to be zeroed … in MatZeroRowsColumns_SeqDense()
112 …PetscCheck(A->rmap->n == A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_SUP, "Only coded for square matric… in MatZeroRowsColumns_SeqDense()
138 …PetscCheck(A->rmap->n == A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_SUP, "Only coded for square matric… in MatZeroRowsColumns_SeqDense()
153 PetscInt *ai = a->i, *aj = a->j, m = A->rmap->N, n = A->cmap->N, i; in MatConvert_SeqAIJ_SeqDense()
205 PetscCall(PetscCalloc3(A->rmap->n, &rows, A->rmap->n, &nnz, A->rmap->n, &vals)); in MatConvert_SeqDense_SeqAIJ()
208 PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); in MatConvert_SeqDense_SeqAIJ()
[all …]
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | aijbaij.c | 9 PetscInt bs = A->rmap->bs, *ai = a->i, *aj = a->j, n = A->rmap->N / bs, i, j, k; in MatConvert_SeqBAIJ_SeqAIJ()
25 PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); in MatConvert_SeqBAIJ_SeqAIJ()
26 PetscCall(MatSetBlockSizes(B, A->rmap->bs, A->cmap->bs)); in MatConvert_SeqBAIJ_SeqAIJ()
62 PetscInt m, n, bs = A->rmap->bs; in MatConvert_SeqAIJ_SeqBAIJ_Preallocate()
97 PetscInt m = A->rmap->N, n = A->cmap->N, *rowlengths, bs = A->rmap->bs; in MatConvert_SeqAIJ_SeqBAIJ()
|
| /petsc/src/mat/impls/aij/mpi/mpiviennacl/ |
| H A D | mpiaijviennacl.cxx | 12 PetscCall(PetscLayoutSetUp(B->rmap)); in MatMPIAIJSetPreallocation_MPIAIJViennaCL()
17 PetscCall(MatSetSizes(b->A, B->rmap->n, B->cmap->n, B->rmap->n, B->cmap->n)); in MatMPIAIJSetPreallocation_MPIAIJViennaCL()
20 PetscCall(MatSetSizes(b->B, B->rmap->n, B->cmap->N, B->rmap->n, B->cmap->N)); in MatMPIAIJSetPreallocation_MPIAIJViennaCL()
|