CMakeLists.txt 7.1 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
Khalique Ahmed's avatar
testing  
Khalique Ahmed committed
49
    eliminate_layout.cpp
50
    eliminate_pad.cpp
Paul's avatar
Paul committed
51
    env.cpp
52
    file_buffer.cpp
53
    fuse_pointwise.cpp
Paul Fultz II's avatar
Paul Fultz II committed
54
    fuse_reduce.cpp
Paul's avatar
Paul committed
55
    generate.cpp
Shucai Xiao's avatar
Shucai Xiao committed
56
    inline_module.cpp
57
    insert_pad.cpp
Paul's avatar
Paul committed
58
    instruction.cpp
59
    json.cpp
60
    layout_nhwc.cpp
61
    load_save.cpp
62
    make_op.cpp
63
    memory_coloring.cpp
64
    module.cpp
65
    msgpack.cpp
66
67
    normalize_attributes.cpp
    normalize_ops.cpp
68
    op_enums.cpp
69
    operation.cpp
70
    optimize_module.cpp
71
    pad_calc.cpp
72
    pass_manager.cpp
73
    permutation.cpp
74
    preallocate_param.cpp
75
    process.cpp
Paul's avatar
Paul committed
76
    program.cpp
77
    propagate_constant.cpp
Charlie Lin's avatar
Charlie Lin committed
78
    promote_literals.cpp
Shucai Xiao's avatar
Shucai Xiao committed
79
    quantization.cpp
80
81
    quantize_fp16.cpp
    quantize_int8.cpp
82
    reduce_dims.cpp
83
84
    register_op.cpp
    register_target.cpp
85
    replace_allocate.cpp
turneram's avatar
turneram committed
86
    simplify_qdq.cpp
87
    sqlite.cpp
88
    rewrite_gelu.cpp
89
    rewrite_pooling.cpp
turneram's avatar
turneram committed
90
    rewrite_quantization.cpp
91
    rewrite_rnn.cpp
Paul's avatar
Paul committed
92
    schedule.cpp
93
    serialize.cpp
94
    shape.cpp
Paul's avatar
Paul committed
95
    simplify_algebra.cpp
Paul's avatar
Paul committed
96
    simplify_reshapes.cpp
97
    split_single_dyn_dim.cpp
98
    target.cpp
Paul Fultz II's avatar
Paul Fultz II committed
99
    tmp_dir.cpp
100
    value.cpp
101
    verify_args.cpp
Paul's avatar
Paul committed
102
)
103
configure_file(version.h.in include/migraphx/version.h)
Paul's avatar
Paul committed
104
rocm_set_soversion(migraphx ${MIGRAPHX_SO_VERSION})
105
106
107
108
109
110
111
112
113
114
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
115
    allocate
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
    argmax
    argmin
    asinh
    asin
    as_shape
    atanh
    atan
    broadcast
    capture
    ceil
    clip
    concat
    contiguous
    convert
    convolution
    cosh
    cos
    deconvolution
turneram's avatar
turneram committed
134
    dequantizelinear
135
136
137
    div
    dot
    elu
138
    equal
139
140
141
142
    erf
    exp
    flatten
    floor
143
    fmod
144
    gather
turneram's avatar
turneram committed
145
    gathernd
Shucai Xiao's avatar
Shucai Xiao committed
146
    get_tuple_elem
147
    greater
148
149
    gru
    identity
Shucai Xiao's avatar
Shucai Xiao committed
150
    if_op
151
    im2col
Charlie Lin's avatar
Charlie Lin committed
152
    isnan
153
    layout
154
    leaky_relu
155
    less
156
157
    load
    log
Shucai Xiao's avatar
Shucai Xiao committed
158
159
160
    logical_and
    logical_or
    logical_xor
161
    logsoftmax
Shucai Xiao's avatar
Shucai Xiao committed
162
    loop
163
164
165
166
    lrn
    lstm
    max
    min
167
    mod
168
169
    mul
    multibroadcast
turneram's avatar
turneram committed
170
    multinomial
171
    neg
172
    nonmaxsuppression
Shucai Xiao's avatar
Shucai Xiao committed
173
    nonzero
174
175
    outline
    pad
176
    pointwise
177
178
    pooling
    pow
turneram's avatar
turneram committed
179
    prefix_scan_sum
180
181
182
    prelu
    quant_convolution
    quant_dot
