| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 35 PetscCallHIP(hipFree(coo->Bjmap2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
97 PetscCallHIP(hipMalloc((void **)&coo_d->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
111 …PetscCallHIP(hipMemcpy(coo_d->Bjmap2, coo_h->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount), hipM… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
144 …alar Aa[], PetscCount Bnnz2, const PetscCount Bimap2[], const PetscCount Bjmap2[], const PetscCoun… in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
183 const auto &Bjmap2 = coo->Bjmap2; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
223 …, PetscDefaultHipStream, v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjmap2, Bperm2, Ba); in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 35 PetscCallCUDA(cudaFree(coo->Bjmap2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
97 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
111 …PetscCallCUDA(cudaMemcpy(coo_d->Bjmap2, coo_h->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount), cu… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 …alar Aa[], PetscCount Bnnz2, const PetscCount Bimap2[], const PetscCount Bjmap2[], const PetscCoun… in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
183 const auto &Bjmap2 = coo->Bjmap2; in MatSetValuesCOO_MPIAIJCUSPARSE() local
223 …z2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjmap2, Bperm2, Ba); in MatSetValuesCOO_MPIAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1510 PetscCountKokkosView Bimap2, Bjmap2, Bperm2; member
1532 …Bjmap2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bjmap2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1603 const auto &Bjmap2 = coo->Bjmap2; 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 D | mpiaij.h | 85 …PetscCount *Bimap2, *Bjmap2, *Bperm2; /* Lengths: [Bnnz2], [Bnnz2+1], [Btot2]. Remote entries t… member
|
| H A D | mpiaij.c | 6332 PetscCall(PetscFree(coo->Bjmap2)); in MatCOOStructDestroy_MPIAIJ()
6557 PetscCount *Ajmap2, *Aperm2, *Bjmap2, *Bperm2; in MatSetPreallocationCOO_MPIAIJ() local
6563 … rowBegin2, rowMid2, rowEnd2, &Atot2, &Aperm2, &Annz2, &Ajmap2, &Btot2, &Bperm2, &Bnnz2, &Bjmap2)); in MatSetPreallocationCOO_MPIAIJ()
6581 …ntries_Internal(mat, j1, j2, rowMid1, rowEnd1, rowMid2, rowEnd2, Bjmap1, Bjmap2, Bimap1, Bimap2, B… in MatSetPreallocationCOO_MPIAIJ()
6692 coo->Bjmap2 = Bjmap2; in MatSetPreallocationCOO_MPIAIJ()
6712 const PetscCount *Bjmap1, *Bjmap2, *Bimap2; in MatSetValuesCOO_MPIAIJ() local
6728 Bjmap2 = coo->Bjmap2; in MatSetValuesCOO_MPIAIJ()
6762 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += recvbuf[Bperm2[k]]; in MatSetValuesCOO_MPIAIJ()
|