Lines Matching refs:dim
46 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatMult_SeqFFTW() local
61 …fftw->p_forward = fftw_plan_dft_1d(dim[0], (fftw_complex *)x_array, (fftw_complex *)y_array, FFTW_… in MatMult_SeqFFTW()
63 …fftw->p_forward = fftw_plan_dft_r2c_1d(dim[0], (double *)x_array, (fftw_complex *)y_array, fftw->p… in MatMult_SeqFFTW()
68 …fftw->p_forward = fftw_plan_dft_2d(dim[0], dim[1], (fftw_complex *)x_array, (fftw_complex *)y_arra… in MatMult_SeqFFTW()
70 …fftw->p_forward = fftw_plan_dft_r2c_2d(dim[0], dim[1], (double *)x_array, (fftw_complex *)y_array,… in MatMult_SeqFFTW()
75 …fftw->p_forward = fftw_plan_dft_3d(dim[0], dim[1], dim[2], (fftw_complex *)x_array, (fftw_complex … in MatMult_SeqFFTW()
77 …fftw->p_forward = fftw_plan_dft_r2c_3d(dim[0], dim[1], dim[2], (double *)x_array, (fftw_complex *)… in MatMult_SeqFFTW()
85 iodims[ndim - 1].n = (ptrdiff_t)dim[ndim - 1]; in MatMult_SeqFFTW()
88 iodims[i].n = (ptrdiff_t)dim[i]; in MatMult_SeqFFTW()
95 iodims[ndim - 1].n = (int)dim[ndim - 1]; in MatMult_SeqFFTW()
98 iodims[i].n = (int)dim[i]; in MatMult_SeqFFTW()
106 …fftw->p_forward = fftw_plan_dft_r2c(ndim, (int *)dim, (double *)x_array, (fftw_complex *)y_array, … in MatMult_SeqFFTW()
141 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatMultTranspose_SeqFFTW() local
164 …fftw->p_backward = fftw_plan_dft_1d(dim[0], (fftw_complex *)x_array, (fftw_complex *)y_array, FFTW… in MatMultTranspose_SeqFFTW()
166 …fftw->p_backward = fftw_plan_dft_c2r_1d(dim[0], (fftw_complex *)x_array, (double *)y_array, fftw->… in MatMultTranspose_SeqFFTW()
171 …fftw->p_backward = fftw_plan_dft_2d(dim[0], dim[1], (fftw_complex *)x_array, (fftw_complex *)y_arr… in MatMultTranspose_SeqFFTW()
173 …fftw->p_backward = fftw_plan_dft_c2r_2d(dim[0], dim[1], (fftw_complex *)x_array, (double *)y_array… in MatMultTranspose_SeqFFTW()
178 …fftw->p_backward = fftw_plan_dft_3d(dim[0], dim[1], dim[2], (fftw_complex *)x_array, (fftw_complex… in MatMultTranspose_SeqFFTW()
180 …fftw->p_backward = fftw_plan_dft_c2r_3d(dim[0], dim[1], dim[2], (fftw_complex *)x_array, (double *… in MatMultTranspose_SeqFFTW()
191 …fftw->p_backward = fftw_plan_dft_c2r((int)ndim, (int *)dim, (fftw_complex *)x_array, (double *)y_a… in MatMultTranspose_SeqFFTW()
226 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatMult_MPIFFTW() local
244 …fftw->p_forward = fftw_mpi_plan_dft_1d(dim[0], (fftw_complex *)x_array, (fftw_complex *)y_array, c… in MatMult_MPIFFTW()
251 …fftw->p_forward = fftw_mpi_plan_dft_2d(dim[0], dim[1], (fftw_complex *)x_array, (fftw_complex *)y_… in MatMult_MPIFFTW()
253 …fftw->p_forward = fftw_mpi_plan_dft_r2c_2d(dim[0], dim[1], (double *)x_array, (fftw_complex *)y_ar… in MatMult_MPIFFTW()
258 …fftw->p_forward = fftw_mpi_plan_dft_3d(dim[0], dim[1], dim[2], (fftw_complex *)x_array, (fftw_comp… in MatMult_MPIFFTW()
260 …fftw->p_forward = fftw_mpi_plan_dft_r2c_3d(dim[0], dim[1], dim[2], (double *)x_array, (fftw_comple… in MatMult_MPIFFTW()
297 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatMultTranspose_MPIFFTW() local
315 …fftw->p_backward = fftw_mpi_plan_dft_1d(dim[0], (fftw_complex *)x_array, (fftw_complex *)y_array, … in MatMultTranspose_MPIFFTW()
322 …fftw->p_backward = fftw_mpi_plan_dft_2d(dim[0], dim[1], (fftw_complex *)x_array, (fftw_complex *)y… in MatMultTranspose_MPIFFTW()
324 …fftw->p_backward = fftw_mpi_plan_dft_c2r_2d(dim[0], dim[1], (fftw_complex *)x_array, (double *)y_a… in MatMultTranspose_MPIFFTW()
329 …fftw->p_backward = fftw_mpi_plan_dft_3d(dim[0], dim[1], dim[2], (fftw_complex *)x_array, (fftw_com… in MatMultTranspose_MPIFFTW()
331 …fftw->p_backward = fftw_mpi_plan_dft_c2r_3d(dim[0], dim[1], dim[2], (fftw_complex *)x_array, (doub… in MatMultTranspose_MPIFFTW()
503 PetscInt ndim = fft->ndim, *dim = fft->dim; 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()
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()
555 PetscCall(PetscIntCast(2 * dim[0] * (dim[1] / 2 + 1), &N1)); 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 …alloc_local = fftw_mpi_local_size_3d_transposed(dim[0], dim[1], dim[2] / 2 + 1, comm, &local_n0, &… in MatCreateVecsFFTW_FFTW()
607 PetscCall(PetscIntCast(2 * dim[0] * dim[1] * (dim[2] / 2 + 1), &N1)); 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()
778 PetscInt ndim = fft->ndim, *dim = fft->dim, n1; 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()
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()
816 PetscCall(PetscIntCast(local_0_start * dim[1], &nstart)); 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()
832 if (dim[1] % 2 == 0) { in VecScatterPetscToFFTW_FFTW()
833 NM = dim[1] + 2; in VecScatterPetscToFFTW_FFTW()
835 NM = dim[1] + 1; in VecScatterPetscToFFTW_FFTW()
838 for (j = 0; j < dim[1]; j++) { in VecScatterPetscToFFTW_FFTW()
839 tempindx = i * dim[1] + j; in VecScatterPetscToFFTW_FFTW()
842 PetscCall(PetscIntCast(local_0_start * dim[1] + tempindx, &indx1[tempindx])); 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()
867 PetscCall(PetscIntCast(local_0_start * dim[1] * dim[2], &nstart)); 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()
885 if (dim[2] % 2 == 0) NM = dim[2] + 2; in VecScatterPetscToFFTW_FFTW()
886 else NM = dim[2] + 1; in VecScatterPetscToFFTW_FFTW()
889 for (j = 0; j < dim[1]; j++) { in VecScatterPetscToFFTW_FFTW()
890 for (k = 0; k < dim[2]; k++) { in VecScatterPetscToFFTW_FFTW()
891 tempindx = i * dim[1] * dim[2] + j * dim[2] + k; in VecScatterPetscToFFTW_FFTW()
892 tempindx1 = i * dim[1] * NM + j * NM + k; in VecScatterPetscToFFTW_FFTW()
894 PetscCall(PetscIntCast(local_0_start * dim[1] * dim[2] + tempindx, &indx1[tempindx])); 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()
944 if (dim[ndim - 1] % 2 == 0) NM = 2; in VecScatterPetscToFFTW_FFTW()
951 if (k % dim[ndim - 1] == 0) j += NM; in VecScatterPetscToFFTW_FFTW()
1025 PetscInt ndim = fft->ndim, *dim = fft->dim; 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()
1063 PetscCall(PetscIntCast(local_0_start * dim[1], &nstart)); 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()
1079 if (dim[1] % 2 == 0) NM = dim[1] + 2; in VecScatterFFTWToPetsc_FFTW()
1080 else NM = dim[1] + 1; in VecScatterFFTWToPetsc_FFTW()
1083 for (j = 0; j < dim[1]; j++) { in VecScatterFFTWToPetsc_FFTW()
1084 tempindx = i * dim[1] + j; in VecScatterFFTWToPetsc_FFTW()
1087 PetscCall(PetscIntCast(local_0_start * dim[1] + tempindx, &indx1[tempindx])); 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()
1111 PetscCall(PetscIntCast(local_0_start * dim[1] * dim[2], &nstart)); 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()
1127 if (dim[2] % 2 == 0) NM = dim[2] + 2; in VecScatterFFTWToPetsc_FFTW()
1128 else NM = dim[2] + 1; in VecScatterFFTWToPetsc_FFTW()
1131 for (j = 0; j < dim[1]; j++) { in VecScatterFFTWToPetsc_FFTW()
1132 for (k = 0; k < dim[2]; k++) { in VecScatterFFTWToPetsc_FFTW()
1133 tempindx = i * dim[1] * dim[2] + j * dim[2] + k; in VecScatterFFTWToPetsc_FFTW()
1134 tempindx1 = i * dim[1] * NM + j * NM + k; in VecScatterFFTWToPetsc_FFTW()
1136 PetscCall(PetscIntCast(local_0_start * dim[1] * dim[2] + tempindx, &indx1[tempindx])); in VecScatterFFTWToPetsc_FFTW()
1142 PetscCall(PetscIntCast(local_n0 * dim[1] * dim[2], &n1)); in VecScatterFFTWToPetsc_FFTW()
1185 if (dim[ndim - 1] % 2 == 0) NM = 2; in VecScatterFFTWToPetsc_FFTW()
1192 if (k % dim[ndim - 1] == 0) j += NM; in VecScatterFFTWToPetsc_FFTW()
1227 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatCreate_FFTW() local
1247 pdim[0] = dim[0]; in MatCreate_FFTW()
1249 tot_dim = 2 * dim[0]; in MatCreate_FFTW()
1252 partial_dim *= dim[ctr]; in MatCreate_FFTW()
1253 pdim[ctr] = dim[ctr]; in MatCreate_FFTW()
1255 if (ctr == ndim - 1) tot_dim *= (dim[ctr] / 2 + 1); in MatCreate_FFTW()
1256 else tot_dim *= dim[ctr]; in MatCreate_FFTW()
1283 …fftw_mpi_local_size_1d(dim[0], comm, FFTW_FORWARD, FFTW_ESTIMATE, &local_n0, &local_0_start, &loca… 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()
1298 …PetscCall(MatSetSizes(A, fft->n, fft->n, 2 * dim[0] * (dim[1] / 2 + 1), 2 * dim[0] * (dim[1] / 2 +… 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()
1311 …PetscCall(MatSetSizes(A, fft->n, fft->n, 2 * dim[0] * dim[1] * (dim[2] / 2 + 1), 2 * dim[0] * dim[… in MatCreate_FFTW()
1355 for (ctr = 0; ctr < ndim; ctr++) (fftw->dim_fftw)[ctr] = dim[ctr]; in MatCreate_FFTW()