Searched refs:pcuda (Results 1 – 1 of 1) sorted by relevance
107 PC_VPBJacobi_CUDA *pcuda = static_cast<PC_VPBJacobi_CUDA *>(jac->spptr); in PCApplyOrTranspose_VPBJacobi_CUDA() local126 …MatMultBatched<<<gridSize, 256>>>(n, pcuda->bs_d, pcuda->bs2_d, pcuda->matIdx_d, pcuda->diag_d, xx… in PCApplyOrTranspose_VPBJacobi_CUDA()131 PetscCall(PetscLogGpuFlops(pcuda->nsize * 2)); /* FMA on entries in all blocks */ in PCApplyOrTranspose_VPBJacobi_CUDA()163 PC_VPBJacobi_CUDA *pcuda = static_cast<PC_VPBJacobi_CUDA *>(jac->spptr); in PCSetUp_VPBJacobi_CUDA() local174 if (pcuda && (pcuda->n != n || pcuda->nblocks != nblocks || pcuda->nsize != nsize)) { in PCSetUp_VPBJacobi_CUDA()175 PetscCallCXX(delete pcuda); in PCSetUp_VPBJacobi_CUDA()176 pcuda = nullptr; in PCSetUp_VPBJacobi_CUDA()179 if (!pcuda) { /* allocate the struct along with the helper arrays from the scratch */ in PCSetUp_VPBJacobi_CUDA()182 PetscCall(pcuda->UpdateOffsetsOnDevice(bsizes, jac->diag)); in PCSetUp_VPBJacobi_CUDA()