From d725e45f1362205ff2b22edc8ebfff9f1390fc68 Mon Sep 17 00:00:00 2001 From: Binyang Li Date: Tue, 28 Mar 2023 14:53:03 +0000 Subject: [PATCH] fix --- src/proxy.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/proxy.cc b/src/proxy.cc index cda3476d..dce62479 100644 --- a/src/proxy.cc +++ b/src/proxy.cc @@ -178,7 +178,7 @@ void* mscclppProxyService(void* _args) PROXYCUDACHECK(cudaStreamDestroy(stream)); if (isP2pProxy) { - PROXYCUDACHECK(cudaStreamDestroy(p2pStream)); + PROXYCUDACHECK(cudaStreamSynchronize(p2pStream)); } *run = MSCCLPP_PROXY_RUN_STATE_IDLE; return NULL;