wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-29 03:21:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix shutdown hook in worker mode (#8098)

Browse Source
This commit is contained in:
Kai Yang
2020-05-26 15:23:44 +08:00
committed by GitHub
parent f39760a4d3
commit 26cffb9c7c
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
java/runtime/src/main/java/io/ray/runtime/RayNativeRuntime.java
+7 -5
View File
@@ -102,12 +102,14 @@ public final class RayNativeRuntime extends AbstractRayRuntime {
@Override
public void shutdown() {
nativeShutdown();
if (null != manager) {
manager.cleanup();
manager = null;
if (rayConfig.workerMode == WorkerType.DRIVER) {
nativeShutdown();
if (null != manager) {
manager.cleanup();
manager = null;
}
RayConfig.reset();
}
RayConfig.reset();
LOGGER.info("RayNativeRuntime shutdown");
}