Searched refs:Bperm1 (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 30 PetscCallHIP(hipFree(coo->Bperm1)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
92 PetscCallHIP(hipMalloc((void **)&coo_d->Bperm1, coo_h->Btot1 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
106 …PetscCallHIP(hipMemcpy(coo_d->Bperm1, coo_h->Bperm1, coo_h->Btot1 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
127 …calar Aa[], PetscCount Bnnz, const PetscCount Bjmap1[], const PetscCount Bperm1[], PetscScalar Ba[… in MatAddLocalCOOValues() argument
138 for (PetscCount k = Bjmap1[i]; k < Bjmap1[i + 1]; k++) sum += kv[Bperm1[k]]; in MatAddLocalCOOValues()
187 const auto &Bperm1 = coo->Bperm1; 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 | 30 PetscCallCUDA(cudaFree(coo->Bperm1)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
92 PetscCallCUDA(cudaMalloc((void **)&coo_d->Bperm1, coo_h->Btot1 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
106 …PetscCallCUDA(cudaMemcpy(coo_d->Bperm1, coo_h->Bperm1, coo_h->Btot1 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
127 …calar Aa[], PetscCount Bnnz, const PetscCount Bjmap1[], const PetscCount Bperm1[], PetscScalar Ba[… in MatAddLocalCOOValues() argument
138 for (PetscCount k = Bjmap1[i]; k < Bjmap1[i + 1]; k++) sum += kv[Bperm1[k]]; in MatAddLocalCOOValues()
187 const auto &Bperm1 = coo->Bperm1; 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
1527 …Bperm1 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bperm1, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1607 const auto &Bperm1 = coo->Bperm1; 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 | 6324 PetscCall(PetscFree(coo->Bperm1)); 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()
6687 coo->Bperm1 = Bperm1; in MatSetPreallocationCOO_MPIAIJ()
6713 const PetscCount *Aperm1, *Aperm2, *Bperm1, *Bperm2; in MatSetValuesCOO_MPIAIJ() local
6732 Bperm1 = coo->Bperm1; in MatSetValuesCOO_MPIAIJ()
6752 for (PetscCount k = Bjmap1[i]; k < Bjmap1[i + 1]; k++) sum += v[Bperm1[k]]; in MatSetValuesCOO_MPIAIJ()
|