Searched refs:host_ptr (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/ksp/impls/hpddm/cuda/ |
| H A D | hpddm.cu | 21 K *ptr, *host_ptr; in KSPSolve_HPDDM_CUDA_Private() local
24 PetscCall(PetscMalloc1(2 * N, &host_ptr)); in KSPSolve_HPDDM_CUDA_Private()
30 PetscCallCUDA(cudaMemcpy(host_ptr, ptr, 2 * N * sizeof(K), cudaMemcpyDeviceToHost)); in KSPSolve_HPDDM_CUDA_Private()
31 PetscCall(HPDDM::IterativeMethod::solve(*data->op, host_ptr, host_ptr + N, n, comm)); in KSPSolve_HPDDM_CUDA_Private()
32 PetscCallCUDA(cudaMemcpy(ptr + N, host_ptr + N, N * sizeof(K), cudaMemcpyHostToDevice)); in KSPSolve_HPDDM_CUDA_Private()
35 PetscCall(PetscFree(host_ptr)); in KSPSolve_HPDDM_CUDA_Private()
39 PetscScalar *host_ptr; in KSPSolve_HPDDM_CUDA_Private() local
41 PetscCall(PetscMalloc1(2 * N, &host_ptr)); in KSPSolve_HPDDM_CUDA_Private()
42 PetscCallCUDA(cudaMemcpy(host_ptr, b, N * sizeof(PetscScalar), cudaMemcpyDeviceToHost)); in KSPSolve_HPDDM_CUDA_Private()
43 PetscCallCUDA(cudaMemcpy(host_ptr + N, x, N * sizeof(PetscScalar), cudaMemcpyDeviceToHost)); in KSPSolve_HPDDM_CUDA_Private()
[all …]
|