Home
last modified time | relevance | path

Searched refs:Bimap2 (Results 1 – 5 of 5) sorted by relevance

/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx34 PetscCallHIP(hipFree(coo->Bimap2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
96 PetscCallHIP(hipMalloc((void **)&coo_d->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
110 …PetscCallHIP(hipMemcpy(coo_d->Bimap2, coo_h->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
144 …etscCount Aperm2[], PetscScalar Aa[], PetscCount Bnnz2, const PetscCount Bimap2[], const PetscCoun… in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
184 const auto &Bimap2 = coo->Bimap2; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
223 … 0, PetscDefaultHipStream, v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjmap2, Bperm2, B… in MatSetValuesCOO_MPIAIJHIPSPARSE()
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu34 PetscCallCUDA(cudaFree(coo->Bimap2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
96 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
110 …PetscCallCUDA(cudaMemcpy(coo_d->Bimap2, coo_h->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 …etscCount Aperm2[], PetscScalar Aa[], PetscCount Bnnz2, const PetscCount Bimap2[], const PetscCoun… in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
184 const auto &Bimap2 = coo->Bimap2; in MatSetValuesCOO_MPIAIJCUSPARSE() local
223 …nnz2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjmap2, Bperm2, B… in MatSetValuesCOO_MPIAIJCUSPARSE()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx1510 PetscCountKokkosView Bimap2, Bjmap2, Bperm2; member
1531 …Bimap2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bimap2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1604 const auto &Bimap2 = coo->Bimap2; in MatSetValuesCOO_MPIAIJKokkos() local
1654 for (PetscCount k = Bjmap2(i); k < Bjmap2(i + 1); k++) Ba(Bimap2(i)) += v2(Bperm2(k)); in MatSetValuesCOO_MPIAIJKokkos()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.h85 …PetscCount *Bimap2, *Bjmap2, *Bperm2; /* Lengths: [Bnnz2], [Bnnz2+1], [Btot2]. Remote entries t… member
H A Dmpiaij.c6328 PetscCall(PetscFree(coo->Bimap2)); in MatCOOStructDestroy_MPIAIJ()
6574 PetscCount *Aimap1, *Bimap1, *Aimap2, *Bimap2; in MatSetPreallocationCOO_MPIAIJ() local
6578 PetscCall(PetscMalloc1(Bnnz2, &Bimap2)); in MatSetPreallocationCOO_MPIAIJ()
6581 …Internal(mat, j1, j2, rowMid1, rowEnd1, rowMid2, rowEnd2, Bjmap1, Bjmap2, Bimap1, Bimap2, Bi, Bj)); in MatSetPreallocationCOO_MPIAIJ()
6691 coo->Bimap2 = Bimap2; in MatSetPreallocationCOO_MPIAIJ()
6712 const PetscCount *Bjmap1, *Bjmap2, *Bimap2; in MatSetValuesCOO_MPIAIJ() local
6729 Bimap2 = coo->Bimap2; in MatSetValuesCOO_MPIAIJ()
6762 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += recvbuf[Bperm2[k]]; in MatSetValuesCOO_MPIAIJ()