Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
94 commits
Select commit Hold shift + click to select a range
8368b26
merge master to devel (v3.1.0) (#4801)
njzjz Jun 11, 2025
1d95c18
build(deps): bump pypa/cibuildwheel from 2.23 to 3.0 (#4805)
dependabot[bot] Jun 16, 2025
1dc5b04
fix: use tuple in `xp.reshape` (#4808)
caic99 Jun 18, 2025
48a7de5
perf: use contiguous memory stride for edge/angle indices (#4804)
caic99 Jun 18, 2025
a3677b6
[pre-commit.ci] pre-commit autoupdate (#4807)
pre-commit-ci[bot] Jun 18, 2025
617d3e2
pd: support different label_dict in CINN (#4795)
HydrogenSulfate Jun 19, 2025
ba09114
pd: fix local_rank and in mutlti nodes training (#4811)
HydrogenSulfate Jun 19, 2025
fdfd4c0
fix: training speed might be incorrect (#4806)
caic99 Jun 20, 2025
3bd950f
pd: update loc_mapping for dpa3 in paddle backend (#4797)
HydrogenSulfate Jun 20, 2025
04da98f
[pre-commit.ci] pre-commit autoupdate (#4817)
pre-commit-ci[bot] Jun 24, 2025
089a5e2
fix(jax): use more safe_for_vector_norm (#4809)
njzjz Jun 24, 2025
e1e433c
Feat: support fparam/aparam in dp calculator (#4819)
anyangml Jun 24, 2025
205f924
fix: omit virial in dp test summary if not available (#4818)
caic99 Jun 25, 2025
ef5869a
fix(CI): clean up mpi4py index (#4822)
njzjz Jun 27, 2025
f8f01cb
fix(jax): fix the usage of `jaxlib.xla_extension` (#4824)
njzjz Jun 27, 2025
c5cd5e5
pd: add flag `CINN_ALLOW_DYNAMIC_SHAPE` for better performance with d…
HydrogenSulfate Jul 2, 2025
70604ba
[pre-commit.ci] pre-commit autoupdate (#4829)
pre-commit-ci[bot] Jul 3, 2025
272f573
[pre-commit.ci] pre-commit autoupdate (#4834)
pre-commit-ci[bot] Jul 8, 2025
c151e04
fix(dpmodel/pt/pd/jax): pass trainable to layer & support JAX trainab…
njzjz Jul 8, 2025
1eefc8e
pd: support dpa3 dynamic shape for pd backend (#4828)
HydrogenSulfate Jul 10, 2025
4ad67d5
feat(pt): add `observed-type` option for dp show (#4820)
iProzd Jul 11, 2025
5450066
feat(pt): add hook to last fitting layer output (#4789)
iProzd Jul 11, 2025
ba42f64
fix: fix pytorch in the cuda11 image (#4841)
njzjz Jul 15, 2025
2b32af5
[pre-commit.ci] pre-commit autoupdate (#4838)
pre-commit-ci[bot] Jul 16, 2025
68f0d21
fix(cc): use insert_or_assign instead of insert (#4844)
CaRoLZhangxy Jul 19, 2025
c41b0dd
[pre-commit.ci] pre-commit autoupdate (#4845)
pre-commit-ci[bot] Jul 25, 2025
46e9542
build(deps): bump pypa/cibuildwheel from 3.0 to 3.1 (#4851)
dependabot[bot] Jul 28, 2025
43504d3
[pre-commit.ci] pre-commit autoupdate (#4852)
pre-commit-ci[bot] Jul 29, 2025
dbc6d7b
fix(CI): prefer stable versions (#4857)
njzjz Aug 1, 2025
1dc1248
feat(pt): add Mean absolute percentage error (MAPE) loss for prop. pr…
SchrodingersCattt Aug 1, 2025
813bbc8
refactor(training): Average training loss for smoother and more repre…
OutisLi Aug 2, 2025
cec1944
docs: add bfloat16 option to the model precision choice (#4866)
caic99 Aug 4, 2025
98fb397
[pre-commit.ci] pre-commit autoupdate (#4868)
pre-commit-ci[bot] Aug 5, 2025
1fbe4c1
fix: merge `get_np_precision` to `get_xp_precision` (#4867)
njzjz Aug 5, 2025
102a2ed
doc: fix inconsistency between the docstring and the implementation o…
A-LOST-WAPITI Aug 6, 2025
bcc267e
chore: bump LAMMPS to stable_22Jul2025 (#4861)
njzjz Aug 7, 2025
3be0755
fix: no pinning memory on CPU (#4874)
caic99 Aug 7, 2025
91ebe34
feat(pd): support dpa2/dpa3 C++ inference (#4870)
HydrogenSulfate Aug 8, 2025
46411c1
Profile bug fix when both enable_profiler and profiling are set to tr…
OutisLi Aug 10, 2025
c7d8da6
docs: add docs about LAMMPS D3 dispersion (#4875)
njzjz Aug 11, 2025
8c545b0
build(deps): bump actions/download-artifact from 4 to 5 (#4881)
dependabot[bot] Aug 11, 2025
88b71e8
[pre-commit.ci] pre-commit autoupdate (#4882)
pre-commit-ci[bot] Aug 12, 2025
cefce47
Fix: support "max:N" and "filter:N" batch_size rules in DeepmdDataSys…
OutisLi Aug 12, 2025
1c29fe4
feat(pt): support zbl finetune (#4849)
iProzd Aug 13, 2025
7601889
fix(pt/pd): fix eta computation (#4886)
HydrogenSulfate Aug 13, 2025
accc331
fix: get correct intensive property prediction when using virtual ato…
Chengqian-Zhang Aug 13, 2025
e73bbff
fix(tf): fix compatibility with TF 2.20 (#4890)
njzjz Aug 15, 2025
565f0af
fix: relax `atol` and `rtol` value of padding atoms UT (#4892)
Chengqian-Zhang Aug 15, 2025
dc99ba2
doc(pd): update paddle installation scripts and paddle related conten…
HydrogenSulfate Aug 15, 2025
1525a79
fix(pt): fix CMake compatibility with PyTorch 2.8 (#4891)
njzjz Aug 18, 2025
1e75042
feat: add yaml input file support (#4894)
caic99 Aug 18, 2025
ed4f9d6
build(deps): bump actions/checkout from 4 to 5 (#4897)
dependabot[bot] Aug 19, 2025
6dac4f9
[pre-commit.ci] pre-commit autoupdate (#4898)
pre-commit-ci[bot] Aug 19, 2025
191759b
Fix(pt): add comm_dict for zbl, linear, dipole, dos, polar model to f…
OutisLi Aug 23, 2025
727ec3c
docs: add comprehensive GitHub Copilot instructions and environment s…
Copilot Aug 25, 2025
73755b3
fix(pt,pd): remove redundant tensor handling to eliminate tensor cons…
Copilot Aug 25, 2025
3f0087b
feat: Add eval-desc CLI command for descriptor evaluation with 3D out…
Copilot Aug 25, 2025
d94157e
build(deps): bump actions/upload-pages-artifact from 3 to 4 (#4918)
dependabot[bot] Aug 26, 2025
630f233
fix: Avoid setting pin_memory in tests (#4919)
caic99 Aug 26, 2025
6349238
[pre-commit.ci] pre-commit autoupdate (#4917)
pre-commit-ci[bot] Aug 26, 2025
64e108f
chore(CI): bump PyTorch from 2.7 to 2.8 (#4884)
njzjz Aug 26, 2025
996d192
fix(pd): change numel function return type from int to size_t to prev…
Copilot Aug 27, 2025
c796862
feat(pd): support gradient accumulation (#4920)
HydrogenSulfate Aug 27, 2025
58ce1aa
feat(pt): add model branch alias (#4883)
iProzd Aug 27, 2025
85ab1b9
feat(ci): skip workflows on bot branches to avoid redundant CI runs (…
Copilot Aug 28, 2025
aa44098
feat: handle masked forces in test (#4893)
caic99 Aug 29, 2025
c57d19f
feat: add comprehensive type hints to core modules excluding backends…
Copilot Aug 29, 2025
db22802
feat: support using train/valid data from input.json for dp test (#4859)
caic99 Aug 31, 2025
581bcb6
feat(tf): implement change-bias command (#4927)
Copilot Sep 2, 2025
aca4e8c
style: complete type annotation enforcement for deepmd.pt (#4943)
Copilot Sep 2, 2025
ea7f828
fix(tf): fix serialization of dipole fitting with sel_type (#4934)
Copilot Sep 2, 2025
adcacf9
feat(infer): add get_model method to DeepEval for accessing backend-s…
Copilot Sep 2, 2025
d1428e0
[pre-commit.ci] pre-commit autoupdate (#4948)
pre-commit-ci[bot] Sep 2, 2025
68ea2aa
docs(lmp): fix the usage of LAMMPS `pair_style hybrid/overlay` (#4951)
njzjz Sep 3, 2025
354c301
feat(dp/pt): add default_fparam (#4888)
iProzd Sep 4, 2025
0fe38c8
chore(deps): bump LAMMPS to stable_22Jul2025_update1 (#4955)
njzjz Sep 5, 2025
dab37b2
style(dpmodel): enforce type annotations (#4953)
Copilot Sep 8, 2025
00721b6
build(deps): bump actions/checkout from 4 to 5 (#4966)
dependabot[bot] Sep 8, 2025
a10f920
build(deps): bump actions/labeler from 5 to 6 (#4964)
dependabot[bot] Sep 8, 2025
ef5959f
build(deps): bump actions/setup-python from 5 to 6 (#4965)
dependabot[bot] Sep 8, 2025
40f0b1c
[pre-commit.ci] pre-commit autoupdate (#4968)
pre-commit-ci[bot] Sep 9, 2025
6d04648
feat: add PyTorch profiler support to LAMMPS MD (#4969)
caic99 Sep 15, 2025
34df2b4
style(jax): enable ANN rule and add comprehensive type hints to JAX b…
Copilot Sep 15, 2025
dd55dff
perf: fix cuda-aware mpi in v3 (#4977)
caic99 Sep 16, 2025
a0af89f
[pre-commit.ci] pre-commit autoupdate (#4976)
pre-commit-ci[bot] Sep 17, 2025
15c2b7d
fix: change eV/A to eV/Å for dp test (#4978)
OutisLi Sep 17, 2025
4ef7705
docs: move copilot-instructions.md to AGENTS.md (#4982)
Copilot Sep 18, 2025
7768832
fix: fix unit display in dp test (#4980)
njzjz Sep 18, 2025
417915a
fix(tf): make dipole, polar, and dos models consistent with dpmodel (…
Copilot Sep 18, 2025
e49741c
docs: clarify atomic_dipole meaning for DPLR models (#4979)
Copilot Sep 19, 2025
58f60c4
[pre-commit.ci] pre-commit autoupdate (#4989)
pre-commit-ci[bot] Sep 23, 2025
58e346a
pd(feat): support python inference with `DP` class (#4987)
HydrogenSulfate Sep 28, 2025
62f83bb
feat(pt): implement DeepTensorPT (#4937)
Copilot Sep 29, 2025
164c8c3
[pre-commit.ci] pre-commit autoupdate (#4997)
pre-commit-ci[bot] Sep 29, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
4 changes: 2 additions & 2 deletions .devcontainer/build_cxx.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,15 @@ NPROC=$(nproc --all)
SCRIPT_PATH=$(dirname $(realpath -s $0))

export CMAKE_PREFIX_PATH=${SCRIPT_PATH}/../libtorch
TENSORFLOW_ROOT=$(python -c 'import importlib,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')
TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')

mkdir -p ${SCRIPT_PATH}/../buildcxx/
cd ${SCRIPT_PATH}/../buildcxx/
cmake -D ENABLE_TENSORFLOW=ON \
-D ENABLE_PYTORCH=ON \
-D ENABLE_PADDLE=ON \
-D CMAKE_INSTALL_PREFIX=${SCRIPT_PATH}/../dp/ \
-D LAMMPS_VERSION=stable_29Aug2024_update1 \
-D LAMMPS_VERSION=stable_22Jul2025_update1 \
-D CMAKE_BUILD_TYPE=Debug \
-D BUILD_TESTING:BOOL=TRUE \
-D TENSORFLOW_ROOT=${TENSORFLOW_ROOT} \
Expand Down
1 change: 0 additions & 1 deletion .devcontainer/devcontainer.json
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
"PATH": "${containerEnv:PATH}:${containerWorkspaceFolder}/.venv/bin",
"DP_ENABLE_PYTORCH": "1",
"DP_VARIANT": "cpu",
"LMP_CXX11_ABI_0": "1",
"UV_EXTRA_INDEX_URL": "https://download.pytorch.org/whl/cpu"
}
}
2 changes: 1 addition & 1 deletion .devcontainer/download_libtorch.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,5 @@ set -ev
SCRIPT_PATH=$(dirname $(realpath -s $0))
cd ${SCRIPT_PATH}/..

wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.7.0%2Bcpu.zip -O ~/libtorch.zip
wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.8.0%2Bcpu.zip -O ~/libtorch.zip
unzip ~/libtorch.zip
2 changes: 1 addition & 1 deletion .devcontainer/gdb_lmp
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
SCRIPT_PATH=$(dirname $(realpath -s $0))

export CMAKE_PREFIX_PATH=${SCRIPT_PATH}/../libtorch
TENSORFLOW_ROOT=$(python -c 'import importlib,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')
TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')

env LAMMPS_PLUGIN_PATH=${SCRIPT_PATH}/../dp/lib/deepmd_lmp \
LD_LIBRARY_PATH=${SCRIPT_PATH}/../dp/lib:${CMAKE_PREFIX_PATH}/lib:${TENSORFLOW_ROOT} \
Expand Down
2 changes: 1 addition & 1 deletion .devcontainer/gdb_pytest_lmp
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
SCRIPT_PATH=$(dirname $(realpath -s $0))/../..

export CMAKE_PREFIX_PATH=${SCRIPT_PATH}/../libtorch
TENSORFLOW_ROOT=$(python -c 'import importlib,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')
TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')

env LAMMPS_PLUGIN_PATH=${SCRIPT_PATH}/../dp/lib/deepmd_lmp \
LD_LIBRARY_PATH=${SCRIPT_PATH}/../dp/lib:${CMAKE_PREFIX_PATH}/lib:${TENSORFLOW_ROOT} \
Expand Down
2 changes: 1 addition & 1 deletion .devcontainer/lmp
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
SCRIPT_PATH=$(dirname $(realpath -s $0))

export CMAKE_PREFIX_PATH=${SCRIPT_PATH}/../libtorch
TENSORFLOW_ROOT=$(python -c 'import importlib,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')
TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')

env LAMMPS_PLUGIN_PATH=${SCRIPT_PATH}/../dp/lib/deepmd_lmp \
LD_LIBRARY_PATH=${SCRIPT_PATH}/../dp/lib:${CMAKE_PREFIX_PATH}/lib:${TENSORFLOW_ROOT} \
Expand Down
2 changes: 1 addition & 1 deletion .devcontainer/pytest_lmp
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
SCRIPT_PATH=$(dirname $(realpath -s $0))/../..

export CMAKE_PREFIX_PATH=${SCRIPT_PATH}/../libtorch
TENSORFLOW_ROOT=$(python -c 'import importlib,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')
TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')

env LAMMPS_PLUGIN_PATH=${SCRIPT_PATH}/../dp/lib/deepmd_lmp \
LD_LIBRARY_PATH=${SCRIPT_PATH}/../dp/lib:${CMAKE_PREFIX_PATH}/lib:${TENSORFLOW_ROOT} \
Expand Down
9 changes: 6 additions & 3 deletions .github/workflows/build_cc.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,9 @@ on:
push:
branches-ignore:
- "gh-readonly-queue/**"
- "copilot/**"
- "dependabot/**"
- "pre-commit-ci-update-config"
pull_request:
merge_group:
concurrency:
Expand All @@ -26,14 +29,14 @@ jobs:
- variant: clang
dp_variant: clang
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
- uses: actions/checkout@v5
- uses: actions/setup-python@v6
with:
python-version: '3.11'
- uses: lukka/get-cmake@latest
- run: python -m pip install uv
- run: source/install/uv_with_retry.sh pip install --system tensorflow
- run: source/install/uv_with_retry.sh pip install --system 'torch==2.7' --index-url https://download.pytorch.org/whl/cpu
- run: source/install/uv_with_retry.sh pip install --system 'torch==2.8.*' --index-url https://download.pytorch.org/whl/cpu
- run: |
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb \
&& sudo dpkg -i cuda-keyring_1.0-1_all.deb \
Expand Down
21 changes: 12 additions & 9 deletions .github/workflows/build_wheel.yml
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,9 @@ on:
push:
branches-ignore:
- "gh-readonly-queue/**"
- "copilot/**"
- "dependabot/**"
- "pre-commit-ci-update-config"
tags:
- "v*"
pull_request:
Expand Down Expand Up @@ -54,7 +57,7 @@ jobs:
platform_id: manylinux_aarch64
dp_variant: cpu
steps:
- uses: actions/checkout@v4
- uses: actions/checkout@v5
with:
# https://github.com/pypa/setuptools_scm/issues/480
fetch-depth: 0
Expand All @@ -70,7 +73,7 @@ jobs:
rm -rf .git
if: matrix.dp_pkg_name == 'deepmd-kit-cu11'
- name: Build wheels
uses: pypa/cibuildwheel@v2.23
uses: pypa/cibuildwheel@v3.1
env:
CIBW_BUILD_VERBOSITY: 1
CIBW_ARCHS: all
Expand All @@ -87,7 +90,7 @@ jobs:
name: Build source distribution
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/checkout@v5
with:
fetch-depth: 0
- name: Build sdist
Expand All @@ -106,7 +109,7 @@ jobs:
id-token: write
if: github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/v')
steps:
- uses: actions/download-artifact@v4
- uses: actions/download-artifact@v5
with:
pattern: cibw-*
path: dist
Expand All @@ -128,8 +131,8 @@ jobs:
steps:
- name: Delete huge unnecessary tools folder
run: rm -rf /opt/hostedtoolcache
- uses: actions/checkout@v4
- uses: actions/download-artifact@v4
- uses: actions/checkout@v5
- uses: actions/download-artifact@v5
with:
path: source/install/docker/dist
pattern: cibw-*-manylinux_x86_64-cu${{ matrix.cuda_version }}*
Expand Down Expand Up @@ -162,12 +165,12 @@ jobs:
needs: [build_wheels, build_sdist]
runs-on: ubuntu-latest
steps:
- uses: actions/download-artifact@v4
- uses: actions/download-artifact@v5
with:
path: dist/packages
pattern: cibw-*
merge-multiple: true
- uses: actions/setup-python@v5
- uses: actions/setup-python@v6
name: Install Python
with:
python-version: '3.11'
Expand All @@ -176,7 +179,7 @@ jobs:
ls dist/packages > package_list.txt
dumb-pypi --output-dir dist --packages-url ../../packages --package-list package_list.txt --title "DeePMD-kit Developed Packages"
- name: Upload Pages artifact
uses: actions/upload-pages-artifact@v3
uses: actions/upload-pages-artifact@v4
with:
path: dist
deploy_pypi_index:
Expand Down
9 changes: 6 additions & 3 deletions .github/workflows/codeql.yml
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,9 @@ on:
push:
branches-ignore:
- "gh-readonly-queue/**"
- "copilot/**"
- "dependabot/**"
- "pre-commit-ci-update-config"
pull_request:
schedule:
- cron: '45 2 * * 2'
Expand All @@ -27,8 +30,8 @@ jobs:

steps:
- name: Checkout repository
uses: actions/checkout@v4
- uses: actions/setup-python@v5
uses: actions/checkout@v5
- uses: actions/setup-python@v6
with:
python-version: '3.11'
cache: 'pip'
Expand All @@ -41,7 +44,7 @@ jobs:
&& sudo apt-get update \
&& sudo apt-get -y install cuda-cudart-dev-12-2 cuda-nvcc-12-2
python -m pip install tensorflow
python -m pip install 'torch==2.7' --index-url https://download.pytorch.org/whl/cpu
python -m pip install 'torch==2.8.*' --index-url https://download.pytorch.org/whl/cpu
env:
DEBIAN_FRONTEND: noninteractive
# Initializes the CodeQL tools for scanning.
Expand Down
69 changes: 69 additions & 0 deletions .github/workflows/copilot-setup-steps.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,69 @@
name: "Copilot Setup Steps"

# Automatically run the setup steps when they are changed to allow for easy validation, and
# allow manual testing through the repository's "Actions" tab
on:
workflow_dispatch:
push:
branches-ignore:
- "copilot/**"
- "dependabot/**"
- "pre-commit-ci-update-config"
paths:
- .github/workflows/copilot-setup-steps.yml
pull_request:
paths:
- .github/workflows/copilot-setup-steps.yml

jobs:
# The job MUST be called `copilot-setup-steps` or it will not be picked up by Copilot.
copilot-setup-steps:
runs-on: ubuntu-latest

# Set the permissions to the lowest permissions possible needed for your steps.
# Copilot will be given its own token for its operations.
permissions:
# If you want to clone the repository as part of your setup steps, for example to install dependencies, you'll need the `contents: read` permission. If you don't clone the repository in your setup steps, Copilot will do this for you automatically after the steps complete.
contents: read

# You can define any steps you want, and they will run before the agent starts.
# If you do not check out your code, Copilot will do this for you.
steps:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: "3.10"

- name: Install uv
uses: astral-sh/setup-uv@v6
with:
enable-cache: true

- name: Create virtual environment
run: uv venv venv

- name: Activate virtual environment
run: echo "VIRTUAL_ENV=$PWD/venv" >> $GITHUB_ENV && echo "$PWD/venv/bin" >> $GITHUB_PATH

- name: Install base dependencies
run: uv pip install tensorflow-cpu

- name: Install PyTorch
run: uv pip install torch --index-url https://download.pytorch.org/whl/cpu

- name: Build Python package
run: uv pip install -e .[cpu,test]

- name: Install pre-commit tools
run: uv tool install pre-commit

- name: Install pre-commit hooks
run: pre-commit install --install-hooks

- name: Verify installation
run: |
dp --version
python -c "import deepmd; import deepmd.tf; print('DeePMD-kit installation verified')"
2 changes: 1 addition & 1 deletion .github/workflows/labeler.yml
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,6 @@ jobs:
pull-requests: write
runs-on: ubuntu-latest
steps:
- uses: actions/labeler@v5
- uses: actions/labeler@v6
with:
repo-token: "${{ secrets.GITHUB_TOKEN }}"
9 changes: 8 additions & 1 deletion .github/workflows/mirror_gitee.yml
Original file line number Diff line number Diff line change
@@ -1,6 +1,13 @@
name: Mirror to Gitee Repo

on: [ push, delete, create ]
on:
push:
branches-ignore:
- "copilot/**"
- "dependabot/**"
- "pre-commit-ci-update-config"
delete:
create:

# Ensures that only one mirror task will run at a time.
concurrency:
Expand Down
11 changes: 7 additions & 4 deletions .github/workflows/package_c.yml
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,9 @@ on:
push:
branches-ignore:
- "gh-readonly-queue/**"
- "copilot/**"
- "dependabot/**"
- "pre-commit-ci-update-config"
tags:
- "v*"
pull_request:
Expand All @@ -22,10 +25,10 @@ jobs:
tensorflow_version: ""
filename: libdeepmd_c.tar.gz
- tensorflow_build_version: "2.14"
tensorflow_version: ">=2.5.0rc0,<2.15"
tensorflow_version: ">=2.5.0,<2.15"
filename: libdeepmd_c_cu11.tar.gz
steps:
- uses: actions/checkout@v4
- uses: actions/checkout@v5
with:
fetch-depth: 0
- name: Package C library
Expand Down Expand Up @@ -53,9 +56,9 @@ jobs:
needs: [build_c]
runs-on: ubuntu-22.04
steps:
- uses: actions/checkout@v4
- uses: actions/checkout@v5
- name: Download artifact
uses: actions/download-artifact@v4
uses: actions/download-artifact@v5
with:
pattern: libdeepmd_c-*
merge-multiple: true
Expand Down
20 changes: 9 additions & 11 deletions .github/workflows/test_cc.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,9 @@ on:
push:
branches-ignore:
- "gh-readonly-queue/**"
- "copilot/**"
- "dependabot/**"
- "pre-commit-ci-update-config"
pull_request:
merge_group:
concurrency:
Expand All @@ -16,23 +19,19 @@ jobs:
matrix:
check_memleak: [true, false]
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
- uses: actions/checkout@v5
- uses: actions/setup-python@v6
with:
python-version: '3.11'
cache: 'pip'
- name: Setup MPI
uses: mpi4py/setup-mpi@v1
with:
mpi: mpich
- uses: lukka/get-cmake@latest
- run: python -m pip install uv
- name: Install Python dependencies
run: |
source/install/uv_with_retry.sh pip install --system tensorflow-cpu~=2.18.0 jax==0.5.0
export TENSORFLOW_ROOT=$(python -c 'import importlib,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')
source/install/uv_with_retry.sh pip install --system -e .[cpu,test,lmp,jax] mpi4py
source/install/uv_with_retry.sh pip install --system 'torch==2.7' --index-url https://download.pytorch.org/whl/cpu
export TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')
source/install/uv_with_retry.sh pip install --system -e .[cpu,test,lmp,jax] mpi4py mpich
source/install/uv_with_retry.sh pip install --system 'torch==2.8.*' --index-url https://download.pytorch.org/whl/cpu
- name: Convert models
run: source/tests/infer/convert-models.sh
# https://github.com/actions/runner-images/issues/9491
Expand All @@ -45,7 +44,6 @@ jobs:
OMP_NUM_THREADS: 1
TF_INTRA_OP_PARALLELISM_THREADS: 1
TF_INTER_OP_PARALLELISM_THREADS: 1
LMP_CXX11_ABI_0: 1
CMAKE_GENERATOR: Ninja
CXXFLAGS: ${{ matrix.check_memleak && '-fsanitize=leak' || '' }}
LSAN_OPTIONS: suppressions=${{ github.workspace }}/.github/workflows/suppr.txt
Expand All @@ -54,13 +52,13 @@ jobs:
cp ${{ github.workspace }}/source/build_tests/paddle_inference_install_dir/paddle/lib/*.so ${{ github.workspace }}/dp_test/lib/
cp ${{ github.workspace }}/source/build_tests/paddle_inference_install_dir/third_party/install/onednn/lib/* ${{ github.workspace }}/dp_test/lib/
cp ${{ github.workspace }}/source/build_tests/paddle_inference_install_dir/third_party/install/mklml/lib/* ${{ github.workspace }}/dp_test/lib/
export LD_LIBRARY_PATH=${{ github.workspace }}/dp_test/lib:$LD_LIBRARY_PATH
pytest --cov=deepmd source/lmp/tests
env:
OMP_NUM_THREADS: 1
TF_INTRA_OP_PARALLELISM_THREADS: 1
TF_INTER_OP_PARALLELISM_THREADS: 1
LAMMPS_PLUGIN_PATH: ${{ github.workspace }}/dp_test/lib/deepmd_lmp
LD_LIBRARY_PATH: ${{ github.workspace }}/dp_test/lib
if: ${{ !matrix.check_memleak }}
# test ipi
- run: |
Expand Down
Loading
Loading