Searched refs:cufft (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/cufft/ |
| H A D | cufft.cu | 18 Mat_CUFFT *cufft = (Mat_CUFFT *)A->data; in MatMult_SeqCUFFT() local
19 cufftComplex *devArray = cufft->devArray; in MatMult_SeqCUFFT()
20 PetscInt ndim = cufft->ndim, *dim = cufft->dim; in MatMult_SeqCUFFT()
26 if (!cufft->p_forward) { in MatMult_SeqCUFFT()
30 PetscCallCUFFT(cufftPlan1d(&cufft->p_forward, dim[0], CUFFT_C2C, 1)); in MatMult_SeqCUFFT()
33 PetscCallCUFFT(cufftPlan2d(&cufft->p_forward, dim[0], dim[1], CUFFT_C2C)); in MatMult_SeqCUFFT()
36 PetscCallCUFFT(cufftPlan3d(&cufft->p_forward, dim[0], dim[1], dim[2], CUFFT_C2C)); in MatMult_SeqCUFFT()
45 PetscCallCUFFT(cufftExecC2C(cufft->p_forward, devArray, devArray, CUFFT_FORWARD)); in MatMult_SeqCUFFT()
55 Mat_CUFFT *cufft = (Mat_CUFFT *)A->data; in MatMultTranspose_SeqCUFFT() local
56 cufftComplex *devArray = cufft->devArray; in MatMultTranspose_SeqCUFFT()
[all …]
|