Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(jax): build nlist in the SavedModel & fix nopbc for StableHLO and SavedModel #4318

Merged
merged 32 commits into from
Nov 11, 2024

clean up

e4bac35
Select commit
Loading
Failed to load commit list.
Merged

feat(jax): build nlist in the SavedModel & fix nopbc for StableHLO and SavedModel #4318

clean up
e4bac35
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch succeeded Nov 9, 2024 in 0s

68.85% of diff hit (within 100.00% threshold of 84.50%)

View this Pull Request on Codecov

68.85% of diff hit (within 100.00% threshold of 84.50%)

Annotations

Check warning on line 78 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L73-L78 

Added lines #L73 - L78 were not covered by tests

Check warning on line 84 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L83-L84 

Added lines #L83 - L84 were not covered by tests

Check warning on line 87 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L87 

Added line #L87 was not covered by tests

Check warning on line 96 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L95-L96 

Added lines #L95 - L96 were not covered by tests

Check warning on line 104 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L104 

Added line #L104 was not covered by tests

Check warning on line 110 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L110 

Added line #L110 was not covered by tests

Check warning on line 67 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L67 

Added line #L67 was not covered by tests

Check warning on line 94 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L94 

Added line #L94 was not covered by tests

Check warning on line 98 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L98 

Added line #L98 was not covered by tests

Check warning on line 188 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L185-L188 

Added lines #L185 - L188 were not covered by tests

Check warning on line 26 in deepmd/jax/jax2tf/region.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/region.py#L25-L26 

Added lines #L25 - L26 were not covered by tests

Check warning on line 72 in deepmd/jax/jax2tf/region.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/region.py#L70-L72 

Added lines #L70 - L72 were not covered by tests

Check warning on line 126 in deepmd/jax/jax2tf/serialization.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/serialization.py#L125-L126 

Added lines #L125 - L126 were not covered by tests

Check warning on line 128 in deepmd/jax/jax2tf/serialization.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/serialization.py#L128 

Added line #L128 was not covered by tests

Check warning on line 130 in deepmd/jax/jax2tf/serialization.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/serialization.py#L130 

Added line #L130 was not covered by tests

Check warning on line 160 in deepmd/jax/jax2tf/serialization.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/serialization.py#L160 

Added line #L160 was not covered by tests

Check warning on line 174 in deepmd/jax/jax2tf/serialization.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/serialization.py#L174 

Added line #L174 was not covered by tests

Check warning on line 193 in deepmd/jax/jax2tf/serialization.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/serialization.py#L193 

Added line #L193 was not covered by tests

Check warning on line 216 in deepmd/jax/jax2tf/serialization.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/serialization.py#L216 

Added line #L216 was not covered by tests

Check warning on line 147 in deepmd/jax/jax2tf/tfmodel.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/tfmodel.py#L147 

Added line #L147 was not covered by tests

Check warning on line 32 in deepmd/jax/jax2tf/transform_output.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/transform_output.py#L31-L32 

Added lines #L31 - L32 were not covered by tests

Check warning on line 59 in deepmd/jax/jax2tf/transform_output.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/transform_output.py#L45-L59 

Added lines #L45 - L59 were not covered by tests

Check warning on line 61 in deepmd/jax/jax2tf/transform_output.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/transform_output.py#L61 

Added line #L61 was not covered by tests

Check warning on line 71 in deepmd/jax/jax2tf/transform_output.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/transform_output.py#L71 

Added line #L71 was not covered by tests

Check warning on line 76 in deepmd/jax/jax2tf/transform_output.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/transform_output.py#L76 

Added line #L76 was not covered by tests
View more details on Codecov