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.

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

pd: support dpa2 #4418

Open
wants to merge 81 commits into
base: devel
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
81 commits
Select commit Hold shift + click to select a range
48f77f3
add core modules of paddle backend and water/se_e2_a example
HydrogenSulfate Nov 2, 2024
2082a59
add paddle code in consistent test
HydrogenSulfate Nov 2, 2024
2ae45b8
clean env and training
HydrogenSulfate Nov 2, 2024
7f03a04
add more test files
HydrogenSulfate Nov 2, 2024
4d1c44c
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 2, 2024
72c9b4e
fix pt->pd
HydrogenSulfate Nov 2, 2024
3b1c348
update test_python.yml
HydrogenSulfate Nov 2, 2024
a46dcb5
restore .pre-commit-config.yaml
HydrogenSulfate Nov 3, 2024
90f9ff9
remove redundant file
HydrogenSulfate Nov 3, 2024
0a6baa6
Skip bfloat16 for some cases
HydrogenSulfate Nov 3, 2024
4b77e55
enable prim by default in unitest
HydrogenSulfate Nov 3, 2024
6e139a2
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Nov 3, 2024
9437957
fix env code
HydrogenSulfate Nov 5, 2024
f1d762f
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 5, 2024
8534597
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 6, 2024
c22b45d
update test_ener.py
HydrogenSulfate Nov 6, 2024
39842ff
add missing pd_class
HydrogenSulfate Nov 6, 2024
07cd98e
use paddle Tensor instead of numpy array in pd/test_auto_batch_size.p…
HydrogenSulfate Nov 6, 2024
bb2d547
add training test and remove ase_calc.py
HydrogenSulfate Nov 7, 2024
5fb6d8e
add training test and remove ase_calc.py
HydrogenSulfate Nov 7, 2024
91066f8
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 7, 2024
90c9c03
upload missing json
HydrogenSulfate Nov 7, 2024
eb7384e
restore pt/test_auto_batch_size.py
HydrogenSulfate Nov 7, 2024
9faf54f
rerun CI for network problem
HydrogenSulfate Nov 7, 2024
4e3a121
add multitask unitest
HydrogenSulfate Nov 7, 2024
18333ab
add more unitest
HydrogenSulfate Nov 7, 2024
f9c6da8
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 7, 2024
3fd979d
remove redundant file and fix typo
HydrogenSulfate Nov 7, 2024
5922e84
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Nov 7, 2024
f5a17a9
update unitest
HydrogenSulfate Nov 8, 2024
8bea1bf
delete record
HydrogenSulfate Nov 8, 2024
8a7875f
remove more unused code and files
HydrogenSulfate Nov 8, 2024
df9f887
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 8, 2024
67b81e1
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 8, 2024
b0bf733
Merge branch 'add_paddle_backend_core_and_water_se_e2_a' of https://g…
HydrogenSulfate Nov 8, 2024
71a3c0a
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Nov 8, 2024
ede5047
remove redundant annotations
HydrogenSulfate Nov 8, 2024
d11bf4d
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 8, 2024
b7a8cec
add nvtx profiler code in training, which is more accurate and detailed
HydrogenSulfate Nov 8, 2024
7567cf8
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 8, 2024
416fec8
update code as devel and fix typo
HydrogenSulfate Nov 9, 2024
1c0161c
fix pth -> json
HydrogenSulfate Nov 9, 2024
02a6f84
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 9, 2024
3354e5c
update unitest and training
HydrogenSulfate Nov 9, 2024
0d3f8cf
install paddle when test_cuda
HydrogenSulfate Nov 9, 2024
18215ff
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 9, 2024
859b94d
fix unitest
HydrogenSulfate Nov 9, 2024
74ee1c2
add eta in logging message for convenient
HydrogenSulfate Nov 9, 2024
f176309
remove hybrid code and enable one unitest
HydrogenSulfate Nov 9, 2024
4935e7b
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 9, 2024
fac51d3
add pd/__init__.py
HydrogenSulfate Nov 11, 2024
d3ca1f0
Merge branch 'devel' into add_paddle_backend_core_and_water_se_e2_a
HydrogenSulfate Nov 11, 2024
db1cd76
fix enable_prim
HydrogenSulfate Nov 11, 2024
36512fd
remove unused layernorm
HydrogenSulfate Nov 11, 2024
351bf7a
update dpa1 code
HydrogenSulfate Nov 25, 2024
01d4179
Merge branch 'devel' into add_dpa1
HydrogenSulfate Nov 25, 2024
701926a
update dpa2 code
HydrogenSulfate Nov 25, 2024
bc1cb38
update code of dpa1
HydrogenSulfate Nov 28, 2024
b4bc9db
Merge branch 'devel' into add_dpa1
HydrogenSulfate Nov 28, 2024
c944b82
restore decomp to paddle function
HydrogenSulfate Nov 28, 2024
4c925f9
remove redundant files
HydrogenSulfate Nov 28, 2024
dd3191a
update unitest and codes
HydrogenSulfate Nov 29, 2024
7df0e2f
fix
HydrogenSulfate Nov 29, 2024
ac479ed
update code
HydrogenSulfate Nov 29, 2024
3e64196
Merge branch 'devel' into add_dpa1
HydrogenSulfate Nov 29, 2024
56e079c
update typos
HydrogenSulfate Nov 29, 2024
3d70e7c
update code
HydrogenSulfate Nov 29, 2024
8b5b4a8
fix coverage
HydrogenSulfate Nov 29, 2024
e74d272
update consistent check of dpa1
HydrogenSulfate Nov 30, 2024
1a79a06
Merge branch 'add_dpa1' into add_dpa2_v2
HydrogenSulfate Dec 1, 2024
4520e97
add unittests code of dpa2 and replace several decomp. API to paddle.…
HydrogenSulfate Dec 1, 2024
79ed0f0
fix typos
HydrogenSulfate Dec 1, 2024
ed80c6d
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Dec 2, 2024
89d82f9
update UT code in test_se_t_tebd
HydrogenSulfate Dec 2, 2024
984b242
Merge branch 'add_dpa2_v22' into add_dpa2_v2
HydrogenSulfate Dec 2, 2024
617a258
update __init__
HydrogenSulfate Dec 2, 2024
6e5ebb3
solve code QL
HydrogenSulfate Dec 3, 2024
3cfb90f
Merge branch 'devel' into add_dpa2_v2
HydrogenSulfate Dec 18, 2024
575726a
fix unitest and typo
HydrogenSulfate Dec 19, 2024
befc111
Merge branch 'devel' into add_dpa2_v2
HydrogenSulfate Dec 23, 2024
abeae6d
update reference in dpa2.py
HydrogenSulfate Dec 23, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions deepmd/pd/model/descriptor/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,20 +9,34 @@
DescrptBlockSeAtten,
DescrptDPA1,
)
from .dpa2 import (
DescrptDPA2,
)
from .env_mat import (
prod_env_mat,
)
from .repformers import (
DescrptBlockRepformers,
)
from .se_a import (
DescrptBlockSeA,
DescrptSeA,
)
from .se_t_tebd import (
DescrptBlockSeTTebd,
DescrptSeTTebd,
)

__all__ = [
"BaseDescriptor",
"DescriptorBlock",
"DescrptBlockRepformers",
"DescrptBlockSeA",
"DescrptBlockSeAtten",
"DescrptBlockSeTTebd",
"DescrptDPA1",
"DescrptDPA2",
"DescrptSeA",
"DescrptSeTTebd",
"prod_env_mat",
]
Loading
Loading