Call GpuDeviceManager.shutdown when the executor plugin is shutting d…

…own (NVIDIA#1713) * Call GpuDeviceManager.shutdown when the executor plugin is shutting down Signed-off-by: Alessandro Bellina <abellina@nvidia.com> * Update copyright
nartal1 · Feb 12, 2021 · ce6f65d · ce6f65d
1 parent 52a61ec
commit ce6f65d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/Plugin.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/Plugin.scala
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2019-2020, NVIDIA CORPORATION.
+ * Copyright (c) 2019-2021, NVIDIA CORPORATION.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -210,6 +210,7 @@ class RapidsExecutorPlugin extends ExecutorPlugin with Logging {
   override def shutdown(): Unit = {
     GpuSemaphore.shutdown()
     PythonWorkerSemaphore.shutdown()
+    GpuDeviceManager.shutdown()
   }
 }