abacus-develop/source/source_base/module_device/CMakeLists.txt at f27e198a674dbed42b25d74e927610dc7a46c012 · 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
# list(APPEND module_device_srcs
#   memory_op.cpp
#   device.cpp
# )

# if(USE_CUDA)
#   list(APPEND module_device_srcs
#     cuda/memory_op.cu
#   )
# endif()


# if(USE_ROCM)
#   hip_add_library(module_device_rocm STATIC
#     rocm/memory_op.hip.cu
#   )
#   target_link_libraries(
#     device
#     module_device_rocm
#     hip::host
#     hip::device
#     hip::hipfft
#     roc::hipblas
#     roc::hipsolver
#   )
# endif()

# add_library(device OBJECT ${device_srcs})

# if(USE_CUDA)
#   target_link_libraries(
#     device
#   )
# elseif(USE_ROCM)
#   target_link_libraries(
#     device
#     device_rocm
#     hip::host
#     hip::device
#     hip::hipfft
#     roc::hipblas
#     roc::hipsolver
#   )
# endif()