Lines Matching refs:maps

93 PetscErrorCode LandauKokkosCreateMatMaps(P4estVertexMaps maps[], pointInterpolationP4est (*pointMaps)[LANDAU_MAX_Q_FACE], PetscInt Nf[], PetscInt grid)
96   const Kokkos::View<pointInterpolationP4est *[LANDAU_MAX_Q_FACE], Kokkos::LayoutRight, Kokkos::HostSpace, Kokkos::MemoryTraits<Kokkos::Unmanaged>>     h_points((pointInterpolationP4est *)pointMaps, maps[grid].num_reduced);
97   const Kokkos::View<LandauIdx *[LANDAU_MAX_SPECIES][LANDAU_MAX_NQND], Kokkos::LayoutRight, Kokkos::HostSpace, Kokkos::MemoryTraits<Kokkos::Unmanaged>> h_gidx((LandauIdx *)maps[grid].gIdx, maps[grid].num_elements);
98   Kokkos::View<pointInterpolationP4est *[LANDAU_MAX_Q_FACE], Kokkos::LayoutRight>     *d_points = new Kokkos::View<pointInterpolationP4est *[LANDAU_MAX_Q_FACE], Kokkos::LayoutRight>("points", maps[grid].num_reduced);
99   Kokkos::View<LandauIdx *[LANDAU_MAX_SPECIES][LANDAU_MAX_NQND], Kokkos::LayoutRight> *d_gidx   = new Kokkos::View<LandauIdx *[LANDAU_MAX_SPECIES][LANDAU_MAX_NQND], Kokkos::LayoutRight>("gIdx", maps[grid].num_elements);
104   h_maps.num_elements = maps[grid].num_elements;
105   h_maps.num_face     = maps[grid].num_face;
106   h_maps.num_reduced  = maps[grid].num_reduced;
107   h_maps.deviceType   = maps[grid].deviceType;
108   h_maps.numgrids     = maps[grid].numgrids;
111   maps[grid].vp1      = (void *)d_points;
113   maps[grid].vp2      = (void *)d_gidx;
118     maps[grid].d_self = d_maps_k->data();
119     maps[grid].vp3    = (void *)d_maps_k;
123 PetscErrorCode LandauKokkosDestroyMatMaps(P4estVertexMaps maps[], PetscInt num_grids)
127     auto a = static_cast<Kokkos::View<pointInterpolationP4est *[LANDAU_MAX_Q_FACE], Kokkos::LayoutRight> *>(maps[grid].vp1);
128     auto b = static_cast<Kokkos::View<LandauIdx *[LANDAU_MAX_SPECIES][LANDAU_MAX_NQND], Kokkos::LayoutRight> *>(maps[grid].vp2);
129     auto c = static_cast<Kokkos::View<P4estVertexMaps *> *>(maps[grid].vp3);
271   SData_d->maps = NULL; // not used
392   P4estVertexMaps *maps[LANDAU_MAX_GRIDS]; // this gets captured
462       maps[grid] = h_maps[grid].d_self;
467     for (PetscInt grid = 0; grid < num_grids; grid++) maps[grid] = NULL;
527             if (!maps[grid]) {
535                   LandauIdx idx = maps[grid]->gIdx[loc_elem][f][b];
541                     for (int q = 0; q < maps[grid]->num_face; q++) {
542                       PetscInt    id    = maps[grid]->c_maps[idx][q].gid;
543                       PetscScalar scale = maps[grid]->c_maps[idx][q].scale;
712                   static_cast<void>(landau_mat_assemble(d_coo_vals, t, f, g, Nb, moffset, loc_elem, fieldA, maps[grid], d_coo_elem_offsets, d_coo_elem_fullNb, d_coo_elem_point_offsets, b_elem_idx, b_id * coo_sz_batch));
769                   static_cast<void>(landau_mat_assemble(d_coo_vals, t, f, g, Nb, moffset, loc_elem, fieldA, maps[grid], d_coo_elem_offsets, d_coo_elem_fullNb, d_coo_elem_point_offsets, b_elem_idx, b_id * coo_sz_batch));