Add device synchronization before destroying proxy thread. (#1631)

This commit ensures that GPU finishes all kernel before destroying
communicator thread.
Αυτή η υποβολή περιλαμβάνεται σε:
Arm Patinyasakdikul
2025-04-10 10:44:16 -05:00
υποβλήθηκε από GitHub
γονέας e40ff4f84a
υποβολή 52654e2301
+1
Προβολή Αρχείου
@@ -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");
}