Searched refs:Aperm1 (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 28 PetscCallHIP(hipFree(coo->Aperm1)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
90 PetscCallHIP(hipMalloc((void **)&coo_d->Aperm1, coo_h->Atot1 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
104 …PetscCallHIP(hipMemcpy(coo_d->Aperm1, coo_h->Aperm1, coo_h->Atot1 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
127 …Mode imode, PetscCount Annz, const PetscCount Ajmap1[], const PetscCount Aperm1[], PetscScalar Aa[… in MatAddLocalCOOValues() argument
134 for (PetscCount k = Ajmap1[i]; k < Ajmap1[i + 1]; k++) sum += kv[Aperm1[k]]; in MatAddLocalCOOValues()
185 const auto &Aperm1 = coo->Aperm1; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
216 …5) / 256), dim3(256), 0, PetscDefaultHipStream, v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1,… in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 28 PetscCallCUDA(cudaFree(coo->Aperm1)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
90 PetscCallCUDA(cudaMalloc((void **)&coo_d->Aperm1, coo_h->Atot1 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
104 …PetscCallCUDA(cudaMemcpy(coo_d->Aperm1, coo_h->Aperm1, coo_h->Atot1 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
127 …Mode imode, PetscCount Annz, const PetscCount Ajmap1[], const PetscCount Aperm1[], PetscScalar Aa[… in MatAddLocalCOOValues() argument
134 for (PetscCount k = Ajmap1[i]; k < Ajmap1[i + 1]; k++) sum += kv[Aperm1[k]]; in MatAddLocalCOOValues()
185 const auto &Aperm1 = coo->Aperm1; 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
1525 …Aperm1 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aperm1, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1605 const auto &Aperm1 = coo->Aperm1; 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 | 6323 PetscCall(PetscFree(coo->Aperm1)); in MatCOOStructDestroy_MPIAIJ()
6554 PetscCount *Ajmap1, *Aperm1, *Bjmap1, *Bperm1; in MatSetPreallocationCOO_MPIAIJ() local
6562 …_Internal(mat, rem, i1, j1, perm1, rowBegin1, rowMid1, rowEnd1, &Atot1, &Aperm1, &Annz1, &Ajmap1, … in MatSetPreallocationCOO_MPIAIJ()
6685 coo->Aperm1 = Aperm1; in MatSetPreallocationCOO_MPIAIJ()
6713 const PetscCount *Aperm1, *Aperm2, *Bperm1, *Bperm2; in MatSetValuesCOO_MPIAIJ() local
6730 Aperm1 = coo->Aperm1; in MatSetValuesCOO_MPIAIJ()
6747 for (PetscCount k = Ajmap1[i]; k < Ajmap1[i + 1]; k++) sum += v[Aperm1[k]]; in MatSetValuesCOO_MPIAIJ()
|