CMakeLists.txt 2.93 KB
Newer Older
Paul's avatar
Paul committed
1
2
3
4

list(APPEND CMAKE_PREFIX_PATH /opt/rocm /opt/rocm/hip /opt/rocm/hcc)
find_package(miopen)

5
6
7
8
# rocblas
find_package(rocblas REQUIRED PATHS /opt/rocm)
message(STATUS "Build with rocblas")

Paul's avatar
Paul committed
9
10
11
12
if(NOT TARGET MIOpen)
    message(SEND_ERROR "Cant find miopen")
endif()

Paul's avatar
Paul committed
13
add_library(migraphx_device
14
    device/acos.cpp
Paul's avatar
Paul committed
15
    device/add.cpp
16
17
18
19
    device/add_clip.cpp
    device/add_relu.cpp
    device/add_sigmoid.cpp
    device/add_tanh.cpp
Shucai Xiao's avatar
Shucai Xiao committed
20
21
    device/argmax.cpp
    device/argmin.cpp
22
23
    device/asin.cpp
    device/atan.cpp
24
25
26
    device/ceil.cpp
    device/clip.cpp
    device/concat.cpp
27
    device/contiguous.cpp
28
    device/convert.cpp
29
30
31
32
33
34
    device/cos.cpp
    device/cosh.cpp
    device/div.cpp
    device/erf.cpp
    device/exp.cpp
    device/floor.cpp
35
    device/gather.cpp
36
    device/int8_gemm_pack.cpp
37
38
39
40
41
42
43
44
45
46
    device/log.cpp
    device/logsoftmax.cpp
    device/max.cpp
    device/min.cpp
    device/mul.cpp
    device/mul_add.cpp
    device/mul_add_relu.cpp
    device/pad.cpp
    device/pow.cpp
    device/reduce_max.cpp
47
    device/reduce_mean.cpp
Shucai Xiao's avatar
Shucai Xiao committed
48
    device/reduce_min.cpp
49
    device/reduce_sum.cpp
Shucai Xiao's avatar
Shucai Xiao committed
50
    device/reduce_prod.cpp
51
52
53
54
    device/relu.cpp
    device/round.cpp
    device/rsqrt.cpp
    device/sigmoid.cpp
Shucai Xiao's avatar
Shucai Xiao committed
55
    device/sign.cpp
56
57
58
59
60
61
62
63
    device/sin.cpp
    device/sinh.cpp
    device/softmax.cpp
    device/sqdiff.cpp
    device/sqrt.cpp
    device/sub.cpp
    device/tan.cpp
    device/tanh.cpp
64
)
Paul's avatar
Paul committed
65
set_target_properties(migraphx_device PROPERTIES EXPORT_NAME device)
Paul's avatar
Paul committed
66
rocm_set_soversion(migraphx_device ${MIGRAPHX_SO_VERSION})
Paul's avatar
Paul committed
67
rocm_clang_tidy_check(migraphx_device)
68
target_compile_options(migraphx_device PRIVATE -std=c++17)
Paul's avatar
Paul committed
69
target_link_libraries(migraphx_device migraphx hip::device -Wno-invalid-command-line-argument -amdgpu-target=gfx803 -amdgpu-target=gfx900 -amdgpu-target=gfx906)
Paul's avatar
Paul committed
70
71
target_include_directories(migraphx_device PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>)
target_include_directories(migraphx_device PRIVATE $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/device/include>)
72

Paul's avatar
Paul committed
73
add_library(migraphx_gpu
Shucai Xiao's avatar
Shucai Xiao committed
74
75
    argmax.cpp
    argmin.cpp
76
    eliminate_workspace.cpp
Paul's avatar
Paul committed
77
    fuse_ops.cpp
Paul's avatar
Paul committed
78
    hip.cpp
Paul's avatar
Paul committed
79
    target.cpp
Paul's avatar
Paul committed
80
    lowering.cpp
wsttiger's avatar
wsttiger committed
81
82
    pooling.cpp
    convolution.cpp
83
    quant_convolution.cpp
84
    softmax.cpp
85
    logsoftmax.cpp
86
    contiguous.cpp
87
    concat.cpp
Khalique's avatar
Khalique committed
88
    leaky_relu.cpp
89
    batchnorm.cpp
Paul's avatar
Paul committed
90
    write_literals.cpp
91
    rocblas.cpp
Khalique's avatar
Khalique committed
92
    abs.cpp
Khalique's avatar
Khalique committed
93
    elu.cpp
94
    pad.cpp
95
    gather.cpp
Shucai Xiao's avatar
Shucai Xiao committed
96
    convert.cpp
Khalique's avatar
Khalique committed
97
    lrn.cpp
Paul's avatar
Paul committed
98
    schedule_model.cpp
99
    adjust_allocation.cpp
100
    pack_int8_args.cpp
Khalique's avatar
Khalique committed
101
    clip.cpp
102
103
    int8_gemm_pack.cpp
    int8_conv_pack.cpp
104
    gemm_impl.cpp
105
    preallocate_param.cpp
Paul's avatar
Paul committed
106
)
Paul's avatar
Paul committed
107
set_target_properties(migraphx_gpu PROPERTIES EXPORT_NAME gpu)
Paul's avatar
Paul committed
108
rocm_set_soversion(migraphx_gpu ${MIGRAPHX_SO_VERSION})
Paul's avatar
Paul committed
109
110
111
rocm_clang_tidy_check(migraphx_gpu)
target_link_libraries(migraphx_gpu PUBLIC migraphx MIOpen roc::rocblas)
target_link_libraries(migraphx_gpu PRIVATE migraphx_device)
mei-ye's avatar
mei-ye committed
112

Paul's avatar
Paul committed
113
rocm_install_targets(
Paul's avatar
Paul committed
114
  TARGETS migraphx_gpu migraphx_device
Paul's avatar
Paul committed
115
116
117
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
mei-ye's avatar
mei-ye committed
118