| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 27 PetscCallHIP(hipFree(coo->Ajmap1)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
89 PetscCallHIP(hipMalloc((void **)&coo_d->Ajmap1, (coo_h->Annz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
103 …PetscCallHIP(hipMemcpy(coo_d->Ajmap1, coo_h->Ajmap1, (coo_h->Annz + 1) * sizeof(PetscCount), hipMe… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
127 …st PetscScalar kv[], InsertMode imode, PetscCount Annz, const PetscCount Ajmap1[], const PetscCoun… in MatAddLocalCOOValues() argument
134 for (PetscCount k = Ajmap1[i]; k < Ajmap1[i + 1]; k++) sum += kv[Aperm1[k]]; in MatAddLocalCOOValues()
179 const auto &Ajmap1 = coo->Ajmap1; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
216 …nnz + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz,… in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 27 PetscCallCUDA(cudaFree(coo->Ajmap1)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
89 PetscCallCUDA(cudaMalloc((void **)&coo_d->Ajmap1, (coo_h->Annz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
103 …PetscCallCUDA(cudaMemcpy(coo_d->Ajmap1, coo_h->Ajmap1, (coo_h->Annz + 1) * sizeof(PetscCount), cud… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
127 …st PetscScalar kv[], InsertMode imode, PetscCount Annz, const PetscCount Ajmap1[], const PetscCoun… in MatAddLocalCOOValues() argument
134 for (PetscCount k = Ajmap1[i]; k < Ajmap1[i + 1]; k++) sum += kv[Aperm1[k]]; in MatAddLocalCOOValues()
179 const auto &Ajmap1 = coo->Ajmap1; in MatSetValuesCOO_MPIAIJCUSPARSE() local
216 …MatAddLocalCOOValues<<<(int)((Annz + Bnnz + 255) / 256), 256>>>(v1, imode, Annz, Ajmap1, Aperm1, A… in MatSetValuesCOO_MPIAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1507 PetscCountKokkosView Ajmap1, Aperm1; member
1524 …Ajmap1 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Ajmap1, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1599 const auto &Ajmap1 = coo->Ajmap1; in MatSetValuesCOO_MPIAIJKokkos() local
1637 for (PetscCount k = Ajmap1(i); k < Ajmap1(i + 1); k++) sum += v1(Aperm1(k)); in MatSetValuesCOO_MPIAIJKokkos()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.h | 82 PetscCount *Ajmap1, *Aperm1; /* Lengths: [Annz+1], [Atot1]. Local entries to diag */ member
|
| H A D | mpiaij.c | 6325 PetscCall(PetscFree(coo->Ajmap1)); in MatCOOStructDestroy_MPIAIJ()
6554 PetscCount *Ajmap1, *Aperm1, *Bjmap1, *Bperm1; in MatSetPreallocationCOO_MPIAIJ() local
6562 …m, i1, j1, perm1, rowBegin1, rowMid1, rowEnd1, &Atot1, &Aperm1, &Annz1, &Ajmap1, &Btot1, &Bperm1, … in MatSetPreallocationCOO_MPIAIJ()
6580 …PetscCall(MatMergeEntries_Internal(mat, j1, j2, rowBegin1, rowMid1, rowBegin2, rowMid2, Ajmap1, Aj… in MatSetPreallocationCOO_MPIAIJ()
6592 PetscCall(ExpandJmap_Internal(Annz1, Annz, Aimap1, Ajmap1, Ajmap1_new)); in MatSetPreallocationCOO_MPIAIJ()
6596 PetscCall(PetscFree(Ajmap1)); in MatSetPreallocationCOO_MPIAIJ()
6604 Ajmap1 = Ajmap1_new; in MatSetPreallocationCOO_MPIAIJ()
6684 coo->Ajmap1 = Ajmap1; in MatSetPreallocationCOO_MPIAIJ()
6711 const PetscCount *Ajmap1, *Ajmap2, *Aimap2; in MatSetValuesCOO_MPIAIJ() local
6724 Ajmap1 = coo->Ajmap1; in MatSetValuesCOO_MPIAIJ()
[all …]
|