Added a synchronize call after memset int the devMemset wrappers.

GlobalArrays · Jul 19, 2023 · 1fb7157 · 1fb7157
1 parent c68aa88
commit 1fb7157
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/comex/src-mpi-pr/dev_utils.cu b/comex/src-mpi-pr/dev_utils.cu
@@ -254,6 +254,7 @@ void deviceMemset(void *ptr, int val, size_t bytes)
     printf("p[%d] cudaMemset ptr: %p bytes: %d msg: %s\n",rank,ptr,bytes,msg);
     MPI_Wrapper_abort(err);
   }
+  cudaDeviceSynchronize();
 }
 
 __global__ void iaxpy_kernel(int *dst, const int *src, int scale, int n)

diff --git a/comex/src-mpi-pr/dev_utils_hip.cpp b/comex/src-mpi-pr/dev_utils_hip.cpp
@@ -244,6 +244,7 @@ void deviceMemset(void *ptr, int val, size_t bytes)
     printf("p[%d] hipMemset ptr: %p bytes: %zu msg: %s\n",rank,ptr,bytes,msg);
     MPI_Wrapper_abort(err);
   }
+  hipDeviceSynchronize();
 }
 
 __global__ void iaxpy_kernel(int *dst, const int *src, int scale, int n)