turneram's avatar
turneram committed
183
    quantizelinear
184
185
186
187
188
189
190
191
    recip
    reduce_max
    reduce_mean
    reduce_min
    reduce_prod
    reduce_sum
    relu
    reshape
Cagri Eryilmaz's avatar
Cagri Eryilmaz committed
192
    reverse
193
194
195
196
    rnn
    rnn_last_cell_output
    rnn_last_hs_output
    rnn_var_sl_last_output
Shucai Xiao's avatar
Shucai Xiao committed
197
    roialign
198
199
    round
    rsqrt
200
    run_on_target
201
    scalar
202
203
204
    scatter_add
    scatter_mul
    scatter_none
turneram's avatar
turneram committed
205
206
    scatternd_add
    scatternd_mul
207
    scatternd_none
Charlie Lin's avatar
Charlie Lin committed
208
    select_module
209
210
211
212
213
214
215
216
217
    sigmoid
    sign
    sinh
    sin
    slice
    softmax
    sqdiff
    sqrt
    squeeze
Shucai Xiao's avatar
Shucai Xiao committed
218
    step
219
220
221
    sub
    tanh
    tan
Shucai Xiao's avatar
Shucai Xiao committed
222
    topk
223
    transpose
224
    unary_not
225
226
227
    undefined
    unknown
    unsqueeze
turneram's avatar
turneram committed
228
    where
229
230
231
)
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
232
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
233
rocm_install_targets(
Paul's avatar
Paul committed
234
  TARGETS migraphx
Paul's avatar
Paul committed
235
236
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
237
    ${CMAKE_CURRENT_BINARY_DIR}/include
Paul's avatar
Paul committed
238
)
Paul Fultz II's avatar
Paul Fultz II committed
239
240
241
242
243
244
245


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

246
247
target_link_libraries(migraphx PRIVATE -ldl)

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

250
251
252
find_package(Threads)
target_link_libraries(migraphx PUBLIC Threads::Threads)

253
254
255
256
257
258
259
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)

260
find_package(msgpackc-cxx QUIET)
261
262
263
if(NOT msgpackc-cxx_FOUND)
    find_package(msgpack REQUIRED)
endif()
264
265
266
267
target_link_libraries(migraphx PRIVATE msgpackc-cxx)
# Make this available to the tests
target_link_libraries(migraphx INTERFACE $<BUILD_INTERFACE:msgpackc-cxx>)

268
269
add_library(migraphx_all_targets INTERFACE)

Paul's avatar
Paul committed
270
set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
271

Paul Fultz II's avatar
Paul Fultz II committed
272
add_subdirectory(api)
Paul's avatar
Paul committed
273
add_subdirectory(driver)
Paul's avatar
Paul committed
274
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
275
add_subdirectory(tf)
Khalique's avatar
Khalique committed
276

Paul's avatar
Paul committed
277
add_subdirectory(py)
278
add_subdirectory(targets/ref)
279
target_link_libraries(migraphx_all_targets INTERFACE migraphx_ref)
280
281
282
283
284
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
285
if(MIGRAPHX_ENABLE_GPU)
Paul Fultz II's avatar
Paul Fultz II committed
286
list(APPEND PACKAGE_DEPENDS PACKAGE MIOpen PACKAGE rocblas)
Paul's avatar
Paul committed
287
add_subdirectory(targets/gpu)
288
289
target_link_libraries(migraphx_all_targets INTERFACE migraphx_gpu)
target_compile_definitions(migraphx_all_targets INTERFACE -DHAVE_GPU)
Paul's avatar
Paul committed
290
endif()
varunsh's avatar
varunsh committed
291
292
293
294
295
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
296

turneram's avatar
turneram committed
297
298
299
300
if(HAVE_HALF_EXPR)
    target_compile_definitions(migraphx PUBLIC -DHAS_HALF_V1)
endif()

301
302
303
304
if(BUILD_DEV)
    target_compile_definitions(migraphx PUBLIC -DBUILD_DEV)
endif()

Paul's avatar
Paul committed
305
rocm_export_targets(
306
  TARGETS migraphx::migraphx_c
Paul's avatar
Paul committed
307
  NAMESPACE migraphx::
Paul's avatar
Paul committed
308
  DEPENDS
309
    Threads
Paul's avatar
Paul committed
310
311
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
312
313