Lines Matching refs:displs
493 PetscMPIInt *sizes, Ntotal, *displs, NN; in PetscIntViewNumColumns() local
506 PetscCall(PetscMalloc1(size, &displs)); in PetscIntViewNumColumns()
507 displs[0] = 0; in PetscIntViewNumColumns()
510 displs[i] = displs[i - 1] + sizes[i - 1]; in PetscIntViewNumColumns()
513 PetscCallMPI(MPI_Gatherv(idx, NN, MPIU_INT, array, sizes, displs, MPIU_INT, 0, comm)); in PetscIntViewNumColumns()
516 PetscCall(PetscFree(displs)); in PetscIntViewNumColumns()
601 PetscMPIInt *sizes, *displs, Ntotal, NN; in PetscRealViewNumColumns() local
614 PetscCall(PetscMalloc1(size, &displs)); in PetscRealViewNumColumns()
615 displs[0] = 0; in PetscRealViewNumColumns()
618 displs[i] = displs[i - 1] + sizes[i - 1]; in PetscRealViewNumColumns()
621 PetscCallMPI(MPI_Gatherv(idx, NN, MPIU_REAL, array, sizes, displs, MPIU_REAL, 0, comm)); in PetscRealViewNumColumns()
624 PetscCall(PetscFree(displs)); in PetscRealViewNumColumns()
713 PetscMPIInt *sizes, Ntotal, *displs, NN; in PetscScalarViewNumColumns() local
726 PetscCall(PetscMalloc1(size, &displs)); in PetscScalarViewNumColumns()
727 displs[0] = 0; in PetscScalarViewNumColumns()
730 displs[i] = displs[i - 1] + sizes[i - 1]; in PetscScalarViewNumColumns()
733 …PetscCallMPI(MPI_Gatherv((void *)idx, NN, MPIU_SCALAR, array, sizes, displs, MPIU_SCALAR, 0, comm)… in PetscScalarViewNumColumns()
736 PetscCall(PetscFree(displs)); in PetscScalarViewNumColumns()