From d180990fc626a8a69970baccd52092475f3230c9 Mon Sep 17 00:00:00 2001
From: Tim Besard <tim.besard@gmail.com>
Date: Mon, 25 May 2020 17:24:11 +0200
Subject: [PATCH] Use regular debug macro.

---
 Project.toml  | 1 +
 src/memory.jl | 8 ++++++--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/Project.toml b/Project.toml
index 7cd47e25ae..7edfcae1de 100644
--- a/Project.toml
+++ b/Project.toml
@@ -14,6 +14,7 @@ GPUCompiler = "61eb1bfa-7361-4325-ad38-22787b887f55"
 LLVM = "929cbde3-209d-540e-8aea-75f648917ca0"
 Libdl = "8f399da3-3557-5675-b5ff-fb832c97cbdb"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
diff --git a/src/memory.jl b/src/memory.jl
index 5d2a3d4447..383c1ef77e 100644
--- a/src/memory.jl
+++ b/src/memory.jl
@@ -1,6 +1,7 @@
 # GPU memory management and pooling
 
 using Printf
+using Logging
 using TimerOutputs
 
 using Base: @lock
@@ -515,10 +516,13 @@ function __init_memory__()
 
     # the user hand-picked an allocator, so be a little verbose
     atexit(()->begin
-      Core.println("""
+      old_logger = global_logger()
+      global_logger(Logging.ConsoleLogger(Core.stderr, old_logger.min_level))
+      @debug """
         CUDA.jl $(nameof(pool[])) statistics:
          - $(alloc_stats.pool_nalloc) pool allocations: $(Base.format_bytes(alloc_stats.pool_alloc)) in $(Base.round(alloc_stats.pool_time; digits=2))s
-         - $(alloc_stats.actual_nalloc) CUDA allocations: $(Base.format_bytes(alloc_stats.actual_alloc)) in $(Base.round(alloc_stats.actual_time; digits=2))s""")
+         - $(alloc_stats.actual_nalloc) CUDA allocations: $(Base.format_bytes(alloc_stats.actual_alloc)) in $(Base.round(alloc_stats.actual_time; digits=2))s"""
+      global_logger(old_logger)
     end)
   end