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

249
250
251
if(WIN32)
    target_compile_definitions(migraphx PUBLIC _CRT_SECURE_NO_WARNINGS _USE_MATH_DEFINES)
else()
Artur Wojcik's avatar
Artur Wojcik committed
252
253
254
255
256
    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
257
258
endif()

Paul's avatar
Paul committed
259
target_include_directories(migraphx SYSTEM PUBLIC $<BUILD_INTERFACE:${HALF_INCLUDE_DIR}>)
260
261
target_link_libraries(migraphx PUBLIC Threads::Threads)

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

Artur Wojcik's avatar
Artur Wojcik committed
268
if(NOT WIN32)
269
    find_package(SQLite3 REQUIRED)
Artur Wojcik's avatar
Artur Wojcik committed
270
endif()
271
target_link_libraries(migraphx PRIVATE SQLite::SQLite3)
272

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

283
284
add_library(migraphx_all_targets INTERFACE)

Paul Fultz II's avatar
Paul Fultz II committed
285
add_subdirectory(api)
Paul's avatar
Paul committed
286
add_subdirectory(driver)
Paul's avatar
Paul committed
287
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
288
add_subdirectory(tf)
Khalique's avatar
Khalique committed
289

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

turneram's avatar
turneram committed
312
313
314
315
if(HAVE_HALF_EXPR)
    target_compile_definitions(migraphx PUBLIC -DHAS_HALF_V1)
endif()

316
317
318
319
if(BUILD_DEV)
    target_compile_definitions(migraphx PUBLIC -DBUILD_DEV)
endif()

Paul's avatar
Paul committed
320
rocm_export_targets(
321
  TARGETS migraphx::migraphx_c
Paul's avatar
Paul committed
322
  NAMESPACE migraphx::
Paul's avatar
Paul committed
323
  DEPENDS
324
    Threads
Paul's avatar
Paul committed
325
326
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
327
328