Skip to content

Commit

Permalink
Sync stream for scatter_op
Browse files Browse the repository at this point in the history
  • Loading branch information
chang-l committed Nov 19, 2024
1 parent 32f7d96 commit 097434c
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions cpp/src/wholememory_ops/scatter_op_impl_mapped.cu
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
#include <wholememory/env_func_ptrs.h>
#include <wholememory/wholememory.h>

#include "cuda_macros.hpp"
#include "wholememory_ops/functions/gather_scatter_func.h"

namespace wholememory_ops {
Expand All @@ -41,6 +42,7 @@ wholememory_error_code_t wholememory_scatter_mapped(
wholememory_desc,
stream,
scatter_sms);
WM_CUDA_CHECK(cudaStreamSynchronize(stream));
}

} // namespace wholememory_ops

0 comments on commit 097434c

Please sign in to comment.