Throw again after logging that RMM could not intialize (#5243)

* Throw again after logging that RMM could not intialize Signed-off-by: Alessandro Bellina <[email protected]> * Let exception as is from RMM, and change slightly the wording from the ExecutorPlugin
NVIDIA · Apr 14, 2022 · b6eaa50 · b6eaa50
1 parent 6c27068
commit b6eaa50
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuDeviceManager.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuDeviceManager.scala
@@ -296,13 +296,9 @@ object GpuDeviceManager extends Logging {
         logInfo("Using legacy default stream")
       }
 
-      try {
-        Cuda.setDevice(gpuId)
-        Rmm.initialize(init, logConf, poolAllocation)
-        RapidsBufferCatalog.init(conf)
-      } catch {
-        case e: Exception => logError("Could not initialize RMM", e)
-      }
+      Cuda.setDevice(gpuId)
+      Rmm.initialize(init, logConf, poolAllocation)
+      RapidsBufferCatalog.init(conf)
     }
   }
 

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
@@ -245,7 +245,7 @@ class RapidsExecutorPlugin extends ExecutorPlugin with Logging {
         // Exceptions in executor plugin can cause a single thread to die but the executor process
         // sticks around without any useful info until it hearbeat times out. Print what happened
         // and exit immediately.
-        logError("Exception in the executor plugin", e)
+        logError("Exception in the executor plugin, shutting down!", e)
         System.exit(1)
     }
   }