Lines Matching refs:partial_dim

23   PetscInt     partial_dim;  member
783 PetscInt i, j, k, partial_dim; in VecScatterPetscToFFTW_FFTW() local
917 PetscCall(PetscIntCast(local_n0 * (fftw->partial_dim), &n1)); in VecScatterPetscToFFTW_FFTW()
918 PetscCall(PetscIntCast(local_0_start * (fftw->partial_dim), &nstart)); in VecScatterPetscToFFTW_FFTW()
939 partial_dim = fftw->partial_dim; 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()
949 indx1[i] = local_0_start * partial_dim + i; 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()
1030 PetscInt i, j, k, partial_dim; in VecScatterFFTWToPetsc_FFTW() local
1160 PetscCall(PetscIntCast(local_n0 * (fftw->partial_dim), &n1)); in VecScatterFFTWToPetsc_FFTW()
1161 PetscCall(PetscIntCast(local_0_start * (fftw->partial_dim), &nstart)); in VecScatterFFTWToPetsc_FFTW()
1180 partial_dim = fftw->partial_dim; 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()
1190 PetscCall(PetscIntCast(local_0_start * partial_dim + i, &indx1[i])); in VecScatterFFTWToPetsc_FFTW()
1195 PetscCall(PetscIntCast(local_n0 * partial_dim, &n1)); in VecScatterFFTWToPetsc_FFTW()
1231 PetscInt p_flag, partial_dim = 1, ctr; in MatCreate_FFTW() local
1252 partial_dim *= dim[ctr]; in MatCreate_FFTW()
1318 PetscCall(PetscIntCast(local_n0 * partial_dim, &fft->n)); in MatCreate_FFTW()
1327 … PetscCall(PetscIntCast(2 * (PetscInt)local_n0 * partial_dim * pdim[ndim - 1] / temp, &fft->n)); in MatCreate_FFTW()
1344 fftw->partial_dim = partial_dim; in MatCreate_FFTW()