Searched refs:bs2_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()
37 … PetscCallCUDA(cudaMemcpy(bs2_d, bs2_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()
60 bs2_h[i + 1] = bs2_h[i] + bsizes[i] * bsizes[i]; in ComputeOffsetsOnHost()
|
| /petsc/src/ksp/pc/impls/vpbjacobi/kokkos/ |
| H A D | vpbjacobi_kok.kokkos.cxx | 45 PetscInt *bs2_h = bs2_dual.view_host().data(); in BuildHelperArrays() local
50 bs_h[0] = bs2_h[0] = 0; in BuildHelperArrays()
54 bs2_h[i + 1] = bs2_h[i] + m * m; in BuildHelperArrays()
|