| /petsc/src/mat/tests/ |
| H A D | ex251.c | 20 PetscInt Bnnz = PETSC_STATIC_ARRAY_LENGTH(Bi); in main() local
48 …for (k = 0; k < Bnnz; k++) PetscCall(MatSetValue(B, Bi[k], Bj[k], Bi[k] + Bj[k] + 2.0, INSERT_VALU… in main()
|
| H A D | ex250.c | 20 PetscInt Bnnz = PETSC_STATIC_ARRAY_LENGTH(Bi); in main() local
48 …for (k = 0; k < Bnnz; k++) PetscCall(MatSetValue(B, Bi[k], Bj[k], Bi[k] + Bj[k] + 2.0, INSERT_VALU… in main()
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 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 …scCount Ajmap1[], const PetscCount Aperm1[], PetscScalar Aa[], PetscCount Bnnz, const PetscCount B… in MatAddLocalCOOValues() argument
131 for (; i < Annz + Bnnz; i += grid_size) { in MatAddLocalCOOValues()
175 const auto &Bnnz = coo->Bnnz; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
215 if (Annz + Bnnz > 0) { in MatSetValuesCOO_MPIAIJHIPSPARSE()
216 …alCOOValues), dim3((Annz + Bnnz + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v1, imode, Ann… in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 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 …scCount Ajmap1[], const PetscCount Aperm1[], PetscScalar Aa[], PetscCount Bnnz, const PetscCount B… in MatAddLocalCOOValues() argument
131 for (; i < Annz + Bnnz; i += grid_size) { in MatAddLocalCOOValues()
175 const auto &Bnnz = coo->Bnnz; 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
1521 Bnnz = coo_h->Bnnz; in MatCOOStruct_MPIAIJKokkos()
1526 …okkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bjmap1, coo_h->Bnnz + 1)); in MatCOOStruct_MPIAIJKokkos()
1595 const auto &Bnnz = coo->Bnnz; in MatSetValuesCOO_MPIAIJKokkos() local
1634 Kokkos::RangePolicy<>(exec, 0, Annz + Bnnz), KOKKOS_LAMBDA(PetscCount i) { 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
6228 Bnnz++; in MatSplitEntries_Internal()
6237 PetscCall(PetscMalloc1(Bnnz + 1, &Bjmap)); in MatSplitEntries_Internal()
6240 Ajmap[0] = Bjmap[0] = Atot = Btot = Annz = Bnnz = 0; in MatSplitEntries_Internal()
6267 Bjmap[Bnnz + 1] = Bjmap[Bnnz] + (p - q); in MatSplitEntries_Internal()
6268 Bnnz++; in MatSplitEntries_Internal()
6277 *Bnnz_ = Bnnz; in MatSplitEntries_Internal()
6586 PetscInt Bnnz = Bi[m]; in MatSetPreallocationCOO_MPIAIJ() local
6590 PetscCall(PetscMalloc1(Bnnz + 1, &Bjmap1_new)); in MatSetPreallocationCOO_MPIAIJ()
6593 PetscCall(ExpandJmap_Internal(Bnnz1, Bnnz, Bimap1, Bjmap1, Bjmap1_new)); in MatSetPreallocationCOO_MPIAIJ()
[all …]
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 4098 PetscInt Annz, Bnnz; in MatSeqAIJHIPSPARSEMergeMats() local
4149 Bnnz = (PetscInt)Bcsr->column_indices->size(); in MatSeqAIJHIPSPARSEMergeMats()
4150 c->nz = Annz + Bnnz; in MatSeqAIJHIPSPARSEMergeMats()
4158 auto Bcoo = new THRUSTINTARRAY32(Bnnz); in MatSeqAIJHIPSPARSEMergeMats()
4180 …PetscCallHIPSPARSE(hipsparseXcsr2coo(Bcusp->handle, Broff->data().get(), Bnnz, m, Bcoo->data().get… in MatSeqAIJHIPSPARSEMergeMats()
4186 auto wPerm = new THRUSTINTARRAY32(Annz + Bnnz); in MatSeqAIJHIPSPARSEMergeMats()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4744 PetscInt Annz, Bnnz; in MatSeqAIJCUSPARSEMergeMats() local
4796 Bnnz = (PetscInt)Bcsr->column_indices->size(); in MatSeqAIJCUSPARSEMergeMats()
4797 c->nz = Annz + Bnnz; in MatSeqAIJCUSPARSEMergeMats()
4805 auto Bcoo = new THRUSTINTARRAY32(Bnnz); in MatSeqAIJCUSPARSEMergeMats()
4828 …stat = cusparseXcsr2coo(Bcusp->handle, Broff->data().get(), Bnnz, m, Bcoo->data().get(), CUSPARSE_… in MatSeqAIJCUSPARSEMergeMats()
4842 auto wPerm = new THRUSTINTARRAY32(Annz + Bnnz); in MatSeqAIJCUSPARSEMergeMats()
|