Lines Matching refs:fft
33 Mat_FFT *fft = (Mat_FFT *)A->data; in MatMult_SeqFFTW() local
34 Mat_FFTW *fftw = (Mat_FFTW *)fft->data; in MatMult_SeqFFTW()
46 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatMult_SeqFFTW()
137 Mat_FFT *fft = (Mat_FFT *)A->data; in MatMultTranspose_SeqFFTW() local
138 Mat_FFTW *fftw = (Mat_FFTW *)fft->data; in MatMultTranspose_SeqFFTW()
141 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatMultTranspose_SeqFFTW()
222 Mat_FFT *fft = (Mat_FFT *)A->data; in MatMult_MPIFFTW() local
223 Mat_FFTW *fftw = (Mat_FFTW *)fft->data; in MatMult_MPIFFTW()
226 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatMult_MPIFFTW()
293 Mat_FFT *fft = (Mat_FFT *)A->data; in MatMultTranspose_MPIFFTW() local
294 Mat_FFTW *fftw = (Mat_FFTW *)fft->data; in MatMultTranspose_MPIFFTW()
297 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatMultTranspose_MPIFFTW()
365 Mat_FFT *fft = (Mat_FFT *)A->data; in MatDestroy_FFTW() local
366 Mat_FFTW *fftw = (Mat_FFTW *)fft->data; in MatDestroy_FFTW()
373 PetscCall(PetscFree(fft->data)); in MatDestroy_FFTW()
481 Mat_FFT *fft = (Mat_FFT *)A->data; in MatCreateVecsFFTW_FFTW() local
492 if (fin) PetscCall(VecCreateSeq(PETSC_COMM_SELF, fft->N, fin)); in MatCreateVecsFFTW_FFTW()
493 if (fout) PetscCall(VecCreateSeq(PETSC_COMM_SELF, fft->N, fout)); in MatCreateVecsFFTW_FFTW()
494 if (bout) PetscCall(VecCreateSeq(PETSC_COMM_SELF, fft->N, bout)); in MatCreateVecsFFTW_FFTW()
496 if (fin) PetscCall(VecCreateSeq(PETSC_COMM_SELF, fft->n, fin)); in MatCreateVecsFFTW_FFTW()
497 if (fout) PetscCall(VecCreateSeq(PETSC_COMM_SELF, fft->n, fout)); in MatCreateVecsFFTW_FFTW()
498 if (bout) PetscCall(VecCreateSeq(PETSC_COMM_SELF, fft->n, bout)); in MatCreateVecsFFTW_FFTW()
502 Mat_FFTW *fftw = (Mat_FFTW *)fft->data; in MatCreateVecsFFTW_FFTW()
503 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatCreateVecsFFTW_FFTW()
525 PetscCall(PetscIntCast(fft->N, &N1)); in MatCreateVecsFFTW_FFTW()
534 PetscCall(PetscIntCast(fft->N, &N1)); in MatCreateVecsFFTW_FFTW()
544 PetscCall(PetscIntCast(fft->N, &N1)); in MatCreateVecsFFTW_FFTW()
583 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_fin, fin)); in MatCreateVecsFFTW_FFTW()
590 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_fout, fout)); in MatCreateVecsFFTW_FFTW()
597 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_bout, bout)); in MatCreateVecsFFTW_FFTW()
634 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_fin, fin)); in MatCreateVecsFFTW_FFTW()
641 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_fout, fout)); in MatCreateVecsFFTW_FFTW()
648 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_bout, bout)); in MatCreateVecsFFTW_FFTW()
660 …N1 = 2 * fft->N * (PetscInt)(fftw->dim_fftw[fftw->ndim_fftw - 1… in MatCreateVecsFFTW_FFTW()
665 PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, N1, (PetscScalar *)data_finr, fin)); in MatCreateVecsFFTW_FFTW()
672 PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, N1, (PetscScalar *)data_fout, fout)); in MatCreateVecsFFTW_FFTW()
679 PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, N1, (PetscScalar *)data_boutr, bout)); in MatCreateVecsFFTW_FFTW()
688 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_fin, fin)); in MatCreateVecsFFTW_FFTW()
695 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_fout, fout)); in MatCreateVecsFFTW_FFTW()
702 … PetscCall(VecCreateMPIWithArray(comm, 1, fft->n, fft->N, (const PetscScalar *)data_bout, bout)); in MatCreateVecsFFTW_FFTW()
753 Mat_FFT *fft = (Mat_FFT *)A->data; in VecScatterPetscToFFTW_FFTW() local
769 PetscCall(ISCreateStride(PETSC_COMM_SELF, fft->N, 0, 1, &list1)); in VecScatterPetscToFFTW_FFTW()
777 Mat_FFTW *fftw = (Mat_FFTW *)fft->data; in VecScatterPetscToFFTW_FFTW()
778 PetscInt ndim = fft->ndim, *dim = fft->dim, n1; in VecScatterPetscToFFTW_FFTW()
1002 Mat_FFT *fft = (Mat_FFT *)A->data; in VecScatterFFTWToPetsc_FFTW() local
1015 PetscCall(ISCreateStride(comm, fft->N, 0, 1, &list1)); in VecScatterFFTWToPetsc_FFTW()
1024 Mat_FFTW *fftw = (Mat_FFTW *)fft->data; in VecScatterFFTWToPetsc_FFTW()
1025 PetscInt ndim = fft->ndim, *dim = fft->dim; in VecScatterFFTWToPetsc_FFTW()
1225 Mat_FFT *fft = (Mat_FFT *)A->data; in MatCreate_FFTW() local
1227 PetscInt ndim = fft->ndim, *dim = fft->dim; in MatCreate_FFTW()
1262 PetscCall(MatSetSizes(A, fft->N, fft->N, fft->N, fft->N)); in MatCreate_FFTW()
1263 fft->n = fft->N; in MatCreate_FFTW()
1266 fft->n = tot_dim; in MatCreate_FFTW()
1284 PetscCall(PetscIntCast(local_n0, &fft->n)); in MatCreate_FFTW()
1286 PetscCall(MatSetSizes(A, n1, fft->n, fft->N, fft->N)); in MatCreate_FFTW()
1292 fft->n = (PetscInt)local_n0 * dim[1]; in MatCreate_FFTW()
1293 PetscCall(MatSetSizes(A, fft->n, fft->n, fft->N, fft->N)); 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()
1305 fft->n = (PetscInt)local_n0 * dim[1] * dim[2]; in MatCreate_FFTW()
1306 PetscCall(MatSetSizes(A, fft->n, fft->n, fft->N, fft->N)); 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()
1318 PetscCall(PetscIntCast(local_n0 * partial_dim, &fft->n)); in MatCreate_FFTW()
1319 PetscCall(MatSetSizes(A, fft->n, fft->n, fft->N, fft->N)); in MatCreate_FFTW()
1327 … PetscCall(PetscIntCast(2 * (PetscInt)local_n0 * partial_dim * pdim[ndim - 1] / temp, &fft->n)); in MatCreate_FFTW()
1328 N1 = 2 * fft->N * (PetscInt)pdim[ndim - 1] / ((PetscInt)temp); in MatCreate_FFTW()
1332 PetscCall(MatSetSizes(A, fft->n, fft->n, N1, N1)); in MatCreate_FFTW()
1341 fft->data = (void *)fftw; in MatCreate_FFTW()
1370 fft->matdestroy = MatDestroy_FFTW; in MatCreate_FFTW()