| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 32 PetscCallHIP(hipFree(coo->Ajmap2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
94 PetscCallHIP(hipMalloc((void **)&coo_d->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
108 …PetscCallHIP(hipMemcpy(coo_d->Ajmap2, coo_h->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount), hipM… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
144 …alar kv[], PetscCount Annz2, const PetscCount Aimap2[], const PetscCount Ajmap2[], const PetscCoun… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
180 const auto &Ajmap2 = coo->Ajmap2; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
223 …2 + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2… in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 32 PetscCallCUDA(cudaFree(coo->Ajmap2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
94 PetscCallCUDA(cudaMalloc((void **)&coo_d->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
108 …PetscCallCUDA(cudaMemcpy(coo_d->Ajmap2, coo_h->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount), cu… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 …alar kv[], PetscCount Annz2, const PetscCount Aimap2[], const PetscCount Ajmap2[], const PetscCoun… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
180 const auto &Ajmap2 = coo->Ajmap2; 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 D | mpiaijkok.kokkos.cxx | 1509 PetscCountKokkosView Aimap2, Ajmap2, Aperm2; member
1529 …Ajmap2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Ajmap2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1600 const auto &Ajmap2 = coo->Ajmap2; 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 D | mpiaij.h | 84 …PetscCount *Aimap2, *Ajmap2, *Aperm2; /* Lengths: [Annz2], [Annz2+1], [Atot2]. Remote entries t… member
|
| H A D | mpiaij.c | 6331 PetscCall(PetscFree(coo->Ajmap2)); in MatCOOStructDestroy_MPIAIJ()
6557 PetscCount *Ajmap2, *Aperm2, *Bjmap2, *Bperm2; in MatSetPreallocationCOO_MPIAIJ() local
6563 …2, i2, j2, perm2, rowBegin2, rowMid2, rowEnd2, &Atot2, &Aperm2, &Annz2, &Ajmap2, &Btot2, &Bperm2, … in MatSetPreallocationCOO_MPIAIJ()
6580 …es_Internal(mat, j1, j2, rowBegin1, rowMid1, rowBegin2, rowMid2, Ajmap1, Ajmap2, Aimap1, Aimap2, A… in MatSetPreallocationCOO_MPIAIJ()
6689 coo->Ajmap2 = Ajmap2; in MatSetPreallocationCOO_MPIAIJ()
6711 const PetscCount *Ajmap1, *Ajmap2, *Aimap2; in MatSetValuesCOO_MPIAIJ() local
6725 Ajmap2 = coo->Ajmap2; in MatSetValuesCOO_MPIAIJ()
6759 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += recvbuf[Aperm2[k]]; in MatSetValuesCOO_MPIAIJ()
|