Add a warning in case we can't configure the cuDF default pool

NVIDIA · May 22, 2024 · e437abd · e437abd
1 parent 5899c89
commit e437abd
Showing 1 changed file with 5 additions and 1 deletion.
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
@@ -448,7 +448,11 @@ object GpuDeviceManager extends Logging {
       (conf.pinnedPoolSize, -1L)
     }
     // disable the cuDF provided default pinned pool for now
-    PinnedMemoryPool.configureDefaultCudfPinnedPoolSize(0L)
+    if (!PinnedMemoryPool.configureDefaultCudfPinnedPoolSize(0L)) {
+      // This is OK in tests because they don't unload/reload our shared
+      // library, and in prod it would be nice to know about it.
+      logWarning("The default cuDF host pool was already configured")
+    }
     if (!PinnedMemoryPool.isInitialized && pinnedSize > 0) {
       logInfo(s"Initializing pinned memory pool (${pinnedSize / 1024 / 1024.0} MiB)")
       PinnedMemoryPool.initialize(pinnedSize, gpuId, setCuioDefaultResource)