Lines Matching refs:smartsim
215 SmartSimData smartsim; in SGS_DD_TrainingSetup() local
216 PetscCall(HoneeGetSmartSimData(honee, &smartsim)); in SGS_DD_TrainingSetup()
234 if (rank % smartsim->collocated_database_num_ranks == 0) { in SGS_DD_TrainingSetup()
244 array_info[4] = smartsim->collocated_database_num_ranks; in SGS_DD_TrainingSetup()
248 …PetscCallSmartRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), array_info, &ar… in SGS_DD_TrainingSetup()
250 PetscCall(SmartRedisVerifyPutTensor(smartsim->client, tensor_name, strlen(tensor_name))); in SGS_DD_TrainingSetup()
260 …PetscCallSmartRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), tensor_overwrit… in SGS_DD_TrainingSetup()
262 PetscCall(SmartRedisVerifyPutTensor(smartsim->client, tensor_name, strlen(tensor_name))); in SGS_DD_TrainingSetup()
272 …PetscCallSmartRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), &num_filter_wid… in SGS_DD_TrainingSetup()
274 PetscCall(SmartRedisVerifyPutTensor(smartsim->client, tensor_name, strlen(tensor_name))); in SGS_DD_TrainingSetup()
295 SmartSimData smartsim; in TSMonitor_SGS_DD_Training() local
300 PetscCall(HoneeGetSmartSimData(honee, &smartsim)); in TSMonitor_SGS_DD_Training()
357 …PetscCall(PetscSNPrintf(array_key, sizeof array_key, "%s.%" PetscInt_FMT, smartsim->rank_id_name, … in TSMonitor_SGS_DD_Training()
359 …intf(array_key, sizeof array_key, "%s.%" PetscInt_FMT "%" PetscInt_FMT, smartsim->rank_id_name, st… in TSMonitor_SGS_DD_Training()
367 …PetscCallSmartRedis(put_tensor(smartsim->client, array_key, array_key_len, (void *)training_data, … in TSMonitor_SGS_DD_Training()
375 if (rank % smartsim->collocated_database_num_ranks == 0) { in TSMonitor_SGS_DD_Training()
381 …PetscCallSmartRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), step_array, dim… in TSMonitor_SGS_DD_Training()
396 SmartSimData smartsim; in TSPostStep_SGS_DD_Training() local
401 PetscCall(HoneeGetSmartSimData(honee, &smartsim)); in TSPostStep_SGS_DD_Training()
404 …PetscCallSmartRedis(unpack_tensor(smartsim->client, check_run_key, check_run_key_size, check_run, … in TSPostStep_SGS_DD_Training()