Lines Matching refs:dim

11   PetscInt     *dim;  member
20 PetscInt ndim = cufft->ndim, *dim = cufft->dim; in MatMult_SeqCUFFT() local
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()
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
67 PetscCallCUFFT(cufftPlan1d(&cufft->p_backward, dim[0], CUFFT_C2C, 1)); in MatMultTranspose_SeqCUFFT()
70 PetscCallCUFFT(cufftPlan2d(&cufft->p_backward, dim[0], dim[1], CUFFT_C2C)); in MatMultTranspose_SeqCUFFT()
73 PetscCallCUFFT(cufftPlan3d(&cufft->p_backward, dim[0], dim[1], dim[2], CUFFT_C2C)); 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()
95 PetscCall(PetscFree(cufft->dim)); in MatDestroy_SeqCUFFT()
124 PetscErrorCode MatCreateSeqCUFFT(MPI_Comm comm, PetscInt ndim, const PetscInt dim[], Mat *A) in MatCreateSeqCUFFT() argument
131 if (ndim) PetscAssertPointer(dim, 3); in MatCreateSeqCUFFT()
135 …PetscCheck(dim[d] >= 0, PETSC_COMM_SELF, PETSC_ERR_USER, "dim[%" PetscInt_FMT "]=%" PetscInt_FMT "… in MatCreateSeqCUFFT()
136 m *= dim[d]; in MatCreateSeqCUFFT()
143 PetscCall(PetscMalloc1(ndim + 1, &cufft->dim)); in MatCreateSeqCUFFT()
144 PetscCall(PetscArraycpy(cufft->dim, dim, ndim)); in MatCreateSeqCUFFT()
149 cufft->dim[ndim] = m; in MatCreateSeqCUFFT()