Searched refs:coo_elem_offsets (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 253 …Traits<Kokkos::Unmanaged>> h_coo_elem_offsets((LandauIdx *)SData_d->coo_elem_offsets, SData_d->coo… in LandauKokkosStaticDataSet()
254 … coo_elem_offsets = new Kokkos:… in LandauKokkosStaticDataSet() local
260 Kokkos::deep_copy(*coo_elem_offsets, h_coo_elem_offsets); in LandauKokkosStaticDataSet()
264 …PetscCall(PetscFree3(SData_d->coo_elem_offsets, SData_d->coo_elem_fullNb, SData_d->coo_elem_point_… in LandauKokkosStaticDataSet()
265 SData_d->coo_elem_offsets = static_cast<void *>(coo_elem_offsets); in LandauKokkosStaticDataSet()
325 …auto coo_elem_offsets = static_cast<Kokkos::View<LandauIdx *, Kokkos::LayoutLeft> *>((void *)SData… in LandauKokkosStaticDataClear() local
326 delete coo_elem_offsets; in LandauKokkosStaticDataClear()
331 SData_d->coo_elem_offsets = NULL; in LandauKokkosStaticDataClear()
343 …nst PetscInt fieldA, const P4estVertexMaps *d_maps, const LandauIdx coo_elem_offsets[], const Land… in landau_mat_assemble() argument
373 …const int idx0 = bid_coo_sz_batch + coo_elem_offsets[glb_elem_idx] + fieldA * fullNb2 + fullNb * c… in landau_mat_assemble()
[all …]
|
| /petsc/src/ts/utils/dmplexlandau/ |
| H A D | plexland.c | 484 …LandauIdx *coo_elem_offsets = (LandauIdx *)ctx->SData_d.coo_elem_offsets, *coo_elem_fullNb = (Land… in LandauFormJacobian_Internal() local
522 …const PetscInt idx0 = b_id * coo_elem_offsets[elem_offset[num_grids]] + coo_elem_offsets[glb_elem_… in LandauFormJacobian_Internal()
1138 ctx->SData_d.coo_elem_offsets = NULL; in ProcessOptions()
1482 …LandauIdx *coo_elem_offsets = NULL, *coo_elem_fullNb = NULL, (*coo_elem_point_offsets)[LANDA… in CreateStaticData() local
1491 …PetscCall(PetscMalloc3(ncellsTot + 1, &coo_elem_offsets, ncellsTot, &coo_elem_fullNb, ncellsTot, &… in CreateStaticData()
1492 …coo_elem_offsets[0] = 0; … in CreateStaticData()
1495 ctx->SData_d.coo_elem_offsets = (void *)coo_elem_offsets; in CreateStaticData()
1517 …if (coo_elem_offsets) coo_elem_offsets[glb_elem_idx + 1] = coo_elem_offsets[glb_elem_idx]; // star… in CreateStaticData()
1594 …coo_elem_offsets[glb_elem_idx + 1] += fullNb * fullNb; // one species block, adds a block for each… in CreateStaticData()
1621 ctx->SData_d.coo_size = coo_elem_offsets[ncellsTot] * ctx->batch_sz; in CreateStaticData()
[all …]
|
| /petsc/include/ |
| H A D | petsclandau.h | 90 void *coo_elem_offsets; member
|