@@ -18,69 +18,69 @@ jobs:
1818 strategy :
1919 matrix :
2020 include :
21- - variant : cpu
22- dp_variant : cpu
23- - variant : cuda120
24- dp_variant : cuda
25- - variant : rocm
26- dp_variant : rocm
27- - variant : clang
28- dp_variant : clang
21+ - variant : cpu
22+ dp_variant : cpu
23+ - variant : cuda120
24+ dp_variant : cuda
25+ - variant : rocm
26+ dp_variant : rocm
27+ - variant : clang
28+ dp_variant : clang
2929 steps :
30- - uses : actions/checkout@v6
31- - uses : actions/setup-python@v6
32- with :
33- python-version : ' 3.11'
34- - uses : lukka/get-cmake@latest
35- - run : python -m pip install uv
36- - run : source/install/uv_with_retry.sh pip install --system --group pin_tensorflow_cpu --group pin_pytorch_cpu --torch-backend cpu
37- - run : |
38- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb \
39- && sudo dpkg -i cuda-keyring_1.0-1_all.deb \
40- && sudo apt-get update \
41- && sudo apt-get -y install cuda-cudart-dev-12-2 cuda-nvcc-12-2
42- if: matrix.variant == 'cuda120'
43- env:
44- DEBIAN_FRONTEND: noninteractive
45- - run : |
46- echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/6.3/ jammy main' | sudo tee /etc/apt/sources.list.d/rocm.list \
47- && printf 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600' | sudo tee /etc/apt/preferences.d/rocm-pin-600 \
48- && curl -s https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add - \
49- && sudo apt-get update \
50- && sudo apt-get install -y rocm-dev hipcub-dev
51- if: matrix.variant == 'rocm'
52- - run : |
53- source/install/build_cc.sh
54- env:
55- DP_VARIANT: ${{ matrix.dp_variant }}
56- DOWNLOAD_TENSORFLOW: "FALSE"
57- CMAKE_GENERATOR: Ninja
58- if: matrix.variant != 'clang'
59- - run : |
60- source/install/build_cc.sh
61- env:
62- DP_VARIANT: cpu
63- DOWNLOAD_TENSORFLOW: "FALSE"
64- CC: clang
65- CXX: clang++
66- CMAKE_GENERATOR: Ninja
67- if: matrix.variant == 'clang'
68- - name : Test files exist
69- run : |
70- test -f dp/bin/dp_ipi &&
71- test -f dp/lib/libdeepmd_cc.so &&
72- test -f dp/lib/libdeepmd_c.so &&
73- test -f dp/lib/libdeepmd_op.so &&
74- test -f dp/lib/libdeepmd_ipi.so &&
75- test -f dp/lib/libdeepmd_lmp.so &&
76- test -f dp/lib/libdeepmd.so
30+ - uses : actions/checkout@v6
31+ - uses : actions/setup-python@v6
32+ with :
33+ python-version : " 3.11"
34+ - uses : lukka/get-cmake@latest
35+ - run : python -m pip install uv
36+ - run : source/install/uv_with_retry.sh pip install --system --group pin_tensorflow_cpu --group pin_pytorch_cpu --torch-backend cpu
37+ - run : |
38+ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb \
39+ && sudo dpkg -i cuda-keyring_1.0-1_all.deb \
40+ && sudo apt-get update \
41+ && sudo apt-get -y install cuda-cudart-dev-12-2 cuda-nvcc-12-2
42+ if: matrix.variant == 'cuda120'
43+ env:
44+ DEBIAN_FRONTEND: noninteractive
45+ - run : |
46+ echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/6.3/ jammy main' | sudo tee /etc/apt/sources.list.d/rocm.list \
47+ && printf 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600' | sudo tee /etc/apt/preferences.d/rocm-pin-600 \
48+ && curl -s https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add - \
49+ && sudo apt-get update \
50+ && sudo apt-get install -y rocm-dev hipcub-dev
51+ if: matrix.variant == 'rocm'
52+ - run : |
53+ source/install/build_cc.sh
54+ env:
55+ DP_VARIANT: ${{ matrix.dp_variant }}
56+ DOWNLOAD_TENSORFLOW: "FALSE"
57+ CMAKE_GENERATOR: Ninja
58+ if: matrix.variant != 'clang'
59+ - run : |
60+ source/install/build_cc.sh
61+ env:
62+ DP_VARIANT: cpu
63+ DOWNLOAD_TENSORFLOW: "FALSE"
64+ CC: clang
65+ CXX: clang++
66+ CMAKE_GENERATOR: Ninja
67+ if: matrix.variant == 'clang'
68+ - name : Test files exist
69+ run : |
70+ test -f dp/bin/dp_ipi &&
71+ test -f dp/lib/libdeepmd_cc.so &&
72+ test -f dp/lib/libdeepmd_c.so &&
73+ test -f dp/lib/libdeepmd_op.so &&
74+ test -f dp/lib/libdeepmd_ipi.so &&
75+ test -f dp/lib/libdeepmd_lmp.so &&
76+ test -f dp/lib/libdeepmd.so
7777 pass :
7878 name : Pass building C++
7979 needs : [buildcc]
8080 runs-on : ubuntu-latest
8181 if : always()
8282 steps :
83- - name : Decide whether the needed jobs succeeded or failed
84- uses : re-actors/alls-green@release/v1
85- with :
86- jobs : ${{ toJSON(needs) }}
83+ - name : Decide whether the needed jobs succeeded or failed
84+ uses : re-actors/alls-green@release/v1
85+ with :
86+ jobs : ${{ toJSON(needs) }}
0 commit comments