Replace paddle.fluid imports with paddle.base

Signed-off-by: Tim Moon <tmoon@nvidia.com>

qa/L0_paddle_lint/test.sh

@@ -22,5 +22,5 @@ then
   cp $TE_PATH/qa/L0_paddle_lint/pylintrc $TE_PATH
   cd $TE_PATH
   echo "Checking Python files"
-  pylint --recursive=y transformer_engine/common transformer_engine/paddle
+  python -m pylint --recursive=y transformer_engine/common transformer_engine/paddle
 fi

transformer_engine/paddle/layer/base.py

@@ -12,8 +12,12 @@
 import numpy as np
 
 import paddle
-from paddle.fluid import core
-from paddle.fluid.framework import _dygraph_tracer
+try:
+    from paddle.base import core
+    from paddle.base.framework import _dygraph_tracer
+except ImportError:
+    from paddle.fluid import core
+    from paddle.fluid.framework import _dygraph_tracer
 
 from ..constants import FP8BwdTensors, dist_group_type
 from ..cpp_extensions import cast_transpose, cast_transpose_bgrad, cast_to_fp8, transpose

transformer_engine/paddle/profile.py

@@ -5,7 +5,10 @@
 
 from contextlib import contextmanager
 
-from paddle.fluid import core
+try:
+    from paddle.base import core
+except ImportError:
+    from paddle.fluid import core
 
 
 @contextmanager