Lines Matching refs:idxs

146     const PetscInt *idxs;  in MatGetNonzeroColumnsLocal_Private()  local
152 PetscCall(ISGetIndices(zd, &idxs)); in MatGetNonzeroColumnsLocal_Private()
154 for (i = 0; i < ctd; i++) aux[i] = (idxs[bs * i] + stc) / bs; in MatGetNonzeroColumnsLocal_Private()
155 PetscCall(ISRestoreIndices(zd, &idxs)); in MatGetNonzeroColumnsLocal_Private()
161 const PetscInt *idxs; in MatGetNonzeroColumnsLocal_Private() local
167 PetscCall(ISGetIndices(zo, &idxs)); in MatGetNonzeroColumnsLocal_Private()
169 for (i = 0; i < cto; i++) aux[i + ctd] = garray[idxs[bs * i] / bs]; in MatGetNonzeroColumnsLocal_Private()
170 PetscCall(ISRestoreIndices(zo, &idxs)); in MatGetNonzeroColumnsLocal_Private()
508 const PetscInt *idxs; in MatMPIXAIJComputeLocalToGlobalMapping_Private() local
512 PetscCall(ISGetIndices(workis[j], &idxs)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
513 PetscCall(MatPreallocateSet(i + A->rmap->rstart, s, idxs, dnz, onz)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
535 const PetscInt *idxs; in MatMPIXAIJComputeLocalToGlobalMapping_Private() local
538 PetscCall(ISGetIndices(workis[j], &idxs)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
539 PetscCall(MatSetValues(A2, 1, &row, s, idxs, vals, INSERT_VALUES)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
540 PetscCall(ISRestoreIndices(workis[j], &idxs)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
979 const PetscInt *idxs; in MatConvert_Nest_IS() local
996 PetscCall(ISGetIndices(isrow[i], &idxs)); in MatConvert_Nest_IS()
998 PetscCall(PetscSFBcastBegin(matis->csf, MPIU_INT, idxs, l2gidxs + stl, MPI_REPLACE)); in MatConvert_Nest_IS()
999 PetscCall(PetscSFBcastEnd(matis->csf, MPIU_INT, idxs, l2gidxs + stl, MPI_REPLACE)); in MatConvert_Nest_IS()
1001 PetscCall(PetscSFBcastBegin(matis->sf, MPIU_INT, idxs, l2gidxs + stl, MPI_REPLACE)); in MatConvert_Nest_IS()
1002 PetscCall(PetscSFBcastEnd(matis->sf, MPIU_INT, idxs, l2gidxs + stl, MPI_REPLACE)); in MatConvert_Nest_IS()
1004 PetscCall(ISRestoreIndices(isrow[i], &idxs)); in MatConvert_Nest_IS()
1015 const PetscInt *idxs; in MatConvert_Nest_IS() local
1031 PetscCall(ISGetIndices(iscol[i], &idxs)); in MatConvert_Nest_IS()
1033 PetscCall(PetscSFBcastBegin(matis->sf, MPIU_INT, idxs, l2gidxs + stl, MPI_REPLACE)); in MatConvert_Nest_IS()
1034 PetscCall(PetscSFBcastEnd(matis->sf, MPIU_INT, idxs, l2gidxs + stl, MPI_REPLACE)); in MatConvert_Nest_IS()
1036 PetscCall(PetscSFBcastBegin(matis->csf, MPIU_INT, idxs, l2gidxs + stl, MPI_REPLACE)); in MatConvert_Nest_IS()
1037 PetscCall(PetscSFBcastEnd(matis->csf, MPIU_INT, idxs, l2gidxs + stl, MPI_REPLACE)); in MatConvert_Nest_IS()
1039 PetscCall(ISRestoreIndices(iscol[i], &idxs)); in MatConvert_Nest_IS()
1371 const PetscInt *idxs; in MatCreateSubMatrix_IS() local
1391 PetscCall(ISGetIndices(irow, &idxs)); in MatCreateSubMatrix_IS()
1392 for (i = 0; i < n; i++) PetscCall(VecSetValue(rtest, idxs[i], 1.0, ADD_VALUES)); in MatCreateSubMatrix_IS()
1400 PetscCall(ISRestoreIndices(irow, &idxs)); in MatCreateSubMatrix_IS()
1402 PetscCall(ISGetIndices(icol, &idxs)); in MatCreateSubMatrix_IS()
1403 for (i = 0; i < n; i++) PetscCall(VecSetValue(ltest, idxs[i], 1.0, ADD_VALUES)); in MatCreateSubMatrix_IS()
1411 PetscCall(ISRestoreIndices(icol, &idxs)); in MatCreateSubMatrix_IS()
1438 PetscCall(ISGetIndices(irow, &idxs)); in MatCreateSubMatrix_IS()
1439 PetscCall(PetscLayoutMapLocal(mat->rmap, m, idxs, &ll, &lidxs, &lgidxs)); in MatCreateSubMatrix_IS()
1440 PetscCall(ISRestoreIndices(irow, &idxs)); in MatCreateSubMatrix_IS()
1472 PetscCall(ISGetIndices(icol, &idxs)); in MatCreateSubMatrix_IS()
1473 PetscCall(PetscLayoutMapLocal(mat->cmap, n, idxs, &ll, &lidxs, &lgidxs)); in MatCreateSubMatrix_IS()
1474 PetscCall(ISRestoreIndices(icol, &idxs)); in MatCreateSubMatrix_IS()
2383 const PetscInt *idxs; in ISLocalToGlobalMappingHasRepeatedLocal_Private() local
2390 PetscCall(ISLocalToGlobalMappingGetBlockIndices(map, &idxs)); in ISLocalToGlobalMappingHasRepeatedLocal_Private()
2395 if (idxs[i] < 0) continue; in ISLocalToGlobalMappingHasRepeatedLocal_Private()
2396 PetscCall(PetscHSetIQueryAdd(ht, idxs[i], &missing)); in ISLocalToGlobalMappingHasRepeatedLocal_Private()
2612 const PetscInt *idxs; in MatISFilterL2GMap() local
2621 PetscCall(ISLocalToGlobalMappingGetBlockIndices(map, &idxs)); in MatISFilterL2GMap()
2626 if (idxs[i] < 0) { in MatISFilterL2GMap()
2630 if (!matis->allow_repeated) PetscCall(PetscHSetIQueryAdd(ht, idxs[i], &missing)); in MatISFilterL2GMap()
2632 else nidxs[c++] = idxs[i]; in MatISFilterL2GMap()
2640 PetscCall(ISLocalToGlobalMappingRestoreBlockIndices(map, &idxs)); in MatISFilterL2GMap()
2652 PetscCall(ISGlobalToLocalMappingApplyBlock(*nmap, IS_GTOLM_MASK, n / bs, idxs, NULL, nidxs)); in MatISFilterL2GMap()
2657 PetscCall(ISLocalToGlobalMappingRestoreBlockIndices(map, &idxs)); in MatISFilterL2GMap()
3250 PetscInt N, M, nrl, i, *idxs; in MatGetLocalSubMatrix_IS() local
3263 PetscCall(PetscMalloc1(nrg, &idxs)); in MatGetLocalSubMatrix_IS()
3265 for (i = 0; i < nrl; i++) idxs[i] = rl[i]; in MatGetLocalSubMatrix_IS()
3266 for (i = nrl; i < nrg; i++) idxs[i] = -1; in MatGetLocalSubMatrix_IS()
3269 PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)A), nrg, idxs, PETSC_OWN_POINTER, &is)); in MatGetLocalSubMatrix_IS()
3287 PetscCall(PetscMalloc1(ncg, &idxs)); in MatGetLocalSubMatrix_IS()
3289 for (i = 0; i < ncl; i++) idxs[i] = cl[i]; in MatGetLocalSubMatrix_IS()
3290 for (i = ncl; i < ncg; i++) idxs[i] = -1; in MatGetLocalSubMatrix_IS()
3293 PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)A), ncg, idxs, PETSC_OWN_POINTER, &is)); in MatGetLocalSubMatrix_IS()