diff --git a/projects/rccl/src/proxy.cc b/projects/rccl/src/proxy.cc index 90f6a7568f..a0b86889ca 100644 --- a/projects/rccl/src/proxy.cc +++ b/projects/rccl/src/proxy.cc @@ -1702,6 +1702,7 @@ void* ncclProxyService(void* _args) { } // Wait for all operations to complete and stop progress thread before freeing any resource + hipDeviceSynchronize(); if (ncclProxyProgressDestroy(proxyState) != ncclSuccess) { WARN("[Proxy Service] proxyDestroy failed"); }