Skip to content

Commit 5e88dcd

Browse files
authored
add GLOBAL_NP_FLOAT_PRECISION to all
Signed-off-by: Jinzhe Zeng <jinzhe.zeng@ustc.edu.cn>
1 parent 5725b5f commit 5e88dcd

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

deepmd/common.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,9 @@
2727
import numpy as np
2828
import yaml
2929

30+
from deepmd.env import (
31+
GLOBAL_NP_FLOAT_PRECISION,
32+
)
3033
from deepmd.utils.path import (
3134
DPPath,
3235
)
@@ -39,6 +42,7 @@
3942
"j_loader",
4043
"make_default_mesh",
4144
"select_idx_map",
45+
"GLOBAL_NP_FLOAT_PRECISION",
4246
]
4347

4448
_PRECISION = Literal["default", "float16", "bfloat16", "float32", "float64"]

0 commit comments

Comments
 (0)