diff --git a/java/runtime/src/main/java/io/ray/runtime/RayNativeRuntime.java b/java/runtime/src/main/java/io/ray/runtime/RayNativeRuntime.java index 5c8f2dc3c..e0df1b521 100644 --- a/java/runtime/src/main/java/io/ray/runtime/RayNativeRuntime.java +++ b/java/runtime/src/main/java/io/ray/runtime/RayNativeRuntime.java @@ -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"); }