fix streamWaitEvent (#700)

Co-authored-by: jingguo-st <fujingguo@sensetime.com>
DeepLink-org · Feb 26, 2024 · 831b1e0 · 831b1e0
1 parent 2f3e94a
commit 831b1e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dipu/torch_dipu/csrc_dipu/vendor/ascend/deviceimpl.cpp b/dipu/torch_dipu/csrc_dipu/vendor/ascend/deviceimpl.cpp
@@ -220,7 +220,7 @@ bool streamNotNull(deviceStream_t stream) {
 }
 
 void streamWaitEvent(deviceStream_t stream, deviceEvent_t event) {
-  DIPU_CALLACLRT(::aclrtSynchronizeEvent(event))
+  DIPU_CALLACLRT(::aclrtStreamWaitEvent(stream, event))
 }
 
 // =====================