Lines Matching refs:sizes
493 PetscMPIInt *sizes, Ntotal, *displs, NN; in PetscIntViewNumColumns() local
503 PetscCall(PetscMalloc1(size, &sizes)); in PetscIntViewNumColumns()
504 PetscCallMPI(MPI_Gather(&NN, 1, MPI_INT, sizes, 1, MPI_INT, 0, comm)); in PetscIntViewNumColumns()
505 Ntotal = sizes[0]; in PetscIntViewNumColumns()
509 Ntotal += sizes[i]; 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()
515 PetscCall(PetscFree(sizes)); in PetscIntViewNumColumns()
601 PetscMPIInt *sizes, *displs, Ntotal, NN; in PetscRealViewNumColumns() local
611 PetscCall(PetscMalloc1(size, &sizes)); in PetscRealViewNumColumns()
612 PetscCallMPI(MPI_Gather(&NN, 1, MPI_INT, sizes, 1, MPI_INT, 0, comm)); in PetscRealViewNumColumns()
613 Ntotal = sizes[0]; in PetscRealViewNumColumns()
617 Ntotal += sizes[i]; 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()
623 PetscCall(PetscFree(sizes)); in PetscRealViewNumColumns()
713 PetscMPIInt *sizes, Ntotal, *displs, NN; in PetscScalarViewNumColumns() local
723 PetscCall(PetscMalloc1(size, &sizes)); in PetscScalarViewNumColumns()
724 PetscCallMPI(MPI_Gather(&NN, 1, MPI_INT, sizes, 1, MPI_INT, 0, comm)); in PetscScalarViewNumColumns()
725 Ntotal = sizes[0]; in PetscScalarViewNumColumns()
729 Ntotal += sizes[i]; 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()
735 PetscCall(PetscFree(sizes)); in PetscScalarViewNumColumns()