Home
last modified time | relevance | path

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

/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx33 PetscCallHIP(hipFree(coo->Aperm2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
95 PetscCallHIP(hipMalloc((void **)&coo_d->Aperm2, coo_h->Atot2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
109 …PetscCallHIP(hipMemcpy(coo_d->Aperm2, coo_h->Aperm2, coo_h->Atot2 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
144 …, const PetscCount Aimap2[], const PetscCount Ajmap2[], const PetscCount Aperm2[], PetscScalar Aa[… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
186 const auto &Aperm2 = coo->Aperm2; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
223 … / 256), dim3(256), 0, PetscDefaultHipStream, v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2… in MatSetValuesCOO_MPIAIJHIPSPARSE()
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu33 PetscCallCUDA(cudaFree(coo->Aperm2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
95 PetscCallCUDA(cudaMalloc((void **)&coo_d->Aperm2, coo_h->Atot2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
109 …PetscCallCUDA(cudaMemcpy(coo_d->Aperm2, coo_h->Aperm2, coo_h->Atot2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 …, const PetscCount Aimap2[], const PetscCount Ajmap2[], const PetscCount Aperm2[], PetscScalar Aa[… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
186 const auto &Aperm2 = coo->Aperm2; in MatSetValuesCOO_MPIAIJCUSPARSE() local
223 …<<<(int)((Annz2 + Bnnz2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2… in MatSetValuesCOO_MPIAIJCUSPARSE()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx1509 PetscCountKokkosView Aimap2, Ajmap2, Aperm2; member
1530 …Aperm2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aperm2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1606 const auto &Aperm2 = coo->Aperm2; 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.c6329 PetscCall(PetscFree(coo->Aperm2)); in MatCOOStructDestroy_MPIAIJ()
6557 PetscCount *Ajmap2, *Aperm2, *Bjmap2, *Bperm2; in MatSetPreallocationCOO_MPIAIJ() local
6563 …s_Internal(mat, n2, i2, j2, perm2, rowBegin2, rowMid2, rowEnd2, &Atot2, &Aperm2, &Annz2, &Ajmap2, … in MatSetPreallocationCOO_MPIAIJ()
6690 coo->Aperm2 = Aperm2; in MatSetPreallocationCOO_MPIAIJ()
6713 const PetscCount *Aperm1, *Aperm2, *Bperm1, *Bperm2; in MatSetValuesCOO_MPIAIJ() local
6731 Aperm2 = coo->Aperm2; in MatSetValuesCOO_MPIAIJ()
6759 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += recvbuf[Aperm2[k]]; in MatSetValuesCOO_MPIAIJ()