Lines Matching refs:coo_h
51 MatCOOStruct_MPIAIJ *coo_h, *coo_d; in MatSetPreallocationCOO_MPIAIJCUSPARSE() local
84 PetscCall(PetscContainerGetPointer(container_h, &coo_h)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
86 *coo_d = *coo_h; // do a shallow copy and then amend fields in coo_d in MatSetPreallocationCOO_MPIAIJCUSPARSE()
89 PetscCallCUDA(cudaMalloc((void **)&coo_d->Ajmap1, (coo_h->Annz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
90 PetscCallCUDA(cudaMalloc((void **)&coo_d->Aperm1, coo_h->Atot1 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
91 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bjmap1, (coo_h->Bnnz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
92 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bperm1, coo_h->Btot1 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
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()
95 PetscCallCUDA(cudaMalloc((void **)&coo_d->Aperm2, coo_h->Atot2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
96 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
97 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
98 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bperm2, coo_h->Btot2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
99 PetscCallCUDA(cudaMalloc((void **)&coo_d->Cperm1, coo_h->sendlen * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
100 PetscCallCUDA(cudaMalloc((void **)&coo_d->sendbuf, coo_h->sendlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
101 PetscCallCUDA(cudaMalloc((void **)&coo_d->recvbuf, coo_h->recvlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
103 …PetscCallCUDA(cudaMemcpy(coo_d->Ajmap1, coo_h->Ajmap1, (coo_h->Annz + 1) * sizeof(PetscCount), cud… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
104 …PetscCallCUDA(cudaMemcpy(coo_d->Aperm1, coo_h->Aperm1, coo_h->Atot1 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
105 …PetscCallCUDA(cudaMemcpy(coo_d->Bjmap1, coo_h->Bjmap1, (coo_h->Bnnz + 1) * sizeof(PetscCount), cud… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
106 …PetscCallCUDA(cudaMemcpy(coo_d->Bperm1, coo_h->Bperm1, coo_h->Btot1 * sizeof(PetscCount), cudaMemc… 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()
109 …PetscCallCUDA(cudaMemcpy(coo_d->Aperm2, coo_h->Aperm2, coo_h->Atot2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
110 …PetscCallCUDA(cudaMemcpy(coo_d->Bimap2, coo_h->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
111 …PetscCallCUDA(cudaMemcpy(coo_d->Bjmap2, coo_h->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount), cu… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
112 …PetscCallCUDA(cudaMemcpy(coo_d->Bperm2, coo_h->Bperm2, coo_h->Btot2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
113 …PetscCallCUDA(cudaMemcpy(coo_d->Cperm1, coo_h->Cperm1, coo_h->sendlen * sizeof(PetscCount), cudaMe… in MatSetPreallocationCOO_MPIAIJCUSPARSE()