feat(pt_expt): implement .pte inference pipeline with dynamic shapes#5284

Merged

wanghan-iapcm merged 16 commits intodeepmodeling:masterfrom

wanghan-iapcm:feat-pt-expt-infer

Mar 5, 2026

Commits on Mar 3, 2026

feat(pt_expt): implement .pte inference pipeline with dynamic shapes
Han Wang
committed

Commits on Mar 4, 2026

support ase nlist
Han Wang
committed
update doc
Han Wang
committed
fix device conversion
Han Wang
committed
Resolve Ruff ARG001 in _build_dynamic_shapes
Han Wang
committed
Updated the backend to use the GPU-aware NeighborStat from deepmd.pt_expt.utils.neighbor_stat (ported from PR #5270)
Han Wang
committed
use PT's AutoBatchSize to catch OOM
Han Wang
committed
build and extend coords on cuda device
Han Wang
committed
fix
Han Wang
committed
assign dtype for torch tensors input to neighbor stat
Han Wang
committed
provide options for make_fx rather than hard coding
Han Wang
committed
align the implementation of env_mat with the training branch
Han Wang
committed
borrow conftest.py
Han Wang
committed

Commits on Mar 5, 2026

fix : test device
Han Wang
committed
Merge branch 'master' into feat-pt-expt-infer
Han Wang
committed
revert change on inputs
Han Wang
committed