Searched refs:matIdx_h (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/vpbjacobi/cuda/ |
| H A D | vpbjacobi_cuda.cu | 16 PetscInt *bs_h, *bs2_h, *matIdx_h; member
24 PetscCallVoid(PetscMalloc3(nblocks + 1, &bs_h, nblocks + 1, &bs2_h, n, &matIdx_h)); in PC_VPBJacobi_CUDA()
38 PetscCallCUDA(cudaMemcpy(matIdx_d, matIdx_h, sizeof(PetscInt) * n, cudaMemcpyHostToDevice)); in UpdateOffsetsOnDevice()
46 PetscCallVoid(PetscFree3(bs_h, bs2_h, matIdx_h)); in ~PC_VPBJacobi_CUDA()
61 for (PetscInt j = 0; j < bsizes[i]; j++) matIdx_h[bs_h[i] + j] = i; in ComputeOffsetsOnHost()
|