Skip to content

Commit 1c08cab

Browse files
author
Riccardo Balin
committed
fixed few bugs
1 parent 4eed59e commit 1c08cab

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

backends/sycl-ref/ceed-sycl-ref-operator.sycl.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -552,8 +552,7 @@ static inline int CeedOperatorLinearAssembleQFunctionCore_Sycl(CeedOperator op,
552552
CeedInt strides[3] = {1, num_elem * Q, Q}; /* *NOPAD* */
553553

554554
// Create output restriction
555-
CeedCallBackend(CeedElemRestrictionCreateStrided(ceed_parent, num_elem, Q, num_active_in * num_active_out,
556-
num_active_in * num_active_out * num_elem * Q, strides, rstr));
555+
CeedCallBackend(CeedElemRestrictionCreateStrided(ceed_parent, num_elem, Q, num_active_in * num_active_out, l_size, strides, rstr));
557556
// Create assembled vector
558557
CeedCallBackend(CeedVectorCreate(ceed_parent, l_size, assembled));
559558
}

examples/fluids/src/smartsim/sgs_dd_training.c

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)