mpiaijcusparse.cu - OpenGrok cross reference for /petsc/src/mat/impls/aij/mpi/mpicusparse/mpiaijcusparse.cu

Lines Matching refs:PetscCallCUDA
27   PetscCallCUDA(cudaFree(coo->Ajmap1));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
28   PetscCallCUDA(cudaFree(coo->Aperm1));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
29   PetscCallCUDA(cudaFree(coo->Bjmap1));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
30   PetscCallCUDA(cudaFree(coo->Bperm1));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
31   PetscCallCUDA(cudaFree(coo->Aimap2));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
32   PetscCallCUDA(cudaFree(coo->Ajmap2));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
33   PetscCallCUDA(cudaFree(coo->Aperm2));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
34   PetscCallCUDA(cudaFree(coo->Bimap2));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
35   PetscCallCUDA(cudaFree(coo->Bjmap2));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
36   PetscCallCUDA(cudaFree(coo->Bperm2));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
37   PetscCallCUDA(cudaFree(coo->Cperm1));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
38   PetscCallCUDA(cudaFree(coo->sendbuf));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
39   PetscCallCUDA(cudaFree(coo->recvbuf));  in MatCOOStructDestroy_MPIAIJCUSPARSE()
68     PetscCallCUDA(cudaMemcpy(i, coo_i, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost));  in MatSetPreallocationCOO_MPIAIJCUSPARSE()
69     PetscCallCUDA(cudaMemcpy(j, coo_j, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost));  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()
193     PetscCallCUDA(cudaMalloc((void **)&v1, coo->n * sizeof(PetscScalar)));  in MatSetValuesCOO_MPIAIJCUSPARSE()
194     PetscCallCUDA(cudaMemcpy((void *)v1, v, coo->n * sizeof(PetscScalar), cudaMemcpyHostToDevice));  in MatSetValuesCOO_MPIAIJCUSPARSE()
209     PetscCallCUDA(cudaPeekAtLastError());  in MatSetValuesCOO_MPIAIJCUSPARSE()
217     PetscCallCUDA(cudaPeekAtLastError());  in MatSetValuesCOO_MPIAIJCUSPARSE()
224     PetscCallCUDA(cudaPeekAtLastError());  in MatSetValuesCOO_MPIAIJCUSPARSE()
235   if (PetscMemTypeHost(memtype)) PetscCallCUDA(cudaFree((void *)v1));  in MatSetValuesCOO_MPIAIJCUSPARSE()