make dtype same as input x to avoid auto floating promotion

deepmodeling · Dec 22, 2024 · 61e05e2 · 61e05e2
1 parent c24498b
commit 61e05e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pd/utils/region.py b/deepmd/pd/utils/region.py
@@ -108,5 +108,5 @@ def normalize_coord(
 
     """
     icoord = phys2inter(coord, cell)
-    icoord = paddle.remainder(icoord, paddle.full([], 1.0))
+    icoord = paddle.remainder(icoord, paddle.full([], 1.0, dtype=icoord.dtype))
     return inter2phys(icoord, cell)