prune redundant files

deepmd/pd/__init__.py

@@ -1,15 +1,11 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 
 # import customized OPs globally
-from deepmd.pd.cxx_op import (
-    ENABLE_CUSTOMIZED_OP,
-)
+
 from deepmd.utils.entry_point import (
     load_entry_point,
 )
 
 load_entry_point("deepmd.pd")
 
-__all__ = [
-    "ENABLE_CUSTOMIZED_OP",
-]
+__all__ = []

deepmd/pd/entrypoints/main.py

@@ -23,18 +23,12 @@
 from deepmd.common import (
     expand_sys_str,
 )
-from deepmd.env import (
-    GLOBAL_CONFIG,
-)
 from deepmd.loggers.loggers import (
     set_log_handles,
 )
 from deepmd.main import (
     parse_args,
 )
-from deepmd.pd.cxx_op import (
-    ENABLE_CUSTOMIZED_OP,
-)
 from deepmd.pd.infer import (
     inference,
 )
@@ -216,18 +210,11 @@ def get_ngpus(self) -> int:
 
     def get_backend_info(self) -> dict:
         """Get backend information."""
-        if ENABLE_CUSTOMIZED_OP:
-            op_info = {
-                "build with PD ver": GLOBAL_CONFIG["pd_version"],
-                "build with PD inc": GLOBAL_CONFIG["pd_include_dir"].replace(";", "\n"),
-                "build with PD lib": GLOBAL_CONFIG["pd_libs"].replace(";", "\n"),
-            }
-        else:
-            op_info = {}
+        op_info = {}
         return {
             "Backend": "Paddle",
             "PD ver": f"v{paddle.__version__}-g{paddle.version.commit[:11]}",
-            "Enable custom OP": ENABLE_CUSTOMIZED_OP,
+            "Enable custom OP": False,
             **op_info,
         }
 

deepmd/pd/train/training.py

@@ -32,6 +32,9 @@
 from deepmd.common import (
     symlink_prefix_files,
 )
+from deepmd.dpmodel.utils.learning_rate import (
+    LearningRateExp,
+)
 from deepmd.loggers.training import (
     format_training_message_per_task,
 )
@@ -59,9 +62,6 @@
     SAMPLER_RECORD,
     enable_prim,
 )
-from deepmd.pd.utils.learning_rate import (
-    LearningRateExp,
-)
 from deepmd.pd.utils.stat import (
     make_stat_input,
 )