Searched refs:Annz2 (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 93 PetscCallHIP(hipMalloc((void **)&coo_d->Aimap2, coo_h->Annz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
94 PetscCallHIP(hipMalloc((void **)&coo_d->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
107 …PetscCallHIP(hipMemcpy(coo_d->Aimap2, coo_h->Aimap2, coo_h->Annz2 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
108 …PetscCallHIP(hipMemcpy(coo_d->Ajmap2, coo_h->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount), hipM… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
144 __global__ static void MatAddRemoteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscC… in MatAddRemoteCOOValues() argument
148 for (; i < Annz2 + Bnnz2; i += grid_size) { in MatAddRemoteCOOValues()
149 if (i < Annz2) { in MatAddRemoteCOOValues()
152 i -= Annz2; in MatAddRemoteCOOValues()
174 const auto &Annz2 = coo->Annz2; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
222 if (Annz2 + Bnnz2 > 0) { in MatSetValuesCOO_MPIAIJHIPSPARSE()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 93 PetscCallCUDA(cudaMalloc((void **)&coo_d->Aimap2, coo_h->Annz2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
94 PetscCallCUDA(cudaMalloc((void **)&coo_d->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
107 …PetscCallCUDA(cudaMemcpy(coo_d->Aimap2, coo_h->Aimap2, coo_h->Annz2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
108 …PetscCallCUDA(cudaMemcpy(coo_d->Ajmap2, coo_h->Ajmap2, (coo_h->Annz2 + 1) * sizeof(PetscCount), cu… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
144 __global__ static void MatAddRemoteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscC… in MatAddRemoteCOOValues() argument
148 for (; i < Annz2 + Bnnz2; i += grid_size) { in MatAddRemoteCOOValues()
149 if (i < Annz2) { in MatAddRemoteCOOValues()
152 i -= Annz2; in MatAddRemoteCOOValues()
174 const auto &Annz2 = coo->Annz2; in MatSetValuesCOO_MPIAIJCUSPARSE() local
222 if (Annz2 + Bnnz2 > 0) { in MatSetValuesCOO_MPIAIJCUSPARSE()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1506 PetscCount Annz2, Bnnz2; member
1522 Annz2 = coo_h->Annz2; in MatCOOStruct_MPIAIJKokkos()
1528 …= Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aimap2, coo_h->Annz2)); in MatCOOStruct_MPIAIJKokkos()
1529 …kkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Ajmap2, coo_h->Annz2 + 1)); in MatCOOStruct_MPIAIJKokkos()
1594 const auto &Annz2 = coo->Annz2; in MatSetValuesCOO_MPIAIJKokkos() local
1649 Kokkos::RangePolicy<>(exec, 0, Annz2 + Bnnz2), KOKKOS_LAMBDA(PetscCount i) { in MatSetValuesCOO_MPIAIJKokkos()
1650 if (i < Annz2) { in MatSetValuesCOO_MPIAIJKokkos()
1653 i -= Annz2; 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 …al(mat, n2, i2, j2, perm2, rowBegin2, rowMid2, rowEnd2, &Atot2, &Aperm2, &Annz2, &Ajmap2, &Btot2, … in MatSetPreallocationCOO_MPIAIJ()
6571 …PetscCall(PetscMalloc1(Annz1 + Annz2, &Aj)); /* Since local and remote entries might have dups, we… in MatSetPreallocationCOO_MPIAIJ()
6577 PetscCall(PetscMalloc1(Annz2, &Aimap2)); in MatSetPreallocationCOO_MPIAIJ()
6608 if (Annz < Annz1 + Annz2) { in MatSetPreallocationCOO_MPIAIJ()
6678 coo->Annz2 = Annz2; in MatSetPreallocationCOO_MPIAIJ()
6758 for (PetscCount i = 0; i < coo->Annz2; i++) { in MatSetValuesCOO_MPIAIJ()
|