Skip to content

pair_e3gnn.cpp uses pair-wise force.#301

Closed
lee2tae wants to merge 4 commits intoMDIL-SNU:mainfrom
lee2tae:pw_e3gnn
Closed

pair_e3gnn.cpp uses pair-wise force.#301
lee2tae wants to merge 4 commits intoMDIL-SNU:mainfrom
lee2tae:pw_e3gnn

Conversation

@lee2tae
Copy link
Copy Markdown
Collaborator

@lee2tae lee2tae commented Mar 25, 2026

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

gshs12051 and others added 4 commits March 21, 2026 22:35
- 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.
@YutackPark
Copy link
Copy Markdown
Member

Deploy 모델도 calculator 에서 호출 가능하게 하는 기능을 없앱시다.

일단 요 작업들은 pr을 새로 여는게 아니라 원본 pr에서 하는게 맞는데 권한이 없으신가요? 어떻게 하는지 모른다면 'gh pr checkout' <<< 이 커맨드를 쓰면 됩니다.

결과적으로 redundant한 추가 cpp 파일이나 deploy 할때 virial flag 같은건 다 없어져야 합니다. (지금 불필요한 diff가 너무 많습니다)

추가로, 이게 lammps serial에서 속도에 변화를 줄 수도 있는데 이 부분 기존코드랑 비교해서 확인을 해봐야합니다.

@lee2tae lee2tae closed this Mar 27, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants