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 index 26fda9aaea6..f5e5a3a940a 100644 --- 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 index 36a7ef86540..f91b310eda4 100644 --- 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) } }