Lines Matching refs:local_n0
504 ptrdiff_t alloc_local, local_n0, local_0_start; in MatCreateVecsFFTW_FFTW() local
521 …alloc_local = fftw_mpi_local_size_1d(dim[0], comm, FFTW_FORWARD, FFTW_ESTIMATE, &local_n0, &local_… in MatCreateVecsFFTW_FFTW()
524 PetscCall(PetscIntCast(local_n0, &n1)); in MatCreateVecsFFTW_FFTW()
540 …alloc_local = fftw_mpi_local_size_1d(dim[0], comm, FFTW_BACKWARD, FFTW_ESTIMATE, &local_n0, &local… in MatCreateVecsFFTW_FFTW()
554 …alloc_local = fftw_mpi_local_size_2d_transposed(dim[0], dim[1] / 2 + 1, comm, &local_n0, &local_0_… in MatCreateVecsFFTW_FFTW()
556 PetscCall(PetscIntCast(2 * local_n0 * (dim[1] / 2 + 1), &n1)); in MatCreateVecsFFTW_FFTW()
580 alloc_local = fftw_mpi_local_size_2d(dim[0], dim[1], comm, &local_n0, &local_0_start); in MatCreateVecsFFTW_FFTW()
606 …ftw_mpi_local_size_3d_transposed(dim[0], dim[1], dim[2] / 2 + 1, comm, &local_n0, &local_0_start, … in MatCreateVecsFFTW_FFTW()
608 PetscCall(PetscIntCast(2 * local_n0 * dim[1] * (dim[2] / 2 + 1), &n1)); in MatCreateVecsFFTW_FFTW()
631 alloc_local = fftw_mpi_local_size_3d(dim[0], dim[1], dim[2], comm, &local_n0, &local_0_start); in MatCreateVecsFFTW_FFTW()
659 … fftw_mpi_local_size_transposed(fftw->ndim_fftw, fftw->dim_fftw, comm, &local_n0, &local_0_start, … in MatCreateVecsFFTW_FFTW()
685 …alloc_local = fftw_mpi_local_size(fftw->ndim_fftw, fftw->dim_fftw, comm, &local_n0, &local_0_start… in MatCreateVecsFFTW_FFTW()
779 ptrdiff_t local_n0, local_0_start; in VecScatterPetscToFFTW_FFTW() local
794 …fftw_mpi_local_size_1d(dim[0], comm, FFTW_FORWARD, FFTW_ESTIMATE, &local_n0, &local_0_start, &loca… in VecScatterPetscToFFTW_FFTW()
796 PetscCall(PetscIntCast(local_n0, &n1)); in VecScatterPetscToFFTW_FFTW()
813 fftw_mpi_local_size_2d(dim[0], dim[1], comm, &local_n0, &local_0_start); in VecScatterPetscToFFTW_FFTW()
815 PetscCall(PetscIntCast(local_n0 * dim[1], &n1)); in VecScatterPetscToFFTW_FFTW()
827 …fftw_mpi_local_size_2d_transposed(dim[0], dim[1] / 2 + 1, comm, &local_n0, &local_0_start, &local_… in VecScatterPetscToFFTW_FFTW()
829 PetscCall(PetscMalloc1((PetscInt)local_n0 * dim[1], &indx1)); in VecScatterPetscToFFTW_FFTW()
830 PetscCall(PetscMalloc1((PetscInt)local_n0 * dim[1], &indx2)); in VecScatterPetscToFFTW_FFTW()
837 for (i = 0; i < local_n0; i++) { in VecScatterPetscToFFTW_FFTW()
847 PetscCall(PetscIntCast(local_n0 * dim[1], &n1)); in VecScatterPetscToFFTW_FFTW()
864 fftw_mpi_local_size_3d(dim[0], dim[1], dim[2], comm, &local_n0, &local_0_start); in VecScatterPetscToFFTW_FFTW()
866 PetscCall(PetscIntCast(local_n0 * dim[1] * dim[2], &n1)); in VecScatterPetscToFFTW_FFTW()
880 …fftw_mpi_local_size_3d_transposed(dim[0], dim[1], dim[2] / 2 + 1, comm, &local_n0, &local_0_start,… in VecScatterPetscToFFTW_FFTW()
882 PetscCall(PetscMalloc1((PetscInt)local_n0 * dim[1] * dim[2], &indx1)); in VecScatterPetscToFFTW_FFTW()
883 PetscCall(PetscMalloc1((PetscInt)local_n0 * dim[1] * dim[2], &indx2)); in VecScatterPetscToFFTW_FFTW()
888 for (i = 0; i < local_n0; i++) { in VecScatterPetscToFFTW_FFTW()
900 … PetscCall(ISCreateGeneral(comm, local_n0 * dim[1] * dim[2], indx1, PETSC_COPY_VALUES, &list1)); in VecScatterPetscToFFTW_FFTW()
901 … PetscCall(ISCreateGeneral(comm, local_n0 * dim[1] * dim[2], indx2, PETSC_COPY_VALUES, &list2)); in VecScatterPetscToFFTW_FFTW()
915 fftw_mpi_local_size(fftw->ndim_fftw, fftw->dim_fftw, comm, &local_n0, &local_0_start); in VecScatterPetscToFFTW_FFTW()
917 PetscCall(PetscIntCast(local_n0 * (fftw->partial_dim), &n1)); in VecScatterPetscToFFTW_FFTW()
935 …fftw_mpi_local_size_transposed(fftw->ndim_fftw, fftw->dim_fftw, comm, &local_n0, &local_0_start, &… in VecScatterPetscToFFTW_FFTW()
941 PetscCall(PetscMalloc1((PetscInt)local_n0 * partial_dim, &indx1)); in VecScatterPetscToFFTW_FFTW()
942 PetscCall(PetscMalloc1((PetscInt)local_n0 * partial_dim, &indx2)); in VecScatterPetscToFFTW_FFTW()
948 for (i = 0, k = 1; i < (PetscInt)local_n0 * partial_dim; i++, k++) { in VecScatterPetscToFFTW_FFTW()
954 PetscCall(ISCreateGeneral(comm, local_n0 * partial_dim, indx1, PETSC_COPY_VALUES, &list1)); in VecScatterPetscToFFTW_FFTW()
955 PetscCall(ISCreateGeneral(comm, local_n0 * partial_dim, indx2, PETSC_COPY_VALUES, &list2)); in VecScatterPetscToFFTW_FFTW()
1026 ptrdiff_t local_n0, local_0_start; in VecScatterFFTWToPetsc_FFTW() local
1041 …fftw_mpi_local_size_1d(dim[0], comm, FFTW_BACKWARD, FFTW_ESTIMATE, &local_n0, &local_0_start, &loc… in VecScatterFFTWToPetsc_FFTW()
1060 fftw_mpi_local_size_2d(dim[0], dim[1], comm, &local_n0, &local_0_start); in VecScatterFFTWToPetsc_FFTW()
1062 PetscCall(PetscIntCast(local_n0 * dim[1], &n1)); in VecScatterFFTWToPetsc_FFTW()
1074 …fftw_mpi_local_size_2d_transposed(dim[0], dim[1] / 2 + 1, comm, &local_n0, &local_0_start, &local_… in VecScatterFFTWToPetsc_FFTW()
1076 PetscCall(PetscMalloc1((PetscInt)local_n0 * dim[1], &indx1)); in VecScatterFFTWToPetsc_FFTW()
1077 PetscCall(PetscMalloc1((PetscInt)local_n0 * dim[1], &indx2)); in VecScatterFFTWToPetsc_FFTW()
1082 for (i = 0; i < local_n0; i++) { in VecScatterFFTWToPetsc_FFTW()
1092 PetscCall(PetscIntCast(local_n0 * dim[1], &n1)); in VecScatterFFTWToPetsc_FFTW()
1108 fftw_mpi_local_size_3d(dim[0], dim[1], dim[2], comm, &local_n0, &local_0_start); in VecScatterFFTWToPetsc_FFTW()
1110 PetscCall(PetscIntCast(local_n0 * dim[1] * dim[2], &n1)); in VecScatterFFTWToPetsc_FFTW()
1122 …fftw_mpi_local_size_3d_transposed(dim[0], dim[1], dim[2] / 2 + 1, comm, &local_n0, &local_0_start,… in VecScatterFFTWToPetsc_FFTW()
1124 PetscCall(PetscMalloc1((PetscInt)local_n0 * dim[1] * dim[2], &indx1)); in VecScatterFFTWToPetsc_FFTW()
1125 PetscCall(PetscMalloc1((PetscInt)local_n0 * dim[1] * dim[2], &indx2)); in VecScatterFFTWToPetsc_FFTW()
1130 for (i = 0; i < local_n0; i++) { in VecScatterFFTWToPetsc_FFTW()
1142 PetscCall(PetscIntCast(local_n0 * dim[1] * dim[2], &n1)); in VecScatterFFTWToPetsc_FFTW()
1158 fftw_mpi_local_size(fftw->ndim_fftw, fftw->dim_fftw, comm, &local_n0, &local_0_start); in VecScatterFFTWToPetsc_FFTW()
1160 PetscCall(PetscIntCast(local_n0 * (fftw->partial_dim), &n1)); in VecScatterFFTWToPetsc_FFTW()
1176 …fftw_mpi_local_size_transposed(fftw->ndim_fftw, fftw->dim_fftw, comm, &local_n0, &local_0_start, &… in VecScatterFFTWToPetsc_FFTW()
1182 PetscCall(PetscMalloc1((PetscInt)local_n0 * partial_dim, &indx1)); in VecScatterFFTWToPetsc_FFTW()
1183 PetscCall(PetscMalloc1((PetscInt)local_n0 * partial_dim, &indx2)); in VecScatterFFTWToPetsc_FFTW()
1189 for (i = 0, k = 1; i < (PetscInt)local_n0 * partial_dim; i++, k++) { in VecScatterFFTWToPetsc_FFTW()
1195 PetscCall(PetscIntCast(local_n0 * partial_dim, &n1)); in VecScatterFFTWToPetsc_FFTW()
1270 ptrdiff_t local_n0, local_0_start, local_n1, local_1_start; in MatCreate_FFTW() local
1283 …fftw_mpi_local_size_1d(dim[0], comm, FFTW_FORWARD, FFTW_ESTIMATE, &local_n0, &local_0_start, &loca… in MatCreate_FFTW()
1284 PetscCall(PetscIntCast(local_n0, &fft->n)); in MatCreate_FFTW()
1291 fftw_mpi_local_size_2d(dim[0], dim[1], comm, &local_n0, &local_0_start); in MatCreate_FFTW()
1292 fft->n = (PetscInt)local_n0 * dim[1]; in MatCreate_FFTW()
1295 …fftw_mpi_local_size_2d_transposed(dim[0], dim[1] / 2 + 1, comm, &local_n0, &local_0_start, &local_… in MatCreate_FFTW()
1297 fft->n = 2 * (PetscInt)local_n0 * (dim[1] / 2 + 1); in MatCreate_FFTW()
1303 fftw_mpi_local_size_3d(dim[0], dim[1], dim[2], comm, &local_n0, &local_0_start); in MatCreate_FFTW()
1305 fft->n = (PetscInt)local_n0 * dim[1] * dim[2]; in MatCreate_FFTW()
1308 …fftw_mpi_local_size_3d_transposed(dim[0], dim[1], dim[2] / 2 + 1, comm, &local_n0, &local_0_start,… in MatCreate_FFTW()
1310 fft->n = 2 * (PetscInt)local_n0 * dim[1] * (dim[2] / 2 + 1); in MatCreate_FFTW()
1316 fftw_mpi_local_size(ndim, pdim, comm, &local_n0, &local_0_start); in MatCreate_FFTW()
1318 PetscCall(PetscIntCast(local_n0 * partial_dim, &fft->n)); in MatCreate_FFTW()
1325 …fftw_mpi_local_size_transposed(ndim, pdim, comm, &local_n0, &local_0_start, &local_n1, &local_1_st… in MatCreate_FFTW()
1327 … PetscCall(PetscIntCast(2 * (PetscInt)local_n0 * partial_dim * pdim[ndim - 1] / temp, &fft->n)); in MatCreate_FFTW()