@@ -231,13 +231,13 @@ PetscErrorCode SGS_DD_TrainingSetup(Ceed ceed, User user, CeedData ceed_data, Pr
231231 array_info [4 ] = smartsim -> collocated_database_num_ranks ;
232232 array_info [5 ] = rank ;
233233
234- PetscSmartRedisCall (put_tensor (smartsim -> client , "sizeInfo" , 8 , array_info , & array_info_dim , 1 , SRTensorTypeInt32 , SRMemLayoutContiguous ));
234+ PetscSmartRedisCall (put_tensor (smartsim -> client , "sizeInfo" , 8 , array_info , & array_info_dim , 1 , SRTensorTypeInt64 , SRMemLayoutContiguous ));
235235 PetscCall (SmartRedisVerifyPutTensor (smartsim -> client , "sizeInfo" , 8 ));
236236
237237 // -- Send array that communicates if tensors are overwritten in database
238238 PetscInt tensor_overwrite [2 ] = {sgs_dd_train -> overwrite_training_data };
239239 size_t dim_2 [1 ] = {2 };
240- PetscSmartRedisCall (put_tensor (smartsim -> client , "tensor-ow" , 9 , tensor_overwrite , dim_2 , 1 , SRTensorTypeInt32 , SRMemLayoutContiguous ));
240+ PetscSmartRedisCall (put_tensor (smartsim -> client , "tensor-ow" , 9 , tensor_overwrite , dim_2 , 1 , SRTensorTypeInt64 , SRMemLayoutContiguous ));
241241 PetscCall (SmartRedisVerifyPutTensor (smartsim -> client , "tensor-ow" , 9 ));
242242 }
243243 }
@@ -320,7 +320,7 @@ PetscErrorCode TSMonitor_SGS_DD_Training(TS ts, PetscInt step_num, PetscReal sol
320320 if (rank % smartsim -> collocated_database_num_ranks == 0 ) {
321321 size_t dim_2 [1 ] = {2 };
322322 PetscInt step_array [2 ] = {step_num , step_num };
323- PetscSmartRedisCall (put_tensor (smartsim -> client , "step" , 4 , step_array , dim_2 , 1 , SRTensorTypeInt32 , SRMemLayoutContiguous ));
323+ PetscSmartRedisCall (put_tensor (smartsim -> client , "step" , 4 , step_array , dim_2 , 1 , SRTensorTypeInt64 , SRMemLayoutContiguous ));
324324 }
325325 }
326326
0 commit comments