From 42dd3261bbdd4606b13adebe352c1cee79a5653b Mon Sep 17 00:00:00 2001 From: Thomas Graves Date: Thu, 17 Dec 2020 13:41:17 -0600 Subject: [PATCH] Log the number of concurrent gpu tasks allowed on Executor startup (#1422) Signed-off-by: Thomas Graves --- .../src/main/scala/com/nvidia/spark/rapids/Plugin.scala | 4 +++- 1 file changed, 3 insertions(+), 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 index 06a6aabcd9a..fd81bc3df92 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 @@ -145,7 +145,9 @@ class RapidsExecutorPlugin extends ExecutorPlugin with Logging { GpuDeviceManager.initializeGpuAndMemory(pluginContext.resources().asScala.toMap) } - GpuSemaphore.initialize(conf.concurrentGpuTasks) + val concurrentGpuTasks = conf.concurrentGpuTasks + logInfo(s"The number of concurrent GPU tasks allowed is $concurrentGpuTasks") + GpuSemaphore.initialize(concurrentGpuTasks) } catch { case e: Throwable => // Exceptions in executor plugin can cause a single thread to die but the executor process