CMakeLists.txt 6.84 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#####################################################################################
# The MIT License (MIT)
#
# Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#####################################################################################
Paul's avatar
Paul committed
24

Paul's avatar
Paul committed
25
26
include(ROCMInstallTargets)
include(ROCMPackageConfigHelpers)
27
include(RegisterOp)
Paul Fultz II's avatar
Paul Fultz II committed
28
include(CheckCXXLinkerFlag)
Paul's avatar
Paul committed
29

Paul's avatar
Paul committed
30
add_library(migraphx 
31
    adjust_allocation.cpp
32
    analyze_streams.cpp
33
    apply_alpha_beta.cpp
Paul Fultz II's avatar
Paul Fultz II committed
34
    argument.cpp
Paul's avatar
Paul committed
35
    auto_contiguous.cpp
36
    common.cpp
37
    compile_src.cpp
38
    convert_to_json.cpp
39
    cpp_generator.cpp
Paul's avatar
Paul committed
40
    dead_code_elimination.cpp
Paul Fultz II's avatar
Paul Fultz II committed
41
    dom_info.cpp
42
    dynamic_loader.cpp
Paul's avatar
Paul committed
43
    eliminate_allocation.cpp
44
    eliminate_common_subexpression.cpp
45
    eliminate_concat.cpp
46
    eliminate_contiguous.cpp
47
    eliminate_data_type.cpp
48
    eliminate_identity.cpp
49
    eliminate_pad.cpp
Paul's avatar
Paul committed
50
    env.cpp
51
    file_buffer.cpp
52
    fuse_pointwise.cpp
Paul's avatar
Paul committed
53
    generate.cpp
Shucai Xiao's avatar
Shucai Xiao committed
54
    inline_module.cpp
55
    insert_pad.cpp
Paul's avatar
Paul committed
56
    instruction.cpp
57
    json.cpp
58
    load_save.cpp
59
    make_op.cpp
60
    module.cpp
61
    msgpack.cpp
62
63
    normalize_attributes.cpp
    normalize_ops.cpp
64
    op_enums.cpp
65
    operation.cpp
66
67
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
68
    pad_calc.cpp
69
    pass_manager.cpp
70
    permutation.cpp
71
    preallocate_param.cpp
72
    process.cpp
Paul's avatar
Paul committed
73
    program.cpp
74
    propagate_constant.cpp
Shucai Xiao's avatar
Shucai Xiao committed
75
    quantization.cpp
76
77
    quantize_fp16.cpp
    quantize_int8.cpp
78
    reduce_dims.cpp
79
80
    register_op.cpp
    register_target.cpp
81
    replace_allocate.cpp
turneram's avatar
turneram committed
82
    simplify_qdq.cpp
83
    sqlite.cpp
84
85
    rewrite_batchnorm.cpp
    rewrite_pooling.cpp
turneram's avatar
turneram committed
86
    rewrite_quantization.cpp
87
    rewrite_rnn.cpp
Paul's avatar
Paul committed
88
    schedule.cpp
89
    serialize.cpp
90
    shape.cpp
Paul's avatar
Paul committed
91
    simplify_algebra.cpp
Paul's avatar
Paul committed
92
    simplify_reshapes.cpp
93
    target_assignments.cpp
Paul Fultz II's avatar
Paul Fultz II committed
94
    tmp_dir.cpp
95
    value.cpp
96
    verify_args.cpp
Paul's avatar
Paul committed
97
)
98
configure_file(version.h.in include/migraphx/version.h)
Paul's avatar
Paul committed
99
rocm_set_soversion(migraphx ${MIGRAPHX_SO_VERSION})
100
101
102
103
104
105
106
107
108
109
function(register_migraphx_ops)
    foreach(OP ${ARGN})
        register_op(migraphx HEADER migraphx/op/${OP}.hpp OPERATORS op::${OP})
    endforeach()
