Searched refs:Bnnz2 (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 96 PetscCallHIP(hipMalloc((void **)&coo_d->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
97 PetscCallHIP(hipMalloc((void **)&coo_d->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
110 …PetscCallHIP(hipMemcpy(coo_d->Bimap2, coo_h->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
111 …PetscCallHIP(hipMemcpy(coo_d->Bjmap2, coo_h->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount), hipM… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
144 …scCount Ajmap2[], const PetscCount Aperm2[], PetscScalar Aa[], PetscCount Bnnz2, const PetscCount … in MatAddRemoteCOOValues() argument
148 for (; i < Annz2 + Bnnz2; i += grid_size) { in MatAddRemoteCOOValues()
176 const auto &Bnnz2 = coo->Bnnz2; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
222 if (Annz2 + Bnnz2 > 0) { in MatSetValuesCOO_MPIAIJHIPSPARSE()
223 …OOValues), dim3((Annz2 + Bnnz2 + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v2, Annz2, Aima… in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 96 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
97 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
110 …PetscCallCUDA(cudaMemcpy(coo_d->Bimap2, coo_h->Bimap2, coo_h->Bnnz2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
111 …PetscCallCUDA(cudaMemcpy(coo_d->Bjmap2, coo_h->Bjmap2, (coo_h->Bnnz2 + 1) * sizeof(PetscCount), cu… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 …scCount Ajmap2[], const PetscCount Aperm2[], PetscScalar Aa[], PetscCount Bnnz2, const PetscCount … in MatAddRemoteCOOValues() argument
148 for (; i < Annz2 + Bnnz2; i += grid_size) { in MatAddRemoteCOOValues()
176 const auto &Bnnz2 = coo->Bnnz2; in MatSetValuesCOO_MPIAIJCUSPARSE() local
222 if (Annz2 + Bnnz2 > 0) { in MatSetValuesCOO_MPIAIJCUSPARSE()
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 | 1506 PetscCount Annz2, Bnnz2; member
1523 Bnnz2 = coo_h->Bnnz2; in MatCOOStruct_MPIAIJKokkos()
1531 …= Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bimap2, coo_h->Bnnz2)); in MatCOOStruct_MPIAIJKokkos()
1532 …kkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bjmap2, coo_h->Bnnz2 + 1)); in MatCOOStruct_MPIAIJKokkos()
1596 const auto &Bnnz2 = coo->Bnnz2; in MatSetValuesCOO_MPIAIJKokkos() local
1649 Kokkos::RangePolicy<>(exec, 0, Annz2 + Bnnz2), KOKKOS_LAMBDA(PetscCount i) { in MatSetValuesCOO_MPIAIJKokkos()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.h | 80 …PetscCount Annz2, Bnnz2; /* Number of unique remote entries belonging to A and B */ member
|
| H A D | mpiaij.c | 6558 PetscCount Annz2, Bnnz2, Atot2, Btot2; in MatSetPreallocationCOO_MPIAIJ() local
6563 … rowBegin2, rowMid2, rowEnd2, &Atot2, &Aperm2, &Annz2, &Ajmap2, &Btot2, &Bperm2, &Bnnz2, &Bjmap2)); in MatSetPreallocationCOO_MPIAIJ()
6572 PetscCall(PetscMalloc1(Bnnz1 + Bnnz2, &Bj)); in MatSetPreallocationCOO_MPIAIJ()
6578 PetscCall(PetscMalloc1(Bnnz2, &Bimap2)); in MatSetPreallocationCOO_MPIAIJ()
6616 if (Bnnz < Bnnz1 + Bnnz2) { in MatSetPreallocationCOO_MPIAIJ()
6679 coo->Bnnz2 = Bnnz2; in MatSetPreallocationCOO_MPIAIJ()
6761 for (PetscCount i = 0; i < coo->Bnnz2; i++) { in MatSetValuesCOO_MPIAIJ()
|