Lines Matching refs:N1
9 ptrdiff_t N0 = 256, N1 = 256, N2 = 256, N3 = 2, dim[4]; in main() local
36 …alloc_local = fftw_mpi_local_size_3d_transposed(N0, N1, N2 / 2 + 1, PETSC_COMM_WORLD, &local_n0, &… in main()
50 N = 2 * N0 * N1 * (N2 / 2 + 1); in main()
51 N_factor = N0 * N1 * N2; in main()
52 n = 2 * local_n0 * N1 * (N2 / 2 + 1); in main()
53 n1 = local_n1 * N0 * 2 * N1; in main()
78 …fplan = fftw_mpi_plan_dft_r2c_3d(N0, N1, N2, (double *)x_arr, (fftw_complex *)y_arr, PETSC_COMM_WO… in main()
79 …bplan = fftw_mpi_plan_dft_c2r_3d(N0, N1, N2, (fftw_complex *)y_arr, (double *)z_arr, PETSC_COMM_WO… in main()
92 VecSetSizes(ini, local_n0 * N1 * N2, N_factor); in main()
93 VecSetSizes(final, local_n0 * N1 * N2, N_factor); in main()
104 PetscCall(PetscMalloc1(local_n0 * N1 * N2, &indx3)); in main()
105 PetscCall(PetscMalloc1(local_n0 * N1 * N2, &indx4)); in main()
107 for (j = 0; j < N1; j++) { in main()
109 tempindx = i * N1 * N2 + j * N2 + k; in main()
110 tempindx1 = i * N1 * NM + j * NM + k; in main()
112 indx3[tempindx] = local_0_start * N1 * N2 + tempindx; in main()
119 VecGetValues(fin, local_n0 * N1 * N2, indx4, x_arr); in main()
120 VecSetValues(ini, local_n0 * N1 * N2, indx3, x_arr, INSERT_VALUES); in main()
124 VecGetValues(fout1, local_n0 * N1 * N2, indx4, y_arr); in main()
125 VecSetValues(final, local_n0 * N1 * N2, indx3, y_arr, INSERT_VALUES); in main()
129 printf("The local index value is %ld from %d", local_n0 * N1 * N2, rank); in main()