endfunction()
register_migraphx_ops(
    abs
    acosh
    acos
    add
110
    allocate
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
    argmax
    argmin
    asinh
    asin
    as_shape
    atanh
    atan
    batch_norm_inference
    broadcast
    capture
    ceil
    clip
    concat
    contiguous
    convert
    convolution
    cosh
    cos
    deconvolution
turneram's avatar
turneram committed
130
    dequantizelinear
131
132
133
    div
    dot
    elu
134
    equal
135
136
137
138
    erf
    exp
    flatten
    floor
139
    fmod
140
    gather
turneram's avatar
turneram committed
141
    gathernd
Shucai Xiao's avatar
Shucai Xiao committed
142
    get_tuple_elem
143
    greater
144
145
    gru
    identity
Shucai Xiao's avatar
Shucai Xiao committed
146
    if_op
147
    im2col
Charlie Lin's avatar
Charlie Lin committed
148
    isnan
149
    leaky_relu
150
    less
151
152
    load
    log
Shucai Xiao's avatar
Shucai Xiao committed
153
154
155
    logical_and
    logical_or
    logical_xor
156
    logsoftmax
Shucai Xiao's avatar
Shucai Xiao committed
157
    loop
158
159
160
161
    lrn
    lstm
    max
    min
162
    mod
163
164
    mul
    multibroadcast
turneram's avatar
turneram committed
165
    multinomial
166
    neg
167
    nonmaxsuppression
Shucai Xiao's avatar
Shucai Xiao committed
168
    nonzero
169
170
    outline
    pad
171
    pointwise
172
173
    pooling
    pow
turneram's avatar
turneram committed
174
    prefix_scan_sum
175
176
177
    prelu
    quant_convolution
    quant_dot
turneram's avatar
turneram committed
178
    quantizelinear
179
180
181
182
183
184
185
186
    recip
    reduce_max
    reduce_mean
    reduce_min
    reduce_prod
    reduce_sum
    relu
    reshape
Cagri Eryilmaz's avatar
Cagri Eryilmaz committed
187
    reverse
188
189
190
191
    rnn
    rnn_last_cell_output
    rnn_last_hs_output
    rnn_var_sl_last_output
Shucai Xiao's avatar
Shucai Xiao committed
192
    roialign
193
194
195
    round
    rsqrt
    scalar
196
197
198
    scatter_add
    scatter_mul
    scatter_none
turneram's avatar
turneram committed
199
200
    scatternd_add
    scatternd_mul
201
    scatternd_none
202
203
204
205
206
207
208
209
210
    sigmoid
    sign
    sinh
    sin
    slice
    softmax
    sqdiff
    sqrt
    squeeze
Shucai Xiao's avatar
Shucai Xiao committed
211
    step
212
213
214
    sub
    tanh
    tan
Shucai Xiao's avatar
Shucai Xiao committed
215
    topk
216
    transpose
217
    unary_not
218
219
220
    undefined
    unknown
    unsqueeze
turneram's avatar
turneram committed
221
    where
222
223
224
)
register_op(migraphx HEADER migraphx/op/rnn_variable_seq_lens.hpp OPERATORS op::rnn_var_sl_shift_output op::rnn_var_sl_shift_sequence)
register_op(migraphx HEADER migraphx/builtin.hpp OPERATORS builtin::literal builtin::param builtin::returns)
Paul's avatar
Paul committed
225
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
226
rocm_install_targets(
Paul's avatar
Paul committed
227
  TARGETS migraphx
Paul's avatar
Paul committed
228
229
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
230
    ${CMAKE_CURRENT_BINARY_DIR}/include
Paul's avatar
Paul committed
231
)
Paul Fultz II's avatar
Paul Fultz II committed
232
233
234
235
236
237
238


check_cxx_linker_flag(-lstdc++fs HAS_LIB_STD_FILESYSTEM)
if(HAS_LIB_STD_FILESYSTEM)
target_link_libraries(migraphx PRIVATE -lstdc++fs)
endif()

239
240
target_link_libraries(migraphx PRIVATE -ldl)

Paul's avatar
Paul committed
241
target_include_directories(migraphx SYSTEM PUBLIC $<BUILD_INTERFACE:${HALF_INCLUDE_DIR}>)
Paul's avatar
Paul committed
242

243
244
245
find_package(Threads)
target_link_libraries(migraphx PUBLIC Threads::Threads)

246
247
248
249
250
251
252
find_package(nlohmann_json 3.8.0 REQUIRED)
target_link_libraries(migraphx PRIVATE nlohmann_json::nlohmann_json)

find_package(PkgConfig)
pkg_check_modules(SQLITE3 REQUIRED IMPORTED_TARGET sqlite3)
target_link_libraries(migraphx PRIVATE PkgConfig::SQLITE3)

253
254
255
256
257
find_package(msgpack REQUIRED)
target_link_libraries(migraphx PRIVATE msgpackc-cxx)
# Make this available to the tests
target_link_libraries(migraphx INTERFACE $<BUILD_INTERFACE:msgpackc-cxx>)

258
259
add_library(migraphx_all_targets INTERFACE)

Paul's avatar
Paul committed
260
set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
261

Paul Fultz II's avatar
Paul Fultz II committed
262
add_subdirectory(api)
Paul's avatar
Paul committed
263
add_subdirectory(driver)
Paul's avatar
Paul committed
264
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
265
add_subdirectory(tf)
Khalique's avatar
Khalique committed
266

Paul's avatar
Paul committed
267
add_subdirectory(py)
268
add_subdirectory(targets/ref)
269
target_link_libraries(migraphx_all_targets INTERFACE migraphx_ref)
270
271
272
273
274
if(MIGRAPHX_ENABLE_CPU)
add_subdirectory(targets/cpu)
target_link_libraries(migraphx_all_targets INTERFACE migraphx_cpu)
target_compile_definitions(migraphx_all_targets INTERFACE -DHAVE_CPU)
endif()
Paul's avatar
Paul committed
275
if(MIGRAPHX_ENABLE_GPU)
Paul Fultz II's avatar
Paul Fultz II committed
276
list(APPEND PACKAGE_DEPENDS PACKAGE MIOpen PACKAGE rocblas)
Paul's avatar
Paul committed
277
add_subdirectory(targets/gpu)
278
279
target_link_libraries(migraphx_all_targets INTERFACE migraphx_gpu)
target_compile_definitions(migraphx_all_targets INTERFACE -DHAVE_GPU)
Paul's avatar
Paul committed
280
endif()
varunsh's avatar
varunsh committed
281
282
283
284
285
if(MIGRAPHX_ENABLE_FPGA)
add_subdirectory(targets/fpga)
target_link_libraries(migraphx_all_targets INTERFACE migraphx_fpga)
target_compile_definitions(migraphx_all_targets INTERFACE -DHAVE_FPGA)
endif()
mei-ye's avatar
mei-ye committed
286

turneram's avatar
turneram committed
287
288
289
290
if(HAVE_HALF_EXPR)
    target_compile_definitions(migraphx PUBLIC -DHAS_HALF_V1)
endif()

Paul's avatar
Paul committed
291
rocm_export_targets(
292
  TARGETS migraphx::migraphx_c
Paul's avatar
Paul committed
293
  NAMESPACE migraphx::
Paul's avatar
Paul committed
294
  DEPENDS
295
    Threads
Paul's avatar
Paul committed
296
297
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
298
299