[Paddle Backend] Support GPU train/test/inference/LAMMPS with PaddlePaddle backend for water se_e2_a(revert code format)

.pre-commit-config.yaml

@@ -48,10 +48,10 @@ repos:
     -   id: clang-format
         exclude: ^source/3rdparty|source/lib/src/cuda/cudart/.+\.inc
 # CSS
--   repo: https://github.com/pre-commit/mirrors-csslint
-    rev: v1.0.5
-    hooks:
-    -   id: csslint
+# -   repo: https://github.com/pre-commit/mirrors-csslint
+#     rev: v1.0.5
+#     hooks:
+#     -   id: csslint
 # Shell
 - repo: https://github.com/scop/pre-commit-shfmt
   rev: v3.6.0-2

deepmd/common.py

@@ -28,8 +28,10 @@
 
 from deepmd.env import (
     GLOBAL_NP_FLOAT_PRECISION,
+    GLOBAL_PD_FLOAT_PRECISION,
     GLOBAL_TF_FLOAT_PRECISION,
     op_module,
+    paddle,
     tf,
 )
 from deepmd.utils.path import (
@@ -50,11 +52,11 @@
 
 # define constants
 PRECISION_DICT = {
-    "default": GLOBAL_TF_FLOAT_PRECISION,
-    "float16": tf.float16,
-    "float32": tf.float32,
-    "float64": tf.float64,
-    "bfloat16": tf.bfloat16,
+    "default": GLOBAL_PD_FLOAT_PRECISION,
+    "float16": paddle.float16,
+    "float32": paddle.float32,
+    "float64": paddle.float64,
+    "bfloat16": paddle.bfloat16,
 }
 
 
@@ -119,11 +121,11 @@ def gelu_wrapper(x):
 data_requirement = {}
 
 ACTIVATION_FN_DICT = {
-    "relu": tf.nn.relu,
-    "relu6": tf.nn.relu6,
-    "softplus": tf.nn.softplus,
-    "sigmoid": tf.sigmoid,
-    "tanh": tf.nn.tanh,
+    "relu": paddle.nn.functional.relu,
+    "relu6": paddle.nn.functional.relu6,
+    "softplus": paddle.nn.functional.softplus,
+    "sigmoid": paddle.nn.functional.sigmoid,
+    "tanh": paddle.nn.functional.tanh,
     "gelu": gelu,
     "gelu_tf": gelu_tf,
     "None": None,