diff --git a/dipu/torch_dipu/csrc_dipu/vendor/ascend/deviceimpl.cpp b/dipu/torch_dipu/csrc_dipu/vendor/ascend/deviceimpl.cpp index c9e3e78a8..d0ca8f1f9 100644 --- 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)) } // =====================