Searched refs:matIdx_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
27 PetscCallCUDAVoid(cudaMalloc(&matIdx_d, sizeof(PetscInt) * n)); in PC_VPBJacobi_CUDA()
38 PetscCallCUDA(cudaMemcpy(matIdx_d, matIdx_h, sizeof(PetscInt) * n, cudaMemcpyHostToDevice)); in UpdateOffsetsOnDevice()
49 PetscCallCUDAVoid(cudaFree(matIdx_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()
|