| /petsc/src/mat/impls/aij/seq/hdf5/ |
| H A D | aijhdf5.c | 21 PetscLayout jmap = NULL; in MatLoad_AIJ_HDF5() local
113 PetscCall(PetscLayoutCreateFromRanges(comm, range, PETSC_OWN_POINTER, 1, &jmap)); in MatLoad_AIJ_HDF5()
122 PetscCall(PetscLayoutDuplicate(jmap, &is_j->map)); in MatLoad_AIJ_HDF5()
129 PetscCall(PetscLayoutDuplicate(jmap, &vec_a->map)); in MatLoad_AIJ_HDF5()
152 PetscCall(PetscLayoutDestroy(&jmap)); in MatLoad_AIJ_HDF5()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm.hpp | 143 …Scalar *PETSC_RESTRICT vv, PetscCount n, const PetscCount *PETSC_RESTRICT jmap, const PetscCount *… in add_coo_values_impl() argument
146 const auto end = jmap[i + 1]; in add_coo_values_impl()
150 for (auto k = jmap[i]; k < end; ++k) sum += vv[perm[k]]; in add_coo_values_impl()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 4548 PetscCall(PetscFree(coo->jmap)); in MatCOOStructDestroy_SeqAIJ()
4564 PetscCount *perm, *jmap; in MatSetPreallocationCOO_SeqAIJ() local
4596 …PetscCall(PetscMalloc1(coo_n - nneg + 1, &jmap)); /* +1 to make a CSR-like data structure. jmap[i]… in MatSetPreallocationCOO_SeqAIJ()
4598 jmap++; /* Inc jmap by 1 for convenience */ in MatSetPreallocationCOO_SeqAIJ()
4665 jmap[q] = 1; in MatSetPreallocationCOO_SeqAIJ()
4672 jmap[q] = 1; /* Number of repeats of this nonzero entry */ in MatSetPreallocationCOO_SeqAIJ()
4679 jmap[q] = 1; in MatSetPreallocationCOO_SeqAIJ()
4684 jmap[q]++; in MatSetPreallocationCOO_SeqAIJ()
4693 jmap--; // Back to the beginning of jmap[] in MatSetPreallocationCOO_SeqAIJ()
4694 jmap[0] = 0; in MatSetPreallocationCOO_SeqAIJ()
[all …]
|
| H A D | aij.h | 174 …PetscCount *jmap; /* perm[jmap[i]..jmap[i+1]) give indices of entries in v[] associated with i-th … member
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1347 PetscCountKokkosView jmap; member
1355 …jmap = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(coo_h->j… in MatCOOStruct_SeqAIJKokkos()
1407 const auto &jmap = coo->jmap; in MatSetValuesCOO_SeqAIJKokkos() local
1424 for (PetscCount k = jmap(i); k < jmap(i + 1); k++) sum += kv(perm(k)); in MatSetValuesCOO_SeqAIJKokkos()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3725 PetscCallHIP(hipFree(coo->jmap)); in MatCOOStructDestroy_SeqAIJHIPSPARSE()
3760 PetscCallHIP(hipMalloc((void **)&coo_d->jmap, (coo_h->nz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3761 …PetscCallHIP(hipMemcpy(coo_d->jmap, coo_h->jmap, (coo_h->nz + 1) * sizeof(PetscCount), hipMemcpyHo… in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3770 … MatAddCOOValues(const PetscScalar kv[], PetscCount nnz, const PetscCount jmap[], const PetscCount… in MatAddCOOValues() argument
3776 for (PetscCount k = jmap[i]; k < jmap[i + 1]; k++) sum += kv[perm[k]]; in MatAddCOOValues()
3809 …((Annz + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v1, Annz, coo->jmap, coo->perm, imode, … in MatSetValuesCOO_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4367 PetscCallCUDA(cudaFree(coo->jmap)); in MatCOOStructDestroy_SeqAIJCUSPARSE()
4403 PetscCallCUDA(cudaMalloc((void **)&coo_d->jmap, (coo_h->nz + 1) * sizeof(PetscCount))); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4404 …PetscCallCUDA(cudaMemcpy(coo_d->jmap, coo_h->jmap, (coo_h->nz + 1) * sizeof(PetscCount), cudaMemcp… in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4413 … MatAddCOOValues(const PetscScalar kv[], PetscCount nnz, const PetscCount jmap[], const PetscCount… in MatAddCOOValues() argument
4419 for (PetscCount k = jmap[i]; k < jmap[i + 1]; k++) sum += kv[perm[k]]; in MatAddCOOValues()
4452 MatAddCOOValues<<<((int)(Annz + 255) / 256), 256>>>(v1, Annz, coo->jmap, coo->perm, imode, Aa); in MatSetValuesCOO_SeqAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 6303 …etscCount nnz1, PetscCount nnz, const PetscCount imap[], const PetscCount jmap[], PetscCount jmap_… in ExpandJmap_Internal() argument
6311 for (; p > imap[k]; p--) jmap_new[p] = jmap[k + 1]; in ExpandJmap_Internal()
6313 for (; p >= 0; p--) jmap_new[p] = jmap[0]; in ExpandJmap_Internal()
|