93   PetscCallCUDA(cudaMalloc((void **)&coo_d->Aimap2, coo_h->Annz2 * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJCUSPARSE()
94   PetscCallCUDA(cudaMalloc((void **)&coo_d->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJCUSPARSE()
107 …PetscCallCUDA(cudaMemcpy(coo_d->Aimap2, coo_h->Aimap2, coo_h->Annz2 * sizeof(PetscCount), cudaMemc…  in MatSetPreallocationCOO_MPIAIJCUSPARSE()
108 …PetscCallCUDA(cudaMemcpy(coo_d->Ajmap2, coo_h->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount), cu…  in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 __global__ static void MatAddRemoteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscC…  in MatAddRemoteCOOValues()  argument
148   for (; i < Annz2 + Bnnz2; i += grid_size) {  in MatAddRemoteCOOValues()
149     if (i < Annz2) {  in MatAddRemoteCOOValues()
152       i -= Annz2;  in MatAddRemoteCOOValues()
174   const auto &Annz2  = coo->Annz2;  in MatSetValuesCOO_MPIAIJCUSPARSE()  local
222   if (Annz2 + Bnnz2 > 0) {  in MatSetValuesCOO_MPIAIJCUSPARSE()
223 …MatAddRemoteCOOValues<<<(int)((Annz2 + Bnnz2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aper…  in MatSetValuesCOO_MPIAIJCUSPARSE()