Searched refs:bs_d (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/vpbjacobi/cuda/ |
| H A D | vpbjacobi_cuda.cu | 17 PetscInt *bs_d, *bs2_d, *matIdx_d; member
25 PetscCallCUDAVoid(cudaMalloc(&bs_d, sizeof(PetscInt) * (nblocks + 1))); in PC_VPBJacobi_CUDA()
36 PetscCallCUDA(cudaMemcpy(bs_d, bs_h, sizeof(PetscInt) * (nblocks + 1), cudaMemcpyHostToDevice)); in UpdateOffsetsOnDevice()
47 PetscCallCUDAVoid(cudaFree(bs_d)); in ~PC_VPBJacobi_CUDA()
126 …MatMultBatched<<<gridSize, 256>>>(n, pcuda->bs_d, pcuda->bs2_d, pcuda->matIdx_d, pcuda->diag_d, xx… in PCApplyOrTranspose_VPBJacobi_CUDA()
|