CMakeLists.txt 7.66 KB
Newer Older
1
2
3
#####################################################################################
# The MIT License (MIT)
#
4
# Copyright (c) 2015-2023 Advanced Micro Devices, Inc. All rights reserved.
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#
# 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

25
26
include(ExportHeader)

Paul's avatar
Paul committed
27
28
include(ROCMInstallTargets)
include(ROCMPackageConfigHelpers)
29
include(RegisterOp)
Paul Fultz II's avatar
Paul Fultz II committed
30
include(CheckCXXLinkerFlag)
31
 
Paul's avatar
Paul committed
32

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

Artur Wojcik's avatar
Artur Wojcik committed
254
255
256
257
258
259
if(NOT WIN32)
    check_cxx_linker_flag(-lstdc++fs HAS_LIB_STD_FILESYSTEM)
    if(HAS_LIB_STD_FILESYSTEM)
        target_link_libraries(migraphx PRIVATE -lstdc++fs)
    endif()
    target_link_libraries(migraphx PRIVATE -ldl)
Paul Fultz II's avatar
Paul Fultz II committed
260
261
endif()

Paul's avatar
Paul committed
262
target_include_directories(migraphx SYSTEM PUBLIC $<BUILD_INTERFACE:${HALF_INCLUDE_DIR}>)
263
264
target_link_libraries(migraphx PUBLIC Threads::Threads)

Artur Wojcik's avatar
Artur Wojcik committed
265
266
267
if(NOT WIN32)
    find_package(nlohmann_json 3.8.0 REQUIRED)
endif()
268
target_link_libraries(migraphx PRIVATE nlohmann_json::nlohmann_json)
269
migraphx_generate_export_header(migraphx)
270

Artur Wojcik's avatar
Artur Wojcik committed
271
272
273
274
if(NOT WIN32)
    find_package(PkgConfig)
    pkg_check_modules(SQLITE3 REQUIRED IMPORTED_TARGET sqlite3)
endif()
275
276
target_link_libraries(migraphx PRIVATE PkgConfig::SQLITE3)

Artur Wojcik's avatar
Artur Wojcik committed
277
278
279
280
281
if(NOT WIN32)
    find_package(msgpackc-cxx QUIET)
    if(NOT msgpackc-cxx_FOUND)
        find_package(msgpack REQUIRED)
    endif()
282
endif()
283
284
285
286
target_link_libraries(migraphx PRIVATE msgpackc-cxx)
# Make this available to the tests
target_link_libraries(migraphx INTERFACE $<BUILD_INTERFACE:msgpackc-cxx>)

287
288
add_library(migraphx_all_targets INTERFACE)

289
290
291
include(CMakeDependentOption)
cmake_dependent_option(MIGRAPHX_ENABLE_PYTHON "Enable python bindings" ON "WIN32" OFF)

Paul Fultz II's avatar
Paul Fultz II committed
292
add_subdirectory(api)
Paul's avatar
Paul committed
293
add_subdirectory(driver)
Paul's avatar
Paul committed
294
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
295
add_subdirectory(tf)
Khalique's avatar
Khalique committed
296

Paul's avatar
Paul committed
297
add_subdirectory(py)
298
add_subdirectory(targets/ref)
299
target_link_libraries(migraphx_all_targets INTERFACE migraphx_ref)
300
301
302
303
304
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
305
if(MIGRAPHX_ENABLE_GPU)
Paul Fultz II's avatar
Paul Fultz II committed
306
list(APPEND PACKAGE_DEPENDS PACKAGE MIOpen PACKAGE rocblas)
Paul's avatar
Paul committed
307
add_subdirectory(targets/gpu)
308
309
target_link_libraries(migraphx_all_targets INTERFACE migraphx_gpu)
target_compile_definitions(migraphx_all_targets INTERFACE -DHAVE_GPU)
Paul's avatar
Paul committed
310
endif()
varunsh's avatar
varunsh committed
311
312
313
314
315
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
316

turneram's avatar
turneram committed
317
318
319
320
if(HAVE_HALF_EXPR)
    target_compile_definitions(migraphx PUBLIC -DHAS_HALF_V1)
endif()

321
322
323
324
if(BUILD_DEV)
    target_compile_definitions(migraphx PUBLIC -DBUILD_DEV)
endif()

Paul's avatar
Paul committed
325
rocm_export_targets(
326
  TARGETS migraphx::migraphx_c
Paul's avatar
Paul committed
327
  NAMESPACE migraphx::
Paul's avatar
Paul committed
328
  DEPENDS
329
    Threads
Paul's avatar
Paul committed
330
331
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
332
333