Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
31 commits
Select commit Hold shift + click to select a range
03cc9ee
feat(dpmodel): relocate Lebedev rules with numpy loader
Jun 11, 2026
eccbe4d
test(dpmodel): cover Lebedev loader error paths; review fixes
Jun 11, 2026
ea0c471
feat(dpmodel): general-lmax real spherical harmonics (e3nn conventions)
Jun 11, 2026
f8798e1
fix(dpmodel): match e3nn zero-vector handling in real SH; e3nn-free c…
Jun 11, 2026
18dde4b
feat(dpmodel): dpa4_nn indexing + utils
Jun 11, 2026
8a12493
test(dpmodel): accurate parity-file docstring, import order, literal …
Jun 11, 2026
f51d986
feat(dpmodel): dpa4_nn radial basis, envelope, radial MLP
Jun 11, 2026
80fc4a8
feat(dpmodel): dpa4_nn Wigner-D calculator + edge quaternions
Jun 11, 2026
3916c21
feat(dpmodel): dpa4_nn norms, SO3 linear, gated activation
Jun 11, 2026
dee01f5
test(dpmodel): tighten fp32 parity tolerances to measured ulp
Jun 11, 2026
23107df
feat(dpmodel): dpa4_nn S2 grid projection + grid nets (Lebedev)
Jun 11, 2026
3115285
test(dpmodel): keep fp32 parity tests tolerance-based for truncation …
Jun 11, 2026
66be7c3
feat(dpmodel): dpa4_nn SO2 linear + convolution with padded-edge aggr…
Jun 11, 2026
a89ce39
docs(dpmodel): clarify EdgeCache padded-layout and cache-reuse contract
Jun 11, 2026
2d6fa3a
feat(dpmodel): dpa4_nn type, geometric, environment embeddings
Jun 11, 2026
fab9bf3
fix(dpmodel): uniform precision-key lookup; honest GIE zonal test sco…
Jun 11, 2026
abc0574
feat(dpmodel): dpa4_nn padded edge-cache builder
Jun 11, 2026
0c7c822
feat(dpmodel): dpa4_nn equivariant FFN + interaction block
Jun 11, 2026
c8ebada
feat(dpmodel): DPA4 descriptor (core path)
Jun 11, 2026
ce2e2a9
feat(dpmodel): dpa4_ener GLU energy fitting
Jun 11, 2026
b2bbfff
test(dpmodel): DPA4 cross-backend consistency tests and model-dict co…
Jun 11, 2026
80b4ba1
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 11, 2026
93e4d0b
fix(dpmodel): address review findings (xp-namespace constants, valida…
Jun 11, 2026
a625528
test(dpmodel): ulp tolerance for fp32/fp16 safe_norm parity (CI BLAS …
Jun 11, 2026
225c8c5
style(dpmodel): NumPy capitalization; explicit dtypes for pylint
Jun 11, 2026
70702ce
fix(dpmodel): xp-namespace conversion for remaining numpy buffers; Ke…
Jun 11, 2026
177c057
test(dpmodel): pin pt reference modules to CPU in parity tests (CUDA …
Jun 11, 2026
cb395c8
ci: retrigger checks
Jun 11, 2026
0b37974
test(dpmodel): also patch import-time device snapshots for CPU parity…
Jun 12, 2026
33180fa
test(dpmodel): run pt parity references on native device with device-…
Jun 12, 2026
ed0e303
test(dpmodel): move edge-cache fixture tensors to the pt device too
Jun 12, 2026
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
4 changes: 4 additions & 0 deletions deepmd/dpmodel/descriptor/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,9 @@
from .dpa3 import (
DescrptDPA3,
)
from .dpa4 import (
DescrptDPA4,
)
from .hybrid import (
DescrptHybrid,
)
Expand All @@ -34,6 +37,7 @@
"DescrptDPA1",
"DescrptDPA2",
"DescrptDPA3",
"DescrptDPA4",
"DescrptHybrid",
"DescrptSeA",
"DescrptSeAttenV2",
Expand Down
Loading
Loading