Searched refs:Bperm2 (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 36 PetscCallHIP(hipFree(coo->Bperm2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
98 PetscCallHIP(hipMalloc((void **)&coo_d->Bperm2, coo_h->Btot2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
112 …PetscCallHIP(hipMemcpy(coo_d->Bperm2, coo_h->Bperm2, coo_h->Btot2 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
144 …, const PetscCount Bimap2[], const PetscCount Bjmap2[], const PetscCount Bperm2[], PetscScalar Ba[… in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
188 const auto &Bperm2 = coo->Bperm2; 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 | 36 PetscCallCUDA(cudaFree(coo->Bperm2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
98 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bperm2, coo_h->Btot2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
112 …PetscCallCUDA(cudaMemcpy(coo_d->Bperm2, coo_h->Bperm2, coo_h->Btot2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 …, const PetscCount Bimap2[], const PetscCount Bjmap2[], const PetscCount Bperm2[], PetscScalar Ba[… in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
188 const auto &Bperm2 = coo->Bperm2; 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
1533 …Bperm2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bperm2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1608 const auto &Bperm2 = coo->Bperm2; 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 | 6330 PetscCall(PetscFree(coo->Bperm2)); 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()
6693 coo->Bperm2 = Bperm2; in MatSetPreallocationCOO_MPIAIJ()
6713 const PetscCount *Aperm1, *Aperm2, *Bperm1, *Bperm2; in MatSetValuesCOO_MPIAIJ() local
6733 Bperm2 = coo->Bperm2; in MatSetValuesCOO_MPIAIJ()
6762 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += recvbuf[Bperm2[k]]; in MatSetValuesCOO_MPIAIJ()
|