Skip to content

Commit 89ab2d3

Browse files
authored
fix(jax): fix typo c_differentiable -> r_differentiable (#4640)
We haven't met a situation where c_differentiable != r_differentiable. <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **Refactor** - Refined internal processing to streamline evaluation logic. These improvements optimize back-end calculations while ensuring consistent performance and user experience. <!-- end of auto-generated comment: release notes by coderabbit.ai --> Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
1 parent e9a6bd4 commit 89ab2d3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepmd/jax/model/base_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def forward_common_atomic(
4747
kk_redu = get_reduce_name(kk)
4848
model_predict[kk_redu] = jnp.sum(vv, axis=atom_axis)
4949
kk_derv_r, kk_derv_c = get_deriv_name(kk)
50-
if vdef.c_differentiable:
50+
if vdef.r_differentiable:
5151

5252
def eval_output(
5353
cc_ext,

0 commit comments

Comments
 (0)