Searched refs:PetscCallCUDAVoid (Results 1 – 3 of 3) sorted by relevance
25 PetscCallCUDAVoid(cudaMalloc(&bs_d, sizeof(PetscInt) * (nblocks + 1))); in PC_VPBJacobi_CUDA()26 PetscCallCUDAVoid(cudaMalloc(&bs2_d, sizeof(PetscInt) * (nblocks + 1))); in PC_VPBJacobi_CUDA()27 PetscCallCUDAVoid(cudaMalloc(&matIdx_d, sizeof(PetscInt) * n)); in PC_VPBJacobi_CUDA()28 PetscCallCUDAVoid(cudaMalloc(&diag_d, sizeof(MatScalar) * nsize)); in PC_VPBJacobi_CUDA()47 PetscCallCUDAVoid(cudaFree(bs_d)); in ~PC_VPBJacobi_CUDA()48 PetscCallCUDAVoid(cudaFree(bs2_d)); in ~PC_VPBJacobi_CUDA()49 PetscCallCUDAVoid(cudaFree(matIdx_d)); in ~PC_VPBJacobi_CUDA()50 PetscCallCUDAVoid(cudaFree(diag_d)); in ~PC_VPBJacobi_CUDA()
36 #define PetscCallCUDAVoid(...) \ macro54 #define PetscCallCUDAVoid(...) \ macro
97 if (!jac->spptr) PetscCallCUDAVoid(cudaMalloc(&jac->spptr, size)); in PCSetUp_PBJacobi_CUDA()98 PetscCallCUDAVoid(cudaMemcpy(jac->spptr, jac->diag, size, cudaMemcpyHostToDevice)); in PCSetUp_PBJacobi_CUDA()