Lines Matching refs:Annz2
93 PetscCallHIP(hipMalloc((void **)&coo_d->Aimap2, coo_h->Annz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
94 PetscCallHIP(hipMalloc((void **)&coo_d->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
107 …PetscCallHIP(hipMemcpy(coo_d->Aimap2, coo_h->Aimap2, coo_h->Annz2 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
108 …PetscCallHIP(hipMemcpy(coo_d->Ajmap2, coo_h->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount), hipM… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
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_MPIAIJHIPSPARSE() local
222 if (Annz2 + Bnnz2 > 0) { in MatSetValuesCOO_MPIAIJHIPSPARSE()
223 …NEL_NAME(MatAddRemoteCOOValues), dim3((Annz2 + Bnnz2 + 255) / 256), dim3(256), 0, PetscDefaultHipS… in MatSetValuesCOO_MPIAIJHIPSPARSE()