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

Add 4 pt descriptor compression #4227

Merged
merged 102 commits into from
Nov 1, 2024
Merged
Changes from 1 commit
Commits
Show all changes
102 commits
Select commit Hold shift + click to select a range
6bb8795
yan devel
cherryWangY Jun 9, 2024
7ecd122
tabulate_fusion_se_t
cherryWangY Jun 10, 2024
ab670ed
tabulate_fusion_all_op_basic_verion
cherryWangY Jun 10, 2024
9fc3fb0
compile safe version
cherryWangY Jun 10, 2024
cab50c9
compile safe version
cherryWangY Jun 10, 2024
e9ccb98
Merge branch 'devel' of https://github.com/cherryWangY/deepmd-kit int…
cherryWangY Jun 13, 2024
87909e2
se_a & se_atten
cherryWangY Jun 13, 2024
2225975
se_r
cherryWangY Jun 13, 2024
c09a7a7
remove print
cherryWangY Jun 13, 2024
ee5b64e
move pt op test
cherryWangY Jun 13, 2024
c7efbce
Merge remote-tracking branch 'upstream/devel' into devel
cherryWangY Jun 13, 2024
763c7b4
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 13, 2024
3fe4b64
remove print
cherryWangY Jun 13, 2024
6f76ccf
fixed for commit
cherryWangY Jun 13, 2024
b89ceed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 13, 2024
5a8b77e
fix pull request warning
cherryWangY Jun 13, 2024
25ca8c1
fix pr warning
cherryWangY Jun 13, 2024
f1c43f5
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 13, 2024
179e175
gpu test debug
cherryWangY Jun 16, 2024
34c664c
merge
cherryWangY Jun 16, 2024
4cc1478
merge
cherryWangY Jun 16, 2024
5921a60
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 16, 2024
b63209c
table_info set cpu
cherryWangY Jun 17, 2024
a824a80
remove print
cherryWangY Jun 17, 2024
8527819
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 17, 2024
e47dcba
add dtype=float64
cherryWangY Jun 17, 2024
95a9566
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 17, 2024
114f7a6
add dtype=float64
cherryWangY Jun 17, 2024
9e677f4
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 17, 2024
22ae3b7
test both float64 and float32
njzjz Jun 17, 2024
9920e57
skip tests if customized ops are not enables
njzjz Jun 17, 2024
7dd7f6a
Merge branch 'devel' into devel
cherryWangY Jun 18, 2024
82a5035
Merge branch 'devel' into devel
cherryWangY Jun 20, 2024
e6bc120
Merge branch 'devel' into tabulate_op
njzjz Jun 21, 2024
1523196
reduce test size from 192 atoms to 4 atoms
njzjz Jun 21, 2024
4c6f69b
Merge branch 'deepmodeling:devel' into devel
cherryWangY Jun 29, 2024
ff57db2
basic descriptor se_a
cherryWangY Jul 7, 2024
fdb13bb
Merge branch 'devel' of https://github.com/cherryWangY/deepmd-kit int…
cherryWangY Jul 7, 2024
882297c
basic descriptor se_a
cherryWangY Jul 7, 2024
4afd634
basic descriptor se_a
cherryWangY Jul 7, 2024
6880e8b
basic torch version
cherryWangY Jul 27, 2024
48a7508
compressed se_a test
cherryWangY Aug 6, 2024
1af34ce
se_a debug
cherryWangY Aug 28, 2024
e9915e3
four descriptors compression
cherryWangY Oct 17, 2024
322ad02
remove redundant code for pt descriptor compression
cherryWangY Oct 21, 2024
d7813d9
align to latest version
cherryWangY Oct 21, 2024
0331c43
align to latest version
cherryWangY Oct 21, 2024
d6cea9b
Merge branch 'devel' into devel
cherryWangY Oct 21, 2024
8984eb9
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 21, 2024
c38472a
remove extra indent
cherryWangY Oct 22, 2024
7d9d1e2
fix pre-commit
cherryWangY Oct 22, 2024
ed1d598
Merge branch 'devel' into devel
cherryWangY Oct 22, 2024
f12fa1b
enhance code robustness
cherryWangY Oct 23, 2024
95e4f9f
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 23, 2024
776816f
fix pre-commit
cherryWangY Oct 23, 2024
a0d7403
solve some problems
cherryWangY Oct 24, 2024
0d46b21
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 24, 2024
2617b6f
fix annotation error
cherryWangY Oct 24, 2024
7b9332b
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 24, 2024
e2106af
fix self.table and remove locals
cherryWangY Oct 25, 2024
6e1f787
merge latest
cherryWangY Oct 25, 2024
2802947
fix se_atten
cherryWangY Oct 25, 2024
b572034
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 25, 2024
8e86382
fix gg undefine
cherryWangY Oct 25, 2024
3eb8d3f
Merge remote-tracking branch 'new-fork/devel' into devel
cherryWangY Oct 25, 2024
7d84da6
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 25, 2024
37b1644
make torchscript happy
njzjz Oct 25, 2024
c0ab006
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 25, 2024
c924750
add fake op for freezing
cherryWangY Oct 26, 2024
78f20fd
add explicit device
cherryWangY Oct 26, 2024
755401e
add parametized tests
cherryWangY Oct 26, 2024
976c8df
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 26, 2024
126a170
fix parameterized
cherryWangY Oct 26, 2024
706d614
remove useless exception
cherryWangY Oct 26, 2024
b92c1f8
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 26, 2024
4d3c1b8
fix prec bug
cherryWangY Oct 26, 2024
619e5fb
Merge branch 'devel' of https://github.com/cherryWangY/deepmd-kit int…
cherryWangY Oct 26, 2024
1b1f0e9
set se_t precesion as 1e-6 when float64
cherryWangY Oct 27, 2024
b3737b9
avoid using env.DEVICE in the forward
cherryWangY Oct 27, 2024
90c5ac3
change enable_compression
cherryWangY Oct 27, 2024
5d8c96a
solve coderabbitai conversations
cherryWangY Oct 27, 2024
feda81b
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 27, 2024
7231776
change error info
cherryWangY Oct 29, 2024
05c9145
avoid calling serialize() multiple times
cherryWangY Oct 29, 2024
dbe92ef
add enable_compression() to BaseBaseModel
cherryWangY Oct 29, 2024
a7b9f66
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 29, 2024
f5d1757
remove enable_compression() in base_model
cherryWangY Oct 30, 2024
3fb094a
add enable_compression() in base_descriptor
cherryWangY Oct 30, 2024
621b45f
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 30, 2024
99f9f6b
simplified code
cherryWangY Oct 30, 2024
4df6df0
update branch
cherryWangY Oct 30, 2024
30c722c
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 30, 2024
7c86385
merge pt and tf similar implementation of tabulate
cherryWangY Oct 31, 2024
2ce4356
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 31, 2024
39d6d17
Merge branch 'devel' into devel
cherryWangY Oct 31, 2024
cb15335
set is_pt
cherryWangY Oct 31, 2024
15bf3e2
fix for loop; fix codeql warnings
njzjz Oct 31, 2024
689748e
add comment at descrpt SeT build()
cherryWangY Nov 1, 2024
9c7534e
Refactor duplicate code in _get_bias and _get_matrix methods
cherryWangY Nov 1, 2024
5794248
fix device inconsistency
cherryWangY Nov 1, 2024
c06f54a
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Nov 1, 2024
006c159
Merge branch 'devel' into devel
cherryWangY Nov 1, 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
Prev Previous commit
Next Next commit
test both float64 and float32
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz committed Jun 17, 2024
commit 22ae3b70113d4265d30f1ecd393cd04c0b1732ce
6 changes: 5 additions & 1 deletion source/tests/pt/test_tabulate_fusion_se_a.py
Original file line number Diff line number Diff line change
@@ -7,11 +7,15 @@
env,
)

