Home
last modified time | relevance | path

Searched refs:coo_h (Results 1 – 6 of 6) sorted by relevance

/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx51 MatCOOStruct_MPIAIJ *coo_h, *coo_d; in MatSetPreallocationCOO_MPIAIJHIPSPARSE() local
84 PetscCall(PetscContainerGetPointer(container_h, &coo_h)); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
86 *coo_d = *coo_h; // do a shallow copy and then amend fields in coo_d 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()
[all …]
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu51 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()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx1514 MatCOOStruct_MPIAIJKokkos(const MatCOOStruct_MPIAIJ *coo_h) in MatCOOStruct_MPIAIJKokkos()
1518 n = coo_h->n; in MatCOOStruct_MPIAIJKokkos()
1519 sf = coo_h->sf; in MatCOOStruct_MPIAIJKokkos()
1520 Annz = coo_h->Annz; in MatCOOStruct_MPIAIJKokkos()
1521 Bnnz = coo_h->Bnnz; in MatCOOStruct_MPIAIJKokkos()
1522 Annz2 = coo_h->Annz2; in MatCOOStruct_MPIAIJKokkos()
1523 Bnnz2 = coo_h->Bnnz2; in MatCOOStruct_MPIAIJKokkos()
1524 …okkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Ajmap1, coo_h->Annz + 1)); in MatCOOStruct_MPIAIJKokkos()
1525 …= Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aperm1, coo_h->Atot1)); in MatCOOStruct_MPIAIJKokkos()
1526 …okkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bjmap1, coo_h->Bnnz + 1)); in MatCOOStruct_MPIAIJKokkos()
[all …]
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx1350 MatCOOStruct_SeqAIJKokkos(const MatCOOStruct_SeqAIJ *coo_h) in MatCOOStruct_SeqAIJKokkos()
1352 nz = coo_h->nz; in MatCOOStruct_SeqAIJKokkos()
1353 n = coo_h->n; in MatCOOStruct_SeqAIJKokkos()
1354 Atot = coo_h->Atot; in MatCOOStruct_SeqAIJKokkos()
1355 …jmap = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(coo_h->j… in MatCOOStruct_SeqAIJKokkos()
1356 …perm = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(coo_h->p… in MatCOOStruct_SeqAIJKokkos()
1372 MatCOOStruct_SeqAIJ *coo_h; in MatSetPreallocationCOO_SeqAIJKokkos() local
1385 PetscCall(PetscContainerGetPointer(container_h, &coo_h)); in MatSetPreallocationCOO_SeqAIJKokkos()
1386 PetscCallCXX(coo_d = new MatCOOStruct_SeqAIJKokkos(coo_h)); in MatSetPreallocationCOO_SeqAIJKokkos()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx3736 MatCOOStruct_SeqAIJ *coo_h, *coo_d; in MatSetPreallocationCOO_SeqAIJHIPSPARSE() local
3757 PetscCall(PetscContainerGetPointer(container_h, &coo_h)); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3759 *coo_d = *coo_h; // do a shallow copy and then amend some fields that need to be different in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3760 PetscCallHIP(hipMalloc((void **)&coo_d->jmap, (coo_h->nz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3761 …PetscCallHIP(hipMemcpy(coo_d->jmap, coo_h->jmap, (coo_h->nz + 1) * sizeof(PetscCount), hipMemcpyHo… in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3762 PetscCallHIP(hipMalloc((void **)&coo_d->perm, coo_h->Atot * sizeof(PetscCount))); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3763 …PetscCallHIP(hipMemcpy(coo_d->perm, coo_h->perm, coo_h->Atot * sizeof(PetscCount), hipMemcpyHostTo… in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu4378 MatCOOStruct_SeqAIJ *coo_h, *coo_d; in MatSetPreallocationCOO_SeqAIJCUSPARSE() local
4400 PetscCall(PetscContainerGetPointer(container_h, &coo_h)); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4402 *coo_d = *coo_h; // do a shallow copy and then amend some fields that need to be different in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4403 PetscCallCUDA(cudaMalloc((void **)&coo_d->jmap, (coo_h->nz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4404 …PetscCallCUDA(cudaMemcpy(coo_d->jmap, coo_h->jmap, (coo_h->nz + 1) * sizeof(PetscCount), cudaMemcp… in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4405 PetscCallCUDA(cudaMalloc((void **)&coo_d->perm, coo_h->Atot * sizeof(PetscCount))); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4406 …PetscCallCUDA(cudaMemcpy(coo_d->perm, coo_h->perm, coo_h->Atot * sizeof(PetscCount), cudaMemcpyHos… in MatSetPreallocationCOO_SeqAIJCUSPARSE()