| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 29 PetscCallHIP(hipFree(coo->Bjmap1)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
91 PetscCallHIP(hipMalloc((void **)&coo_d->Bjmap1, (coo_h->Bnnz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
105 …PetscCallHIP(hipMemcpy(coo_d->Bjmap1, coo_h->Bjmap1, (coo_h->Bnnz + 1) * sizeof(PetscCount), hipMe… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
127 …PetscCount Aperm1[], PetscScalar Aa[], PetscCount Bnnz, const PetscCount Bjmap1[], const PetscCoun… in MatAddLocalCOOValues() argument
138 for (PetscCount k = Bjmap1[i]; k < Bjmap1[i + 1]; k++) sum += kv[Bperm1[k]]; in MatAddLocalCOOValues()
182 const auto &Bjmap1 = coo->Bjmap1; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
216 …im3(256), 0, PetscDefaultHipStream, v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm1, Ba); in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 29 PetscCallCUDA(cudaFree(coo->Bjmap1)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
91 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bjmap1, (coo_h->Bnnz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
105 …PetscCallCUDA(cudaMemcpy(coo_d->Bjmap1, coo_h->Bjmap1, (coo_h->Bnnz + 1) * sizeof(PetscCount), cud… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
127 …PetscCount Aperm1[], PetscScalar Aa[], PetscCount Bnnz, const PetscCount Bjmap1[], const PetscCoun… in MatAddLocalCOOValues() argument
138 for (PetscCount k = Bjmap1[i]; k < Bjmap1[i + 1]; k++) sum += kv[Bperm1[k]]; in MatAddLocalCOOValues()
182 const auto &Bjmap1 = coo->Bjmap1; in MatSetValuesCOO_MPIAIJCUSPARSE() local
216 …((Annz + Bnnz + 255) / 256), 256>>>(v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm1, Ba); in MatSetValuesCOO_MPIAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1508 PetscCountKokkosView Bjmap1, Bperm1; member
1526 …Bjmap1 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bjmap1, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1602 const auto &Bjmap1 = coo->Bjmap1; in MatSetValuesCOO_MPIAIJKokkos() local
1641 for (PetscCount k = Bjmap1(i); k < Bjmap1(i + 1); k++) sum += v1(Bperm1(k)); in MatSetValuesCOO_MPIAIJKokkos()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.h | 83 …PetscCount *Bjmap1, *Bperm1; /* Lengths: [Bnnz+1], [Btot1]. Local entries to offdiag */ member
|
| H A D | mpiaij.c | 6326 PetscCall(PetscFree(coo->Bjmap1)); in MatCOOStructDestroy_MPIAIJ()
6554 PetscCount *Ajmap1, *Aperm1, *Bjmap1, *Bperm1; in MatSetPreallocationCOO_MPIAIJ() local
6562 … rowBegin1, rowMid1, rowEnd1, &Atot1, &Aperm1, &Annz1, &Ajmap1, &Btot1, &Bperm1, &Bnnz1, &Bjmap1)); in MatSetPreallocationCOO_MPIAIJ()
6581 …PetscCall(MatMergeEntries_Internal(mat, j1, j2, rowMid1, rowEnd1, rowMid2, rowEnd2, Bjmap1, Bjmap2… in MatSetPreallocationCOO_MPIAIJ()
6593 PetscCall(ExpandJmap_Internal(Bnnz1, Bnnz, Bimap1, Bjmap1, Bjmap1_new)); in MatSetPreallocationCOO_MPIAIJ()
6598 PetscCall(PetscFree(Bjmap1)); in MatSetPreallocationCOO_MPIAIJ()
6605 Bjmap1 = Bjmap1_new; in MatSetPreallocationCOO_MPIAIJ()
6686 coo->Bjmap1 = Bjmap1; in MatSetPreallocationCOO_MPIAIJ()
6712 const PetscCount *Bjmap1, *Bjmap2, *Bimap2; in MatSetValuesCOO_MPIAIJ() local
6727 Bjmap1 = coo->Bjmap1; in MatSetValuesCOO_MPIAIJ()
[all …]
|