Home
last modified time | relevance | path

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

/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx31 PetscCallHIP(hipFree(coo->Aimap2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
93 PetscCallHIP(hipMalloc((void **)&coo_d->Aimap2, coo_h->Annz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
107 …PetscCallHIP(hipMemcpy(coo_d->Aimap2, coo_h->Aimap2, coo_h->Annz2 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
144 …moteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscCount Aimap2[], const PetscCoun… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
181 const auto &Aimap2 = coo->Aimap2; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
223 …2 + Bnnz2 + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v2, Annz2, Aimap2, Ajmap2, Aperm2, A… in MatSetValuesCOO_MPIAIJHIPSPARSE()
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu31 PetscCallCUDA(cudaFree(coo->Aimap2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
93 PetscCallCUDA(cudaMalloc((void **)&coo_d->Aimap2, coo_h->Annz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
107 …PetscCallCUDA(cudaMemcpy(coo_d->Aimap2, coo_h->Aimap2, coo_h->Annz2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 …moteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscCount Aimap2[], const PetscCoun… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
181 const auto &Aimap2 = coo->Aimap2; in MatSetValuesCOO_MPIAIJCUSPARSE() local
223 …MatAddRemoteCOOValues<<<(int)((Annz2 + Bnnz2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aper… in MatSetValuesCOO_MPIAIJCUSPARSE()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx1509 PetscCountKokkosView Aimap2, Ajmap2, Aperm2; member
1528 …Aimap2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aimap2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1601 const auto &Aimap2 = coo->Aimap2; in MatSetValuesCOO_MPIAIJKokkos() local
1651 for (PetscCount k = Ajmap2(i); k < Ajmap2(i + 1); k++) Aa(Aimap2(i)) += v2(Aperm2(k)); in MatSetValuesCOO_MPIAIJKokkos()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.h84 …PetscCount *Aimap2, *Ajmap2, *Aperm2; /* Lengths: [Annz2], [Annz2+1], [Atot2]. Remote entries t… member
H A Dmpiaij.c6327 PetscCall(PetscFree(coo->Aimap2)); in MatCOOStructDestroy_MPIAIJ()
6574 PetscCount *Aimap1, *Bimap1, *Aimap2, *Bimap2; in MatSetPreallocationCOO_MPIAIJ() local
6577 PetscCall(PetscMalloc1(Annz2, &Aimap2)); in MatSetPreallocationCOO_MPIAIJ()
6580 …rnal(mat, j1, j2, rowBegin1, rowMid1, rowBegin2, rowMid2, Ajmap1, Ajmap2, Aimap1, Aimap2, Ai, Aj)); in MatSetPreallocationCOO_MPIAIJ()
6688 coo->Aimap2 = Aimap2; in MatSetPreallocationCOO_MPIAIJ()
6711 const PetscCount *Ajmap1, *Ajmap2, *Aimap2; in MatSetValuesCOO_MPIAIJ() local
6726 Aimap2 = coo->Aimap2; in MatSetValuesCOO_MPIAIJ()
6759 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += recvbuf[Aperm2[k]]; in MatSetValuesCOO_MPIAIJ()