Add device synchronization before destroying proxy thread. (#1631)
This commit ensures that GPU finishes all kernel before destroying communicator thread.
Αυτή η υποβολή περιλαμβάνεται σε:
υποβλήθηκε από
GitHub
γονέας
e40ff4f84a
υποβολή
52654e2301
@@ -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");
|
||||
}
|
||||
|
||||
Αναφορά σε νέο ζήτημα
Block a user