wanghan-iapcm
diff --git a/‎deepmd/dpmodel/descriptor/se_e2_a.py‎
Lines changed: 1 addition & 1 deletion b/‎deepmd/dpmodel/descriptor/se_e2_a.py‎
Lines changed: 1 addition & 1 deletion
@@ -609,7 +609,7 @@ def call(
         ng = self.neuron[-1]
         gr = xp.zeros(
             [nf * nloc, ng, 4],
-            dtype=self.dstd.dtype,
+            dtype=input_dtype,
             device=array_api_compat.device(coord_ext),
         )
         exclude_mask = self.emask.build_type_exclude_mask(nlist, atype_ext)
Original file line number	Diff line number	Diff line change
`@@ -609,7 +609,7 @@ def call(`
`609`	`609`	`ng = self.neuron[-1]`
`610`	`610`	`gr = xp.zeros(`
`611`	`611`	`[nf * nloc, ng, 4],`
`612`		`- dtype=self.dstd.dtype,`
	`612`	`+ dtype=input_dtype,`
`613`	`613`	`device=array_api_compat.device(coord_ext),`
`614`	`614`	`)`
`615`	`615`	`exclude_mask = self.emask.build_type_exclude_mask(nlist, atype_ext)`