From 5725b5fd8e48cb7552f05727985c33302ceb7b02 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 4 Aug 2025 21:21:15 +0800 Subject: [PATCH 1/3] chore: merge get_np_precision to get_xp_precision --- deepmd/common.py | 18 +++++------------- 1 file changed, 5 insertions(+), 13 deletions(-) diff --git a/deepmd/common.py b/deepmd/common.py index 5581f3533b..083d1e04e9 100644 --- a/deepmd/common.py +++ b/deepmd/common.py @@ -27,9 +27,6 @@ import numpy as np import yaml -from deepmd.env import ( - GLOBAL_NP_FLOAT_PRECISION, -) from deepmd.utils.path import ( DPPath, ) @@ -249,16 +246,11 @@ def get_np_precision(precision: "_PRECISION") -> np.dtype: RuntimeError if string is invalid """ - if precision == "default": - return GLOBAL_NP_FLOAT_PRECISION - elif precision == "float16": - return np.float16 - elif precision == "float32": - return np.float32 - elif precision == "float64": - return np.float64 - else: - raise RuntimeError(f"{precision} is not a valid precision") + from deepmd.dpmodel.common import ( + get_xp_precision, + ) + + return get_xp_precision(np, precision) def symlink_prefix_files(old_prefix: str, new_prefix: str) -> None: From 5e88dcd51922f8ef40b5c2bbdf696ce5a635fca6 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Tue, 5 Aug 2025 14:41:54 +0800 Subject: [PATCH 2/3] add GLOBAL_NP_FLOAT_PRECISION to all Signed-off-by: Jinzhe Zeng --- deepmd/common.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/deepmd/common.py b/deepmd/common.py index 083d1e04e9..c0542255df 100644 --- a/deepmd/common.py +++ b/deepmd/common.py @@ -27,6 +27,9 @@ import numpy as np import yaml +from deepmd.env import ( + GLOBAL_NP_FLOAT_PRECISION, +) from deepmd.utils.path import ( DPPath, ) @@ -39,6 +42,7 @@ "j_loader", "make_default_mesh", "select_idx_map", + "GLOBAL_NP_FLOAT_PRECISION", ] _PRECISION = Literal["default", "float16", "bfloat16", "float32", "float64"] From d8dfdfc65012aa3049c45e5c3fc56a1d86019c00 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 5 Aug 2025 06:43:29 +0000 Subject: [PATCH 3/3] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- deepmd/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/common.py b/deepmd/common.py index c0542255df..03afdbc2c2 100644 --- a/deepmd/common.py +++ b/deepmd/common.py @@ -35,6 +35,7 @@ ) __all__ = [ + "GLOBAL_NP_FLOAT_PRECISION", "VALID_ACTIVATION", "VALID_PRECISION", "expand_sys_str", @@ -42,7 +43,6 @@ "j_loader", "make_default_mesh", "select_idx_map", - "GLOBAL_NP_FLOAT_PRECISION", ] _PRECISION = Literal["default", "float16", "bfloat16", "float32", "float64"]