pair_e3gnn.cpp uses pair-wise force.#301
Closed
lee2tae wants to merge 4 commits intoMDIL-SNU:mainfrom
Closed
Conversation
- Support inferred_atomic_virial across deploy, inference, calculator, and util flows. - Introduce atomic_stress option in LAMMPS patching for atomic-stress pair sources. - Maintain backward compatibility when atomic options are disabled.
… virial unit regression tests
for more information, see https://pre-commit.ci
Member
|
Deploy 모델도 calculator 에서 호출 가능하게 하는 기능을 없앱시다. 일단 요 작업들은 pr을 새로 여는게 아니라 원본 pr에서 하는게 맞는데 권한이 없으신가요? 어떻게 하는지 모른다면 'gh pr checkout' <<< 이 커맨드를 쓰면 됩니다. 결과적으로 redundant한 추가 cpp 파일이나 deploy 할때 virial flag 같은건 다 없어져야 합니다. (지금 불필요한 diff가 너무 많습니다) 추가로, 이게 lammps serial에서 속도에 변화를 줄 수도 있는데 이 부분 기존코드랑 비교해서 확인을 해봐야합니다. |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Started from pr-297
pair_e3gnn_parallel.cpp 코드 참고하면서 C++에서 autograd를 통해 F, s 구하도록 구현하였고, atomic stress 계산하는 로직도 force_output.py 참고해서 추가했습니다(2번 방법인 force_output 모듈 지우고 LAMMPS에서 autograd).
그러다 보니 deploy된 torchscript를 SevenNetCalculator에서 input으로 사용하지 못 하게 되는 거 아닌가? 하는 생각이 들어서 우선 기존의 deploy를 deploy_ts의 형태로 임시로 따로 떼어 놨습니다. (get_model에서 --atomic_virial flag를 deploy_ts로 연결) (제가 제대로 이해한 게 맞나요? deploy된 ts를 LAMMPS potential로만 사용한다면 SevenNetCalculator가 input으로 ts를 받아서 사용할 필요가 있나 궁금합니다.)
atomic stress 관련된 pytest --> test_lammps