Home
last modified time | relevance | path

Searched refs:coo_d (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
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()
[all …]
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu51 MatCOOStruct_MPIAIJ *coo_h, *coo_d; in MatSetPreallocationCOO_MPIAIJCUSPARSE() local
85 PetscCall(PetscMalloc1(1, &coo_d)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
86 *coo_d = *coo_h; // do a shallow copy and then amend fields in coo_d in MatSetPreallocationCOO_MPIAIJCUSPARSE()
88 …PetscCall(PetscObjectReference((PetscObject)coo_d->sf)); // Since we destroy the sf in both coo_h … 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()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx1554 MatCOOStruct_MPIAIJKokkos *coo_d; in MatSetPreallocationCOO_MPIAIJKokkos() local
1566 PetscCallCXX(coo_d = new MatCOOStruct_MPIAIJKokkos(coo_h)); in MatSetPreallocationCOO_MPIAIJKokkos()
1570 PetscCall(PetscContainerSetPointer(container_d, coo_d)); in MatSetPreallocationCOO_MPIAIJKokkos()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx3736 MatCOOStruct_SeqAIJ *coo_h, *coo_d; in MatSetPreallocationCOO_SeqAIJHIPSPARSE() local
3758 PetscCall(PetscMalloc1(1, &coo_d)); 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()
3766 …PetscCall(PetscObjectContainerCompose((PetscObject)mat, "__PETSc_MatCOOStruct_Device", coo_d, MatC… in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu4378 MatCOOStruct_SeqAIJ *coo_h, *coo_d; in MatSetPreallocationCOO_SeqAIJCUSPARSE() local
4401 PetscCall(PetscMalloc1(1, &coo_d)); 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()
4409 …PetscCall(PetscObjectContainerCompose((PetscObject)mat, "__PETSc_MatCOOStruct_Device", coo_d, MatC… in MatSetPreallocationCOO_SeqAIJCUSPARSE()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx1373 MatCOOStruct_SeqAIJKokkos *coo_d; in MatSetPreallocationCOO_SeqAIJKokkos() local
1386 PetscCallCXX(coo_d = new MatCOOStruct_SeqAIJKokkos(coo_h)); in MatSetPreallocationCOO_SeqAIJKokkos()
1389 …PetscCall(PetscObjectContainerCompose((PetscObject)mat, "__PETSc_MatCOOStruct_Device", coo_d, MatC… in MatSetPreallocationCOO_SeqAIJKokkos()