dtype = torch.float64
from ..consistent.common import (
parameterized,
)


@parameterized((torch.float64, torch.float32))
class TestTabulateFusionSeAOp(unittest.TestCase):
def setUp(self):
(dtype,) = self.param
if dtype == torch.float64:
self.prec = 1e-10
elif dtype == torch.float32:
6 changes: 5 additions & 1 deletion source/tests/pt/test_tabulate_fusion_se_atten.py
Original file line number Diff line number Diff line change
@@ -7,11 +7,15 @@
env,
)

dtype = torch.float64
from ..consistent.common import (
parameterized,
)


@parameterized((torch.float64, torch.float32))
class TestTabulateFusionSeAttenOp(unittest.TestCase):
def setUp(self):
(dtype,) = self.param
if dtype == torch.float64:
self.prec = 1e-10
elif dtype == torch.float32:
6 changes: 5 additions & 1 deletion source/tests/pt/test_tabulate_fusion_se_r.py
Original file line number Diff line number Diff line change
@@ -7,11 +7,15 @@
env,
)

dtype = torch.float64
from ..consistent.common import (
parameterized,
)


@parameterized((torch.float64, torch.float32))
class TestTabulateFusionSeAOp(unittest.TestCase):
def setUp(self):
(dtype,) = self.param
if dtype == torch.float64:
self.prec = 1e-10
elif dtype == torch.float32:
9 changes: 7 additions & 2 deletions source/tests/pt/test_tabulate_fusion_se_t.py
Original file line number Diff line number Diff line change
@@ -7,15 +7,20 @@
env,
)

dtype = torch.float64
from ..consistent.common import (
parameterized,
)


@parameterized((torch.float64, torch.float32))
class TestTabulateFusionSeTOp(unittest.TestCase):
def setUp(self):
(dtype,) = self.param
if dtype == torch.float64:
self.prec = 1e-10
elif dtype == torch.float32:
self.prec = 1e-5
# JZ: not sure the reason, but 1e-5 cannot pass the grad test
self.prec = 1e-3
self.table_tensor = torch.tensor(
[
-1.0600000163027882e02,