25 static PetscErrorCode SGS_DD_TrainingDataDestroy(SGS_DD_TrainingData *sgs_dd_train) {  in SGS_DD_TrainingDataDestroy()  argument
26   SGS_DD_TrainingData sgs_dd_train_ = *sgs_dd_train;  in SGS_DD_TrainingDataDestroy()
35   *sgs_dd_train = NULL;  in SGS_DD_TrainingDataDestroy()
91   SGS_DD_TrainingData sgs_dd_train;  in SetupTrainingDataCalculation()  local
100   PetscCall(HoneeGetContainer(honee, SGS_DD_TRAIN_KEY, &sgs_dd_train));  in SetupTrainingDataCalculation()
103 …PetscCall(DMPlexCeedElemRestrictionCreate(ceed, sgs_dd_train->dm_dd_training, DMLABEL_DEFAULT, DML…  in SetupTrainingDataCalculation()
105 …PetscCall(GetInverseMultiplicity(ceed, sgs_dd_train->dm_dd_training, DMLABEL_DEFAULT, DMLABEL_DEFA…  in SetupTrainingDataCalculation()
113 …PetscCallCeed(ceed, CeedOperatorGetFieldByName(sgs_dd_train->diff_filter->op_rhs_ctx->op, "v0", &o…  in SetupTrainingDataCalculation()
117                                               &sgs_dd_train->filtered_grad_velo_proj));  in SetupTrainingDataCalculation()
120 …PetscCallCeed(ceed, CeedOperatorGetFieldByName(sgs_dd_train->filtered_grad_velo_proj->l2_rhs_ctx->…  in SetupTrainingDataCalculation()
129 …PetscCallCeed(ceed, CeedOperatorGetFieldByName(sgs_dd_train->diff_filter->op_rhs_ctx->op, "v1", &o…  in SetupTrainingDataCalculation()
145 …PetscCallCeed(ceed, CeedQFunctionAddOutput(qf_sgs_dd_train, "training data", sgs_dd_train->num_com…  in SetupTrainingDataCalculation()
157 …PetscCall(OperatorApplyContextCreate(sgs_dd_train->filtered_grad_velo_proj->dm, sgs_dd_train->dm_d…  in SetupTrainingDataCalculation()
158                                        NULL, NULL, &sgs_dd_train->op_training_data_calc_ctx));  in SetupTrainingDataCalculation()
174   SGS_DD_TrainingData      sgs_dd_train;  in SGS_DD_TrainingSetup()  local
180   PetscCall(PetscNew(&sgs_dd_train));  in SGS_DD_TrainingSetup()
181   *sgs_dd_train = (struct SGS_DD_TrainingData_){  in SGS_DD_TrainingSetup()
184 ….num_filter_widths       = sizeof(sgs_dd_train->filter_widths) / sizeof(sgs_dd_train->filter_width…  in SGS_DD_TrainingSetup()
186 …PetscCall(HoneeSetContainer(honee, SGS_DD_TRAIN_KEY, sgs_dd_train, (PetscCtxDestroyFn *)SGS_DD_Tra…  in SGS_DD_TrainingSetup()
187   PetscCall(DifferentialFilterSetup(honee, &sgs_dd_train->diff_filter));  in SGS_DD_TrainingSetup()
191 …                     sgs_dd_train->write_data_interval, &sgs_dd_train->write_data_interval, NULL));  in SGS_DD_TrainingSetup()
192 …overwrite_data", "Overwrite old training data in the database", NULL, sgs_dd_train->overwrite_trai…  in SGS_DD_TrainingSetup()
193                              &sgs_dd_train->overwrite_training_data, NULL));  in SGS_DD_TrainingSetup()
195 …                             sgs_dd_train->filter_widths, &sgs_dd_train->num_filter_widths, NULL));  in SGS_DD_TrainingSetup()
199 …PetscCall(SGS_DD_TrainingCreateDM(honee->dm, &sgs_dd_train->dm_dd_training, honee->app_ctx->degree…  in SGS_DD_TrainingSetup()
200                                     &sgs_dd_train->num_comp_dd_inputs));  in SGS_DD_TrainingSetup()
224       PetscCall(DMGetGlobalSection(sgs_dd_train->dm_dd_training, &global_section));  in SGS_DD_TrainingSetup()
225       PetscCall(DMGetGlobalVectorInfo(sgs_dd_train->dm_dd_training, &num_dofs, NULL, NULL));  in SGS_DD_TrainingSetup()
230       sgs_dd_train->training_data_array_dims[0] = global_min_max[0] / num_comps;  in SGS_DD_TrainingSetup()
231       sgs_dd_train->training_data_array_dims[1] = num_comps;  in SGS_DD_TrainingSetup()
240         array_info[0] = sgs_dd_train->training_data_array_dims[0];  in SGS_DD_TrainingSetup()
241         array_info[1] = sgs_dd_train->training_data_array_dims[1];  in SGS_DD_TrainingSetup()
256         PetscInt64 tensor_overwrite[2] = {sgs_dd_train->overwrite_training_data};  in SGS_DD_TrainingSetup()
268         PetscInt64 num_filter_widths = sgs_dd_train->num_filter_widths;  in SGS_DD_TrainingSetup()
294   SGS_DD_TrainingData sgs_dd_train;  in TSMonitor_SGS_DD_Training()  local
301   PetscCall(HoneeGetContainer(honee, SGS_DD_TRAIN_KEY, &sgs_dd_train));  in TSMonitor_SGS_DD_Training()
304   if (step_num % sgs_dd_train->write_data_interval != 0) PetscFunctionReturn(PETSC_SUCCESS);  in TSMonitor_SGS_DD_Training()
305   PetscCall(DMGetGlobalVector(sgs_dd_train->dm_dd_training, &TrainingData));  in TSMonitor_SGS_DD_Training()
307   for (PetscInt filter_index = 0; filter_index < sgs_dd_train->num_filter_widths; filter_index++) {  in TSMonitor_SGS_DD_Training()
319 …   for (int j = 0; j < 3; j++) filter_width_scaling[j] = sgs_dd_train->filter_widths[filter_index];  in TSMonitor_SGS_DD_Training()
320         PetscCall(KSPGetOperators(sgs_dd_train->diff_filter->ksp, &A_mat, NULL));  in TSMonitor_SGS_DD_Training()
322 …PetscCall(CeedOperatorSetContextDouble(op_mat, sgs_dd_train->diff_filter->filter_width_scaling_lab…  in TSMonitor_SGS_DD_Training()
325       PetscCall(DMGetGlobalVector(sgs_dd_train->diff_filter->dm_filter, &FilteredFields));  in TSMonitor_SGS_DD_Training()
326       PetscCall(DMGetLocalVector(sgs_dd_train->diff_filter->dm_filter, &FilteredFields_loc));  in TSMonitor_SGS_DD_Training()
328 …PetscCall(DifferentialFilterApply(honee, sgs_dd_train->diff_filter, solution_time, Q, FilteredFiel…  in TSMonitor_SGS_DD_Training()
329 …PetscCall(DMGlobalToLocal(sgs_dd_train->diff_filter->dm_filter, FilteredFields, INSERT_VALUES, Fil…  in TSMonitor_SGS_DD_Training()
331 …PetscCall(DMGetGlobalVector(sgs_dd_train->filtered_grad_velo_proj->dm, &FilteredVelocityGradient));  in TSMonitor_SGS_DD_Training()
332 …PetscCall(VelocityGradientProjectionApply(sgs_dd_train->filtered_grad_velo_proj, FilteredFields_lo…  in TSMonitor_SGS_DD_Training()
337 …PetscCallCeed(ceed, CeedOperatorGetFieldByName(sgs_dd_train->op_training_data_calc_ctx->op, "q", &…  in TSMonitor_SGS_DD_Training()
342 …plyCeedOperatorGlobalToGlobal(FilteredVelocityGradient, TrainingData, sgs_dd_train->op_training_da…  in TSMonitor_SGS_DD_Training()
345 …PetscCall(DMRestoreGlobalVector(sgs_dd_train->filtered_grad_velo_proj->dm, &FilteredVelocityGradie…  in TSMonitor_SGS_DD_Training()
346       PetscCall(DMRestoreGlobalVector(sgs_dd_train->diff_filter->dm_filter, &FilteredFields));  in TSMonitor_SGS_DD_Training()
347       PetscCall(DMRestoreLocalVector(sgs_dd_train->diff_filter->dm_filter, &FilteredFields_loc));  in TSMonitor_SGS_DD_Training()
356       if (sgs_dd_train->overwrite_training_data) {  in TSMonitor_SGS_DD_Training()
367 …or(smartsim->client, array_key, array_key_len, (void *)training_data, sgs_dd_train->training_data_…  in TSMonitor_SGS_DD_Training()
386   PetscCall(DMRestoreGlobalVector(sgs_dd_train->dm_dd_training, &TrainingData));  in TSMonitor_SGS_DD_Training()