abacus-develop/source/source_hsolver/CMakeLists.txt at f0a49e316b32fa8a204206390a76bc08455f3225 · cheerly-pku/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
list(APPEND objects
    diag_const_nums.cpp
    diago_cg.cpp
    diago_david.cpp
    diago_dav_subspace.cpp
    diago_bpcg.cpp
    diago_ppcg.cpp
    para_linear_transform.cpp
    hsolver_pw.cpp
    hsolver_lcaopw.cpp
    hsolver_pw_sdft.cpp
    diago_iter_assist.cpp
    hsolver.cpp
    diago_pxxxgvx.cpp
    diag_hs_para.cpp
    diago_params.cpp

)

if(ENABLE_LCAO)
  if (ENABLE_MPI)
    list(APPEND objects
        hsolver_lcao.cpp
        diago_scalapack.cpp
        parallel_k2d.cpp
        diago_lapack.cpp
    )
  else ()
    list(APPEND objects
        hsolver_lcao.cpp
        diago_lapack.cpp
    )
  endif ()
  if (ENABLE_ELPA)
    list(APPEND objects
        diago_elpa.cpp
        diago_elpa_native.cpp
    )
  endif ()

  if(USE_CUDA)
      set(cuda_objects
          ./kernels/hegvd_op.cpp
          ./kernels/cuda/diag_cusolver.cu
          diago_cusolver.cpp
          diago_cusolver.h
      )
      if(ENABLE_CUSOLVERMP)
        list(APPEND cuda_objects
            ./kernels/cuda/diag_cusolvermp.cu
            diago_cusolvermp.cpp
            diago_cusolvermp.h
        )
      endif()
    add_library(
        diag_cusolver
        OBJECT
        ${cuda_objects}
    )
    if(ENABLE_COVERAGE)
      add_coverage(diag_cusolver)
    endif()
  endif()

  if(ENABLE_PEXSI)
  list(APPEND objects
      diago_pexsi.cpp
    )
    add_subdirectory(module_pexsi)
  endif()

endif()

add_library(
    hsolver
    OBJECT
    ${objects}
)

if(ENABLE_COVERAGE)
  add_coverage(hsolver)
endif()

if(ENABLE_LCAO AND ENABLE_ELPA)
  add_subdirectory(module_genelpa)
endif()

if(BUILD_TESTING)
  add_subdirectory(test)
  if(ENABLE_MPI)
    add_subdirectory(kernels/test)
  endif()
  message(STATUS "Building tests")
endif()