10   PetscInt      ndim;  member
20   PetscInt      ndim = cufft->ndim, *dim = cufft->dim;  in MatMult_SeqCUFFT()  local
28     switch (ndim) {  in MatMult_SeqCUFFT()
39 …COMM_SELF, PETSC_ERR_USER, "Cannot create plan for %" PetscInt_FMT "-dimensional transform", ndim);  in MatMult_SeqCUFFT()
43 …PetscCallCUDA(cudaMemcpy(devArray, x_array, sizeof(cufftComplex) * dim[ndim], cudaMemcpyHostToDevi…  in MatMult_SeqCUFFT()
47 …PetscCallCUDA(cudaMemcpy(y_array, devArray, sizeof(cufftComplex) * dim[ndim], cudaMemcpyDeviceToHo…  in MatMult_SeqCUFFT()
57   PetscInt      ndim = cufft->ndim, *dim = cufft->dim;  in MatMultTranspose_SeqCUFFT()  local
65     switch (ndim) {  in MatMultTranspose_SeqCUFFT()
76 …COMM_SELF, PETSC_ERR_USER, "Cannot create plan for %" PetscInt_FMT "-dimensional transform", ndim);  in MatMultTranspose_SeqCUFFT()
80 …PetscCallCUDA(cudaMemcpy(devArray, x_array, sizeof(cufftComplex) * dim[ndim], cudaMemcpyHostToDevi…  in MatMultTranspose_SeqCUFFT()
84 …PetscCallCUDA(cudaMemcpy(y_array, devArray, sizeof(cufftComplex) * dim[ndim], cudaMemcpyDeviceToHo…  in MatMultTranspose_SeqCUFFT()
124 PetscErrorCode MatCreateSeqCUFFT(MPI_Comm comm, PetscInt ndim, const PetscInt dim[], Mat *A)  in MatCreateSeqCUFFT()  argument
130 …PetscCheck(ndim >= 0, PETSC_COMM_SELF, PETSC_ERR_USER, "ndim %" PetscInt_FMT " must be > 0", ndim);  in MatCreateSeqCUFFT()
131   if (ndim) PetscAssertPointer(dim, 3);  in MatCreateSeqCUFFT()
134   for (PetscInt d = 0; d < ndim; ++d) {  in MatCreateSeqCUFFT()
143   PetscCall(PetscMalloc1(ndim + 1, &cufft->dim));  in MatCreateSeqCUFFT()
144   PetscCall(PetscArraycpy(cufft->dim, dim, ndim));  in MatCreateSeqCUFFT()
146   cufft->ndim       = ndim;  in MatCreateSeqCUFFT()
149   cufft->dim[ndim]  = m;  in MatCreateSeqCUFFT()