Lines Matching refs:maps
27 P4estVertexMaps *maps = *(P4estVertexMaps **)ptr; in LandauGPUMapsDestroy() local
31 if (maps[0].deviceType != LANDAU_CPU) { in LandauGPUMapsDestroy()
33 …if (maps[0].deviceType == LANDAU_KOKKOS) PetscCall(LandauKokkosDestroyMatMaps(maps, maps[0].numgri… in LandauGPUMapsDestroy()
37 for (PetscInt grid = 0; grid < maps[0].numgrids; grid++) { in LandauGPUMapsDestroy()
38 PetscCall(PetscFree(maps[grid].c_maps)); in LandauGPUMapsDestroy()
39 PetscCall(PetscFree(maps[grid].gIdx)); in LandauGPUMapsDestroy()
41 PetscCall(PetscFree(maps)); in LandauGPUMapsDestroy()
96 P4estVertexMaps *maps; in LandauFormJacobian_Internal() local
110 PetscCall(PetscContainerGetPointer(container, &maps)); in LandauFormJacobian_Internal()
111 PetscCheck(maps, ctx->comm, PETSC_ERR_ARG_WRONG, "empty GPU matrix container"); in LandauFormJacobian_Internal()
118 maps = NULL; in LandauFormJacobian_Internal()
259 LandauIdx *const Idxs = &maps[grid].gIdx[loc_elem][f][0]; in LandauFormJacobian_Internal()
267 for (q = 0; q < maps[grid].num_face; q++) { in LandauFormJacobian_Internal()
268 PetscInt id = maps[grid].c_maps[idx][q].gid; in LandauFormJacobian_Internal()
269 PetscScalar scale = maps[grid].c_maps[idx][q].scale; in LandauFormJacobian_Internal()
487 LandauIdx *const Idxs = &maps[grid].gIdx[loc_elem][fieldA][0]; in LandauFormJacobian_Internal()
496 for (q = 0, nr = 0; q < maps[grid].num_face; q++, nr++) { in LandauFormJacobian_Internal()
497 if (maps[grid].c_maps[idx][q].gid < 0) break; in LandauFormJacobian_Internal()
498 rows0[q] = maps[grid].c_maps[idx][q].gid; in LandauFormJacobian_Internal()
499 row_scale[q] = maps[grid].c_maps[idx][q].scale; in LandauFormJacobian_Internal()
510 nc = maps[grid].num_face; in LandauFormJacobian_Internal()
511 for (q = 0, nc = 0; q < maps[grid].num_face; q++, nc++) { in LandauFormJacobian_Internal()
512 if (maps[grid].c_maps[idx][q].gid < 0) break; in LandauFormJacobian_Internal()
513 cols0[q] = maps[grid].c_maps[idx][q].gid; in LandauFormJacobian_Internal()
514 col_scale[q] = maps[grid].c_maps[idx][q].scale; in LandauFormJacobian_Internal()
1480 P4estVertexMaps *maps; in CreateStaticData() local
1486 PetscCall(PetscMalloc(sizeof(*maps) * ctx->num_grids, &maps)); in CreateStaticData()
1507 maps[grid].d_self = NULL; in CreateStaticData()
1508 maps[grid].num_elements = numCells[grid]; in CreateStaticData()
1509 maps[grid].num_face = (PetscInt)(pow(Nq, 1. / ((double)dim)) + .001); // Q in CreateStaticData()
1510 … maps[grid].num_face = (PetscInt)(pow(maps[grid].num_face, (double)(dim - 1)) + .001); // Q^2 in CreateStaticData()
1511 maps[grid].num_reduced = 0; in CreateStaticData()
1512 maps[grid].deviceType = ctx->deviceType; in CreateStaticData()
1513 maps[grid].numgrids = ctx->num_grids; in CreateStaticData()
1515 PetscCall(PetscMalloc(maps[grid].num_elements * sizeof(*maps[grid].gIdx), &maps[grid].gIdx)); in CreateStaticData()
1528 … for (PetscInt q = 0; q < numindices; ++q) maps[grid].gIdx[eidx][fieldA][q] = indices[q]; in CreateStaticData()
1536 maps[grid].gIdx[eidx][fieldA][q] = plex_batch[indices[f]]; in CreateStaticData()
1538 maps[grid].gIdx[eidx][fieldA][q] = indices[f]; in CreateStaticData()
1545 …maps[grid].gIdx[eidx][fieldA][q] = -maps[grid].num_reduced - 1; // store (-)index: id = -(idx+1): … in CreateStaticData()
1549 …for (ii = 0, pointMaps[maps[grid].num_reduced][jj].scale = 0; ii < maps[grid].num_face; ii++) { //… in CreateStaticData()
1551 … pointMaps[maps[grid].num_reduced][jj].scale += PetscRealPart(elMat[f * numindices + ff + ii]); in CreateStaticData()
1554 sum += pointMaps[maps[grid].num_reduced][jj].scale; // diagnostic in CreateStaticData()
1556 …if (pointMaps[maps[grid].num_reduced][jj].scale == 0) pointMaps[maps[grid].num_reduced][jj].gid = … in CreateStaticData()
1559 pointMaps[maps[grid].num_reduced][jj].gid = plex_batch[indices[f]]; in CreateStaticData()
1561 pointMaps[maps[grid].num_reduced][jj].gid = indices[f]; in CreateStaticData()
1565 … } while (++jj < maps[grid].num_face && ++f < numindices); // jj is incremented if we hit the end in CreateStaticData()
1566 while (jj < maps[grid].num_face) { in CreateStaticData()
1567 pointMaps[maps[grid].num_reduced][jj].scale = 0; in CreateStaticData()
1568 pointMaps[maps[grid].num_reduced][jj].gid = -1; in CreateStaticData()
1575 …face=%" PetscInt_FMT ")\n", eidx, q, fieldA, (double)sum, LANDAU_MAX_Q_FACE, maps[grid].num_face)); in CreateStaticData()
1582 maps[grid].num_reduced++; in CreateStaticData()
1583 …scCheck(maps[grid].num_reduced < MAP_BF_SIZE, PETSC_COMM_SELF, PETSC_ERR_PLIB, "maps[grid].num_red… in CreateStaticData()
1603 … PetscCall(PetscMalloc(maps[grid].num_reduced * sizeof(*maps[grid].c_maps), &maps[grid].c_maps)); in CreateStaticData()
1604 for (PetscInt ej = 0; ej < maps[grid].num_reduced; ++ej) { in CreateStaticData()
1605 for (PetscInt q = 0; q < maps[grid].num_face; ++q) { in CreateStaticData()
1606 maps[grid].c_maps[ej][q].scale = pointMaps[ej][q].scale; in CreateStaticData()
1607 maps[grid].c_maps[ej][q].gid = pointMaps[ej][q].gid; in CreateStaticData()
1611 …if (ctx->deviceType == LANDAU_KOKKOS) PetscCall(LandauKokkosCreateMatMaps(maps, pointMaps, Nf, gri… in CreateStaticData()
1628 …const LandauIdx *const Idxs = &maps[grid].gIdx[ej][0][0]; // just use field-0 maps, Th… in CreateStaticData()
1638 for (PetscInt q = 0; q < maps[grid].num_face; q++) { in CreateStaticData()
1639 if (maps[grid].c_maps[idx][q].gid < 0) break; in CreateStaticData()
1657 const LandauIdx *const Idxs = &maps[grid].gIdx[ej][fieldA][0]; in CreateStaticData()
1664 for (PetscInt q = 0; q < nr; q++) rows[q] = maps[grid].c_maps[idx][q].gid; in CreateStaticData()
1671 for (PetscInt q = 0; q < nc; q++) cols[q] = maps[grid].c_maps[idx][q].gid; in CreateStaticData()
1692 PetscCall(PetscContainerSetPointer(container, (void *)maps)); in CreateStaticData()