abacus-develop/.github/workflows/cuda.yml at develop · deepmodeling/abacus-develop · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
name: CUDA Test

on:
  workflow_dispatch:
  pull_request:

defaults:
  run:
    shell: bash

concurrency:
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: true

jobs:
  test:
    name: Test on CUDA Build
    runs-on: gpu
    if: github.repository_owner == 'deepmodeling'
    container:
      image: ghcr.io/deepmodeling/abacus-cuda
      volumes:
        - /tmp/ccache:/github/home/.ccache
      options: --gpus all

    steps:
      - name: Checkout
        uses: actions/checkout@v7.0.1
        with:
          submodules: recursive

      - name: Install CI tools
        run: |
          sudo apt-get update
          sudo apt-get install -y ccache xz-utils ninja-build pkg-config

      - name: Install external tools from toolchain
        run: |
          cd toolchain
          ./install_abacus_toolchain_new.sh --with-dftd4=install --dry-run -j8
          ./scripts/stage4/install_stage4.sh
          cd ..

      - name: Configure & Build
        run: |
          nvidia-smi
          source toolchain/install/setup
          rm -rf build
          cmake -B build -G Ninja -DUSE_CUDA=ON -DBUILD_TESTING=ON -DENABLE_FLOAT_FFTW=ON \
            -DCMAKE_CUDA_ARCHITECTURES=70
          cmake --build build -j "$(nproc)"
          cmake --install build

      - name: Module_LCAO CUDA Unittests
        env:
          GTEST_COLOR: 'yes'
          OMP_NUM_THREADS: '2'
        run: |
          ctest --test-dir build -V --timeout 1700 -R '^(MODULE_LCAO_tddft_radial_interpolation_cuda_test|MODULE_LCAO_tddft_snap_psibeta_half_test)$'

      - name: Test 11_PW_GPU
        run: |
          cd tests/11_PW_GPU
          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt

      - name: Test 12_NAO_Gamma_GPU
        run: |
          cd tests/12_NAO_Gamma_GPU
          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt

      - name: Test 13_NAO_multik_GPU
        run: |
          cd tests/13_NAO_multik_GPU
          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt

      - name: Test 15_rtTDDFT_GPU
        run: |
          cd tests/15_rtTDDFT_GPU
          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt

      - name: Test 16_SDFT_GPU
        run: |
          cd tests/16_SDFT_GPU
          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt

      - name: Test 01_PW on GPU
        run: |
          cd tests/01_PW
          find . -name INPUT | while read f; do
              if grep -q "^device" "$f"; then
                  sed -i 's/^device.*/device    gpu/' "$f"
              else
                  echo "device    gpu" >> "$f"
              fi
          done
          bash ../integrate/Autotest.sh -n 1 -a abacus -f CASES_GPU.txt