| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 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 …MatAddLocalCOOValues(const PetscScalar kv[], InsertMode imode, PetscCount Annz, const PetscCount A… in MatAddLocalCOOValues() argument
131 for (; i < Annz + Bnnz; i += grid_size) { in MatAddLocalCOOValues()
133 if (i < Annz) { in MatAddLocalCOOValues()
137 i -= Annz; in MatAddLocalCOOValues()
173 const auto &Annz = coo->Annz; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
215 if (Annz + Bnnz > 0) { in MatSetValuesCOO_MPIAIJHIPSPARSE()
216 …EL_NAME(MatAddLocalCOOValues), dim3((Annz + Bnnz + 255) / 256), dim3(256), 0, PetscDefaultHipStrea… in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/tests/ |
| H A D | ex251.c | 19 PetscInt Annz = PETSC_STATIC_ARRAY_LENGTH(Ai); in main() local
34 …for (k = 0; k < Annz; k++) PetscCall(MatSetValue(A, Ai[k], Aj[k], Ai[k] + Aj[k] + 1.0, INSERT_VALU… in main()
|
| H A D | ex250.c | 19 PetscInt Annz = PETSC_STATIC_ARRAY_LENGTH(Ai); in main() local
34 …for (k = 0; k < Annz; k++) PetscCall(MatSetValue(A, Ai[k], Aj[k], Ai[k] + Aj[k] + 1.0, INSERT_VALU… in main()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 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 …MatAddLocalCOOValues(const PetscScalar kv[], InsertMode imode, PetscCount Annz, const PetscCount A… in MatAddLocalCOOValues() argument
131 for (; i < Annz + Bnnz; i += grid_size) { in MatAddLocalCOOValues()
133 if (i < Annz) { in MatAddLocalCOOValues()
137 i -= Annz; in MatAddLocalCOOValues()
173 const auto &Annz = coo->Annz; in MatSetValuesCOO_MPIAIJCUSPARSE() local
215 if (Annz + Bnnz > 0) { in MatSetValuesCOO_MPIAIJCUSPARSE()
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 | 1505 PetscCount Annz, Bnnz; member
1520 Annz = coo_h->Annz; in MatCOOStruct_MPIAIJKokkos()
1524 …okkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Ajmap1, coo_h->Annz + 1)); in MatCOOStruct_MPIAIJKokkos()
1593 const auto &Annz = coo->Annz; in MatSetValuesCOO_MPIAIJKokkos() local
1634 Kokkos::RangePolicy<>(exec, 0, Annz + Bnnz), KOKKOS_LAMBDA(PetscCount i) { in MatSetValuesCOO_MPIAIJKokkos()
1636 if (i < Annz) { in MatSetValuesCOO_MPIAIJKokkos()
1640 i -= Annz; in MatSetValuesCOO_MPIAIJKokkos()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.h | 79 PetscCount Annz, Bnnz; /* Number of entries in diagonal A and off-diagonal B */ member
|
| H A D | mpiaij.c | 6175 …PetscCount Annz = 0, Bnnz = 0; /* Number of unique nonzeros in the diagonal and off-diagonal bloc… in MatSplitEntries_Internal() local
6219 Annz++; in MatSplitEntries_Internal()
6236 PetscCall(PetscMalloc1(Annz + 1, &Ajmap)); in MatSplitEntries_Internal()
6240 Ajmap[0] = Bjmap[0] = Atot = Btot = Annz = Bnnz = 0; in MatSplitEntries_Internal()
6257 Ajmap[Annz + 1] = Ajmap[Annz] + (p - q); in MatSplitEntries_Internal()
6258 Annz++; in MatSplitEntries_Internal()
6273 *Annz_ = Annz; in MatSplitEntries_Internal()
6585 PetscInt Annz = Ai[m]; in MatSetPreallocationCOO_MPIAIJ() local
6589 PetscCall(PetscMalloc1(Annz + 1, &Ajmap1_new)); in MatSetPreallocationCOO_MPIAIJ()
6592 PetscCall(ExpandJmap_Internal(Annz1, Annz, Aimap1, Ajmap1, Ajmap1_new)); in MatSetPreallocationCOO_MPIAIJ()
[all …]
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3785 PetscCount Annz = seq->nz; in MatSetValuesCOO_SeqAIJHIPSPARSE() local
3808 if (Annz) { in MatSetValuesCOO_SeqAIJHIPSPARSE()
3809 …GGL(HIP_KERNEL_NAME(MatAddCOOValues), dim3((Annz + 255) / 256), dim3(256), 0, PetscDefaultHipStrea… in MatSetValuesCOO_SeqAIJHIPSPARSE()
4098 PetscInt Annz, Bnnz; in MatSeqAIJHIPSPARSEMergeMats() local
4148 Annz = (PetscInt)Acsr->column_indices->size(); in MatSeqAIJHIPSPARSEMergeMats()
4150 c->nz = Annz + Bnnz; in MatSeqAIJHIPSPARSEMergeMats()
4157 auto Acoo = new THRUSTINTARRAY32(Annz); in MatSeqAIJHIPSPARSEMergeMats()
4179 …PetscCallHIPSPARSE(hipsparseXcsr2coo(Acusp->handle, Aroff->data().get(), Annz, m, Acoo->data().get… in MatSeqAIJHIPSPARSEMergeMats()
4186 auto wPerm = new THRUSTINTARRAY32(Annz + Bnnz); in MatSeqAIJHIPSPARSEMergeMats()
4194 thrust::advance(p2, Annz); in MatSeqAIJHIPSPARSEMergeMats()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4428 PetscCount Annz = seq->nz; in MatSetValuesCOO_SeqAIJCUSPARSE() local
4451 if (Annz) { in MatSetValuesCOO_SeqAIJCUSPARSE()
4452 MatAddCOOValues<<<((int)(Annz + 255) / 256), 256>>>(v1, Annz, coo->jmap, coo->perm, imode, Aa); in MatSetValuesCOO_SeqAIJCUSPARSE()
4744 PetscInt Annz, Bnnz; in MatSeqAIJCUSPARSEMergeMats() local
4795 Annz = (PetscInt)Acsr->column_indices->size(); in MatSeqAIJCUSPARSEMergeMats()
4797 c->nz = Annz + Bnnz; in MatSeqAIJCUSPARSEMergeMats()
4804 auto Acoo = new THRUSTINTARRAY32(Annz); in MatSeqAIJCUSPARSEMergeMats()
4826 …stat = cusparseXcsr2coo(Acusp->handle, Aroff->data().get(), Annz, m, Acoo->data().get(), CUSPARSE_… in MatSeqAIJCUSPARSEMergeMats()
4842 auto wPerm = new THRUSTINTARRAY32(Annz + Bnnz); in MatSeqAIJCUSPARSEMergeMats()
4851 cuda::std::advance(p2, Annz); in MatSeqAIJCUSPARSEMergeMats()
[all …]
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1406 const auto &Annz = coo->nz; in MatSetValuesCOO_SeqAIJKokkos() local
1422 …Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, Annz), KOKKOS_LAMBDA(const PetscCount i) { in MatSetValuesCOO_SeqAIJKokkos()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 4742 PetscCount i, j, Annz = aseq->nz; in MatSetValuesCOO_SeqAIJ() local
4755 for (i = 0; i < Annz; i++) { in MatSetValuesCOO_SeqAIJ()
|