51   MatCOOStruct_MPIAIJ *coo_h, *coo_d;  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()  local
85   PetscCall(PetscMalloc1(1, &coo_d));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
86   *coo_d = *coo_h; // do a shallow copy and then amend fields in coo_d  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
88 …PetscCall(PetscObjectReference((PetscObject)coo_d->sf)); // Since we destroy the sf in both coo_h …  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
89   PetscCallHIP(hipMalloc((void **)&coo_d->Ajmap1, (coo_h->Annz + 1) * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
90   PetscCallHIP(hipMalloc((void **)&coo_d->Aperm1, coo_h->Atot1 * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
91   PetscCallHIP(hipMalloc((void **)&coo_d->Bjmap1, (coo_h->Bnnz + 1) * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
92   PetscCallHIP(hipMalloc((void **)&coo_d->Bperm1, coo_h->Btot1 * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
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()
95   PetscCallHIP(hipMalloc((void **)&coo_d->Aperm2, coo_h->Atot2 * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
96   PetscCallHIP(hipMalloc((void **)&coo_d->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
97   PetscCallHIP(hipMalloc((void **)&coo_d->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
98   PetscCallHIP(hipMalloc((void **)&coo_d->Bperm2, coo_h->Btot2 * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
99   PetscCallHIP(hipMalloc((void **)&coo_d->Cperm1, coo_h->sendlen * sizeof(PetscCount)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
100   PetscCallHIP(hipMalloc((void **)&coo_d->sendbuf, coo_h->sendlen * sizeof(PetscScalar)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
101   PetscCallHIP(hipMalloc((void **)&coo_d->recvbuf, coo_h->recvlen * sizeof(PetscScalar)));  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
103 …PetscCallHIP(hipMemcpy(coo_d->Ajmap1, coo_h->Ajmap1, (coo_h->Annz + 1) * sizeof(PetscCount), hipMe…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
104 …PetscCallHIP(hipMemcpy(coo_d->Aperm1, coo_h->Aperm1, coo_h->Atot1 * sizeof(PetscCount), hipMemcpyH…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
105 …PetscCallHIP(hipMemcpy(coo_d->Bjmap1, coo_h->Bjmap1, (coo_h->Bnnz + 1) * sizeof(PetscCount), hipMe…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
106 …PetscCallHIP(hipMemcpy(coo_d->Bperm1, coo_h->Bperm1, coo_h->Btot1 * sizeof(PetscCount), hipMemcpyH…  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()
109 …PetscCallHIP(hipMemcpy(coo_d->Aperm2, coo_h->Aperm2, coo_h->Atot2 * sizeof(PetscCount), hipMemcpyH…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
110 …PetscCallHIP(hipMemcpy(coo_d->Bimap2, coo_h->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount), hipMemcpyH…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
111 …PetscCallHIP(hipMemcpy(coo_d->Bjmap2, coo_h->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount), hipM…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
112 …PetscCallHIP(hipMemcpy(coo_d->Bperm2, coo_h->Bperm2, coo_h->Btot2 * sizeof(PetscCount), hipMemcpyH…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
113 …PetscCallHIP(hipMemcpy(coo_d->Cperm1, coo_h->Cperm1, coo_h->sendlen * sizeof(PetscCount), hipMemcp…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
116 …PetscCall(PetscObjectContainerCompose((PetscObject)mat, "__PETSc_MatCOOStruct_Device", coo_d, MatC…  in MatSetPreallocationCOO_MPIAIJHIPSPARSE()