fluids: Add -sgs_train_enable flag

CEED · May 31, 2023 · aa84b68 · aa84b68
1 parent 635288d
commit aa84b68
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/examples/fluids/navierstokes.h b/examples/fluids/navierstokes.h
@@ -135,6 +135,7 @@ struct AppCtx_private {
   } wall_forces;
   // Subgrid Stress Model
   SGSModelType sgs_model_type;
+  PetscBool    sgs_train_enable;
   // Differential Filtering
   PetscBool diff_filter_monitor;
 };

diff --git a/examples/fluids/src/cloptions.c b/examples/fluids/src/cloptions.c
@@ -186,6 +186,9 @@ PetscErrorCode ProcessCommandLineOptions(MPI_Comm comm, AppCtx app_ctx, SimpleBC
   PetscCall(PetscOptionsBool("-diff_filter_monitor", "Enable differential filtering TSMonitor", NULL, app_ctx->diff_filter_monitor,
                              &app_ctx->diff_filter_monitor, NULL));
 
+  PetscCall(
+      PetscOptionsBool("-sgs_train_enable", "Enable Data-Driven SGS training", NULL, app_ctx->sgs_train_enable, &app_ctx->sgs_train_enable, NULL));
+
   PetscOptionsEnd();
 
   PetscFunctionReturn(0);

diff --git a/examples/fluids/src/setuplibceed.c b/examples/fluids/src/setuplibceed.c
@@ -448,6 +448,7 @@ PetscErrorCode SetupLibceed(Ceed ceed, CeedData ceed_data, DM dm, User user, App
 
   if (app_ctx->turb_spanstats_enable) PetscCall(TurbulenceStatisticsSetup(ceed, user, ceed_data, problem));
   if (app_ctx->diff_filter_monitor && !user->diff_filter) PetscCall(DifferentialFilterSetup(ceed, user, ceed_data, problem));
+  if (app_ctx->sgs_train_enable) PetscCall(SGS_DD_TrainingSetup(ceed, user, ceed_data, problem));
 
   CeedElemRestrictionDestroy(&elem_restr_jd_i);
   CeedOperatorDestroy(&op_ijacobian_vol);

diff --git a/examples/fluids/src/setupts.c b/examples/fluids/src/setupts.c
@@ -494,6 +494,10 @@ PetscErrorCode TSSolve_NS(DM dm, User user, AppCtx app_ctx, Physics phys, Vec *Q
   }
   if (app_ctx->diff_filter_monitor) PetscCall(TSMonitorSet(*ts, TSMonitor_DifferentialFilter, user, NULL));
 
+  if (app_ctx->sgs_train_enable) {
+    PetscCall(TSMonitorSet(*ts, TSMonitor_SGS_DD_Training, user, NULL));
+    PetscCall(TSSetPostStep(*ts, TSPostStep_SGS_DD_Training));
+  }
   // Solve
   PetscReal start_time;
   PetscInt  start_step;