Searched refs:bs_h (Results 1 – 2 of 2) 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()
36 PetscCallCUDA(cudaMemcpy(bs_d, bs_h, sizeof(PetscInt) * (nblocks + 1), cudaMemcpyHostToDevice)); in UpdateOffsetsOnDevice()
46 PetscCallVoid(PetscFree3(bs_h, bs2_h, matIdx_h)); in ~PC_VPBJacobi_CUDA()
57 bs_h[0] = bs2_h[0] = 0; in ComputeOffsetsOnHost()
59 bs_h[i + 1] = bs_h[i] + bsizes[i]; in ComputeOffsetsOnHost()
61 for (PetscInt j = 0; j < bsizes[i]; j++) matIdx_h[bs_h[i] + j] = i; in ComputeOffsetsOnHost()
|
| /petsc/src/ksp/pc/impls/vpbjacobi/kokkos/ |
| H A D | vpbjacobi_kok.kokkos.cxx | 44 PetscInt *bs_h = bs_dual.view_host().data(); in BuildHelperArrays() local
50 bs_h[0] = bs2_h[0] = 0; in BuildHelperArrays()
53 bs_h[i + 1] = bs_h[i] + m; in BuildHelperArrays()
55 for (PetscInt j = 0; j < m; j++) blkMap_h[bs_h[i] + j] = i; in BuildHelperArrays()
|