fluids: Add SmartSimDataDestroy

CEED · Nov 27, 2023 · b306c3f · b306c3f
1 parent e17e35b
commit b306c3f
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/examples/fluids/navierstokes.c b/examples/fluids/navierstokes.c
@@ -259,6 +259,7 @@ int main(int argc, char **argv) {
   PetscCall(SgsDDDataDestroy(user->sgs_dd_data));
   PetscCall(DifferentialFilterDataDestroy(user->diff_filter));
   PetscCall(SGS_DD_TrainingDataDestroy(user->sgs_dd_train));
+  PetscCall(SmartSimDataDestroy(user->smartsim));
 
   // -- Vectors
   PetscCallCeed(ceed, CeedVectorDestroy(&ceed_data->x_coord));

diff --git a/examples/fluids/navierstokes.h b/examples/fluids/navierstokes.h
@@ -492,6 +492,7 @@ PetscErrorCode DifferentialFilterMmsICSetup(ProblemData *problem);
 // SGS Data-Driven Training via SmartSim
 // -----------------------------------------------------------------------------
 PetscErrorCode SmartSimSetup(User user);
+PetscErrorCode SmartSimDataDestroy(SmartSimData smartsim);
 PetscErrorCode SGS_DD_TrainingSetup(Ceed ceed, User user, CeedData ceed_data, ProblemData *problem);
 PetscErrorCode TSMonitor_SGS_DD_Training(TS ts, PetscInt step_num, PetscReal solution_time, Vec Q, void *ctx);
 PetscErrorCode TSPostStep_SGS_DD_Training(TS ts);

diff --git a/examples/fluids/src/smartsim/smartsim.c b/examples/fluids/src/smartsim/smartsim.c
@@ -63,3 +63,12 @@ PetscErrorCode SmartSimSetup(User user) {
   PetscCall(SmartSimTrainingSetup(user));
   PetscFunctionReturn(PETSC_SUCCESS);
 }
+
+PetscErrorCode SmartSimDataDestroy(SmartSimData smartsim) {
+  PetscFunctionBeginUser;
+  if (!smartsim) PetscFunctionReturn(PETSC_SUCCESS);
+
+  SmartRedisCall(DeleteCClient(&smartsim->client));
+  PetscCall(PetscFree(smartsim));
+  PetscFunctionReturn(PETSC_SUCCESS);
+}
diff --git a/examples/fluids/src/smartsim_weak.c b/examples/fluids/src/smartsim_weak.c
@@ -35,3 +35,12 @@ PetscErrorCode TSPostStep_SGS_DD_Training(TS ts) {
   PetscFunctionBeginUser;
   SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_SUP, "Must build with SMARTREDIS_DIR set to run %s", __func__);
 };
+
+PetscErrorCode SmartSimDataDestroy(SmartSimData smartsim) __attribute__((weak));
+PetscErrorCode SmartSimDataDestroy(SmartSimData smartsim) {
+  PetscFunctionBeginUser;
+  if (!smartsim) PetscFunctionReturn(PETSC_SUCCESS);
+  PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Warning: SmartSimData struct should not be initialized if SMARTREDIS_DIR isn't set on build..."));
+
+  PetscFunctionReturn(PETSC_SUCCESS);
+}