Skip to content

Commit

Permalink
fix(jax): fix typo c_differentiable -> r_differentiable (#4640)
Browse files Browse the repository at this point in the history
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 <[email protected]>
  • Loading branch information
njzjz authored Mar 7, 2025
1 parent e9a6bd4 commit 89ab2d3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepmd/jax/model/base_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def forward_common_atomic(
kk_redu = get_reduce_name(kk)
model_predict[kk_redu] = jnp.sum(vv, axis=atom_axis)
kk_derv_r, kk_derv_c = get_deriv_name(kk)
if vdef.c_differentiable:
if vdef.r_differentiable:

def eval_output(
cc_ext,
Expand Down

0 comments on commit 89ab2d3

Please sign in to comment.