Lines Matching refs:nidx
36 PetscInt *nidx = NULL; in ISAdjustForBlockSize() local
50 PetscCall(PetscMalloc1(n, &nidx)); in ISAdjustForBlockSize()
54 if (!PetscBTLookupSet(bt, idx[j] / bs - Nmin)) nidx[n++] = idx[j] / bs; in ISAdjustForBlockSize()
61 PetscCall(ISCreateBlock(comm, bs, n, nidx, PETSC_OWN_POINTER, is + i)); in ISAdjustForBlockSize()
88 static PetscErrorCode MatIncreaseOverlap_MPIAIJ_Once_Scalable(Mat mat, PetscInt nidx, IS is[]) in MatIncreaseOverlap_MPIAIJ_Once_Scalable() argument
109 PetscCall(PetscMalloc2(nidx, (PetscInt ***)&indices, nidx, &length)); in MatIncreaseOverlap_MPIAIJ_Once_Scalable()
111 for (PetscInt i = 0; i < nidx; i++) { in MatIncreaseOverlap_MPIAIJ_Once_Scalable()
120 for (PetscInt i = 0; i < nidx; i++) { in MatIncreaseOverlap_MPIAIJ_Once_Scalable()
148 PetscCall(MatIncreaseOverlap_MPIAIJ_Local_Scalable(mat, nidx, is)); in MatIncreaseOverlap_MPIAIJ_Once_Scalable()
198 …PetscCall(MatIncreaseOverlap_MPIAIJ_Send_Scalable(mat, nidx, nfrom, fromranks, fromsizes, fromdata… in MatIncreaseOverlap_MPIAIJ_Once_Scalable()
221 PetscCall(MatIncreaseOverlap_MPIAIJ_Local_Scalable(mat, nidx, is)); in MatIncreaseOverlap_MPIAIJ_Once_Scalable()
225 PetscCall(MatIncreaseOverlap_MPIAIJ_Receive_Scalable(mat, nidx, is, nrecvrows, todata)); in MatIncreaseOverlap_MPIAIJ_Once_Scalable()
232 static PetscErrorCode MatIncreaseOverlap_MPIAIJ_Receive_Scalable(Mat mat, PetscInt nidx, IS is[], P… in MatIncreaseOverlap_MPIAIJ_Receive_Scalable() argument
241 PetscCall(PetscMalloc1(nidx, &isz)); in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
242 for (i = 0; i < nidx; i++) { in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
247 PetscCall(PetscMalloc2((max_lsize + nrecvs) * nidx, &indices_temp, nidx, &iscomms)); in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
248 for (i = 0; i < nidx; i++) { in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
268 for (i = 0; i < nidx; i++) { in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
280 static PetscErrorCode MatIncreaseOverlap_MPIAIJ_Send_Scalable(Mat mat, PetscInt nidx, PetscMPIInt n… in MatIncreaseOverlap_MPIAIJ_Send_Scalable() argument
310 …PetscCall(PetscCalloc5(max_fszs * nidx, &rows_data_ptr, nidx, &rows_data, nidx, &rows_pos_i, nfrom… in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
311 for (i = 0; i < nidx; i++) rows_data[i] = PetscSafePointerPlusOffset(rows_data_ptr, max_fszs * i); in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
315 PetscCall(PetscArrayzero(rows_pos_i, nidx)); in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
323 for (j = 0; j < nidx; j++) { in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
342 indv_counts[i * nidx + j] = indvc_ij; in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
347 PetscCall(PetscCalloc2(totalrows + nidx * nfrom * 2, &sbdata, 2 * nfrom, &sbsizes)); in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
352 PetscCall(PetscArrayzero(rows_pos_i, nidx)); in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
359 for (j = 0; j < nidx; j++) { in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
360 if (!indv_counts[i * nidx + j]) continue; in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
363 sbdata[totalrows++] = indv_counts[i * nidx + j]; in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
402 static PetscErrorCode MatIncreaseOverlap_MPIAIJ_Local_Scalable(Mat mat, PetscInt nidx, IS is[]) in MatIncreaseOverlap_MPIAIJ_Local_Scalable() argument
430 for (i = 0; i < nidx; i++) { in MatIncreaseOverlap_MPIAIJ_Local_Scalable()