Log the number of concurrent gpu tasks allowed on Executor startup (#…

…1422) Signed-off-by: Thomas Graves <tgraves@nvidia.com>
NVIDIA · Dec 17, 2020 · 42dd326 · 42dd326
1 parent 801ba80
commit 42dd326
Showing 1 changed file with 3 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
@@ -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