Lines Matching refs:num_grids
112 for (PetscInt i = 0; i < ctx->num_grids * ctx->batch_sz; i++) subJ[i] = NULL; in LandauFormJacobian_Internal()
116 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) PetscCall(DMCreateMatrix(ctx->plex[grid], &… in LandauFormJacobian_Internal()
127 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauFormJacobian_Internal()
148 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauFormJacobian_Internal()
155 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) cellClosure_sz += Nb * Nf[grid] * numCells[… in LandauFormJacobian_Internal()
163 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauFormJacobian_Internal()
204 …PetscCall(LandauKokkosJacobian(ctx->plex, Nq, Nb, ctx->batch_sz, ctx->num_grids, numCells, Eq_m, c… in LandauFormJacobian_Internal()
210 …1], elem_offset[LANDAU_MAX_GRIDS + 1], IPf_sz_glb, IPf_sz_tot, num_grids = ctx->num_grids, Nf[LAND… in LandauFormJacobian_Internal() local
216 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauFormJacobian_Internal()
225 for (PetscInt grid = 0; grid < num_grids; grid++) { in LandauFormJacobian_Internal()
231 IPf_sz_glb = ipf_offset[num_grids]; in LandauFormJacobian_Internal()
243 …for (PetscInt tid = 0; tid < ctx->batch_sz * elem_offset[num_grids]; tid++) { … in LandauFormJacobian_Internal()
244 …const PetscInt b_Nelem = elem_offset[num_grids], b_elem_idx = tid % b_Nelem, b_id = tid / b_Nelem;… in LandauFormJacobian_Internal()
250 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset);… in LandauFormJacobian_Internal()
313 … for (PetscInt tid = 0; tid < ctx->batch_sz * elem_offset[num_grids]; tid++) { // for each element in LandauFormJacobian_Internal()
314 const PetscInt b_Nelem = elem_offset[num_grids]; in LandauFormJacobian_Internal()
324 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset… in LandauFormJacobian_Internal()
349 …for (PetscInt grid_r = 0, f_off = 0, ipidx = 0; grid_r < ctx->num_grids; grid_r++, f_off = ctx->sp… in LandauFormJacobian_Internal()
522 …const PetscInt idx0 = b_id * coo_elem_offsets[elem_offset[num_grids]] + coo_elem_offsets[glb_elem_… in LandauFormJacobian_Internal()
554 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauFormJacobian_Internal()
555 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs… in LandauFormJacobian_Internal()
594 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauDMCreateVMeshes()
733 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauDMCreateVMeshes()
1043 for (PetscInt gridi = 0; gridi < ctx->num_grids; gridi++) { in makeLambdas()
1047 for (PetscInt gridj = gridi; gridj < ctx->num_grids; gridj++) { in makeLambdas()
1207 ctx->num_grids = nt; in ProcessOptions()
1208 for (ii = nt = 0; ii < ctx->num_grids; ii++) nt += num_species_grid[ii]; in ProcessOptions()
1210 ctx->num_grids, LANDAU_MAX_GRIDS); in ProcessOptions()
1215 ctx->num_grids = 2; in ProcessOptions()
1217 ctx->num_grids = ctx->num_species; in ProcessOptions()
1218 for (ii = 0; ii < ctx->num_grids; ii++) num_species_grid[ii] = 1; in ProcessOptions()
1221 …for (ctx->species_offset[0] = ii = 0; ii < ctx->num_grids; ii++) ctx->species_offset[ii + 1] = ctx… in ProcessOptions()
1222 …num_grids] == ctx->num_species, ctx->comm, PETSC_ERR_ARG_WRONG, "ctx->species_offset[ctx->num_grid… in ProcessOptions()
1224 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in ProcessOptions()
1248 …num_grids, ctx->comm, PETSC_ERR_ARG_WRONG, "-dm_landau_domain_radius: given %" PetscInt_FMT " radi… in ProcessOptions()
1253 …num_grids, ctx->comm, PETSC_ERR_ARG_WRONG, "-dm_landau_domain_max_par: given %" PetscInt_FMT " rad… in ProcessOptions()
1255 …num_grids, ctx->comm, PETSC_ERR_ARG_WRONG, "-dm_landau_domain_max_perp: given %" PetscInt_FMT " ra… in ProcessOptions()
1257 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in ProcessOptions()
1272 …x->num_grids, ctx->comm, PETSC_ERR_ARG_WRONG, "-dm_landau_amr_levels_max: given %" PetscInt_FMT " … in ProcessOptions()
1275 …for (ii = 1; ii < ctx->num_grids; ii++) ctx->postAMRRefine[ii] = ctx->postAMRRefine[0]; // all gri… in ProcessOptions()
1289 if (flg && nt < ctx->num_grids) { in ProcessOptions()
1290 …for (PetscInt grid = nt; grid < ctx->num_grids; grid++) ctx->sphere_inner_radius_90degree[grid] = … in ProcessOptions()
1293 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) ctx->sphere_inner_radius_90degree[grid] = 0… in ProcessOptions()
1296 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) ctx->sphere_inner_radius_90degree[grid] = 0… in ProcessOptions()
1298 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) ctx->sphere_inner_radius_90degree[grid] = 0… in ProcessOptions()
1304 if (flg && nt < ctx->num_grids) { in ProcessOptions()
1305 …for (PetscInt grid = nt; grid < ctx->num_grids; grid++) ctx->sphere_inner_radius_45degree[grid] = … in ProcessOptions()
1308 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) ctx->sphere_inner_radius_45degree[grid] = 0… in ProcessOptions()
1310 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) ctx->sphere_inner_radius_45degree[grid] = 0… in ProcessOptions()
1315 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in ProcessOptions()
1353 …for (ii = 1; ii < ctx->num_grids; ii++) PetscCall(PetscPrintf(PETSC_COMM_WORLD, ", %" PetscInt_FMT… in ProcessOptions()
1416 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1425 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1430 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1435 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1438 for (PetscInt gridj = 0; gridj < ctx->num_grids; gridj++) { in CreateStaticData()
1446 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1449 for (PetscInt gridj = 0; gridj < ctx->num_grids; gridj++) { in CreateStaticData()
1465 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1486 PetscCall(PetscMalloc(sizeof(*maps) * ctx->num_grids, &maps)); in CreateStaticData()
1501 for (PetscInt grid = 0, glb_elem_idx = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1513 maps[grid].numgrids = ctx->num_grids; in CreateStaticData()
1625 for (PetscInt grid = 0, glb_elem_idx = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1651 for (PetscInt grid = 0, glb_elem_idx = 0; grid < ctx->num_grids; grid++) { in CreateStaticData()
1652 … const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset); in CreateStaticData()
1704 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) nip_glb += Nq * numCells[grid]; in CreateStaticData()
1711 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) ncells += numCells[grid]; in CreateStaticData()
1722 for (grid = 0, outer_ipidx = 0, outer_ej = 0; grid < ctx->num_grids; grid++) { // OpenMP (once) in CreateStaticData()
1816 …PetscCall(LandauKokkosStaticDataSet(ctx->plex[0], Nq, Nb, ctx->batch_sz, ctx->num_grids, numCells,… in CreateStaticData()
1884 …jacobian_field_major_order) PetscCall(PetscMalloc1(ctx->mat_offset[ctx->num_grids] * ctx->batch_sz… in LandauCreateJacobianMatrix()
1885 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauCreateJacobianMatrix()
1916 PetscInt N = ctx->mat_offset[ctx->num_grids], n0 = ctx->mat_offset[grid] + b_id * N; in LandauCreateJacobianMatrix()
1928 …field_major_order) PetscCall(ISCreateGeneral(comm, ctx->mat_offset[ctx->num_grids] * ctx->batch_sz… in LandauCreateJacobianMatrix()
1930 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in LandauCreateJacobianMatrix()
1936 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs… in LandauCreateJacobianMatrix()
1956 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) PetscCall(MatDestroy(&subM[grid])); in LandauCreateJacobianMatrix()
2068 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in DMPlexLandauCreateVelocitySpace()
2105 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) PetscCall(DMCompositeAddDM(*pack, ctx->plex… in DMPlexLandauCreateVelocitySpace()
2110 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in DMPlexLandauCreateVelocitySpace()
2125 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in DMPlexLandauCreateVelocitySpace()
2130 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs… in DMPlexLandauCreateVelocitySpace()
2138 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) PetscCall(VecDestroy(&Xsub[grid])); in DMPlexLandauCreateVelocitySpace()
2223 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in DMPlexLandauAccess()
2236 … const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset); in DMPlexLandauAccess()
2310 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) PetscCall(DMDestroy(&ctx->plex[grid])); in DMPlexLandauDestroyVelocitySpace()
2411 …nDMs == ctx->num_grids * ctx->batch_sz, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "#DM wrong %" PetscInt_F… in DMPlexLandauPrintNorms()
2414 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in DMPlexLandauPrintNorms()
2466 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { // yes a grid loop in a grid loop to prin… in DMPlexLandauPrintNorms()
2476 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { // yes a grid loop in a grid loop to prin… in DMPlexLandauPrintNorms()
2574 for (PetscInt grid = 0, ix = 0; grid < ctx->num_grids; grid++) { in DMPlexLandauCreateMassMatrix()
2595 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) PetscCall(DMCompositeAddDM(mass_pack, massD… in DMPlexLandauCreateMassMatrix()
2605 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in DMPlexLandauCreateMassMatrix()
2618 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) { in DMPlexLandauCreateMassMatrix()
2624 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs… in DMPlexLandauCreateMassMatrix()
2645 for (PetscInt grid = 0; grid < ctx->num_grids; grid++) PetscCall(MatDestroy(&subM[grid])); in DMPlexLandauCreateMassMatrix()