Let exception as is from RMM, and change slightly the wording from th…

…e ExecutorPlugin
NVIDIA · Apr 13, 2022 · f6545d3 · f6545d3
1 parent 4f4172f
commit f6545d3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 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,15 +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: CudfException =>
-          logError("Could not initialize RMM, exiting!", e)
-          throw 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
@@ -238,7 +238,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)
     }
   }