fix logger

Author: eee4017

python/paddle/device/cuda/cuda_graphed_layer.py

@@ -14,7 +14,6 @@
 
 import logging
 import os
-import sys
 from collections import deque
 from enum import Enum
 
@@ -23,10 +22,6 @@
 
 from .graphs import CUDAGraph
 
-logger = log_helper.get_logger(
-    __name__, logging.INFO, fmt='[%(levelname)s] %(message)s'
-)
-
 # CUDAGraphedLayer Debug tools
 enable_debug_print = bool(
     int(os.getenv('PADDLE_DEBUG_ENABLE_CUDAGRAPH_LAYER_LOGGING', '0'))
@@ -35,11 +30,15 @@
     int(os.getenv('PADDLE_DEBUG_CUDAGRAPHEDLAYER_FALLBACK_TO_DEFAULT', '0'))
 )
 
+logger = log_helper.get_logger(
+    __name__, logging.INFO, fmt='[%(levelname)s] %(message)s'
+)
+
 
 def debug_print(x):
     if not enable_debug_print:
         return
-    print(x, file=sys.stderr)
+    logger.info(x)
 
 
 def print_tensor(