Lines Matching refs:smartsim
11 static PetscErrorCode SmartSimDataDestroy(SmartSimData *smartsim) { in SmartSimDataDestroy() argument
12 SmartSimData smartsim_ = *smartsim; in SmartSimDataDestroy()
19 *smartsim = NULL; in SmartSimDataDestroy()
24 SmartSimData smartsim; in SmartSimTrainingSetup() local
30 PetscCall(HoneeGetSmartSimData(honee, &smartsim)); in SmartSimTrainingSetup()
33 if (rank % smartsim->collocated_database_num_ranks == 0) { in SmartSimTrainingSetup()
36 …PetscCallSmartRedis(put_tensor(smartsim->client, "check-run", 9, checkrun, dim_2, 1, SRTensorTypeD… in SmartSimTrainingSetup()
37 PetscCall(SmartRedisVerifyPutTensor(smartsim->client, "check-run", 9)); in SmartSimTrainingSetup()
46 SmartSimData smartsim; in SmartSimSetup() local
49 PetscCall(PetscNew(&smartsim)); in SmartSimSetup()
51 smartsim->collocated_database_num_ranks = 1; in SmartSimSetup()
54 … smartsim->collocated_database_num_ranks, &smartsim->collocated_database_num_ranks, NULL)); in SmartSimSetup()
59 PetscCall(PetscSNPrintf(smartsim->rank_id_name, sizeof(smartsim->rank_id_name), "y.%d", rank)); in SmartSimSetup()
62 …rtRedisCClient(num_orchestrator_nodes != 1, smartsim->rank_id_name, strlen(smartsim->rank_id_name)… in SmartSimSetup()
65 …PetscCall(HoneeSetContainer(honee, SMARTSIM_KEY, smartsim, (PetscCtxDestroyFn *)SmartSimDataDestro… in SmartSimSetup()
79 PetscErrorCode HoneeGetSmartSimData(Honee honee, SmartSimData *smartsim) { in HoneeGetSmartSimData() argument
85 PetscCall(HoneeGetContainer(honee, SMARTSIM_KEY, smartsim)); in HoneeGetSmartSimData()