Lines Matching refs:N1
10 const ptrdiff_t N0 = 2056, N1 = 2056; in main() local
35 …alloc_local = fftw_mpi_local_size_2d_transposed(N0, N1 / 2 + 1, PETSC_COMM_WORLD, &local_n0, &loca… in main()
50 N = 2 * N0 * (N1 / 2 + 1); in main()
51 N_factor = N0 * N1; in main()
52 n = 2 * local_n0 * (N1 / 2 + 1); in main()
89 …fplan = fftw_mpi_plan_dft_r2c_2d(N0, N1, (double *)x_arr, (fftw_complex *)y_arr, PETSC_COMM_WORLD,… in main()
90 …bplan = fftw_mpi_plan_dft_c2r_2d(N0, N1, (fftw_complex *)y_arr, (double *)z_arr, PETSC_COMM_WORLD,… in main()
102 PetscCall(VecSetSizes(ini, local_n0 * N1, N0 * N1)); in main()
103 PetscCall(VecSetSizes(final, local_n0 * N1, N0 * N1)); in main()
107 if (N1 % 2 == 0) { in main()
108 NM = N1 + 2; in main()
110 NM = N1 + 1; in main()
115 PetscCall(PetscMalloc1(local_n0 * N1, &indx3)); in main()
116 PetscCall(PetscMalloc1(local_n0 * N1, &indx4)); in main()
118 for (j = 0; j < N1; j++) { in main()
119 tempindx = i * N1 + j; in main()
122 indx3[tempindx] = local_0_start * N1 + tempindx; in main()
129 …PetscCall(PetscMalloc2(local_n0 * N1, &x_arr, local_n0 * N1, &y_arr)); /* arr must be allocated fo… in main()
130 PetscCall(VecGetValues(fin, local_n0 * N1, indx4, (PetscScalar *)x_arr)); in main()
131 PetscCall(VecSetValues(ini, local_n0 * N1, indx3, x_arr, INSERT_VALUES)); in main()
136 PetscCall(VecGetValues(fout1, local_n0 * N1, indx4, y_arr)); in main()
137 PetscCall(VecSetValues(final, local_n0 * N1, indx3, y_arr, INSERT_VALUES)); in main()