CMakeLists.txt 5.01 KB
Newer Older
Paul's avatar
Paul committed
1

Paul's avatar
Paul committed
2
3
include(ROCMInstallTargets)
include(ROCMPackageConfigHelpers)
4
include(RegisterOp)
Paul Fultz II's avatar
Paul Fultz II committed
5
include(CheckCXXLinkerFlag)
Paul's avatar
Paul committed
6

Paul's avatar
Paul committed
7
add_library(migraphx 
8
    adjust_allocation.cpp
9
    analyze_streams.cpp
10
    apply_alpha_beta.cpp
Paul Fultz II's avatar
Paul Fultz II committed
11
    argument.cpp
Paul's avatar
Paul committed
12
    auto_contiguous.cpp
13
    common.cpp
14
    compile_src.cpp
15
    convert_to_json.cpp
16
    cpp_generator.cpp
Paul's avatar
Paul committed
17
    dead_code_elimination.cpp
Paul Fultz II's avatar
Paul Fultz II committed
18
    dom_info.cpp
19
    dynamic_loader.cpp
Paul's avatar
Paul committed
20
    eliminate_allocation.cpp
21
    eliminate_common_subexpression.cpp
22
    eliminate_concat.cpp
23
    eliminate_contiguous.cpp
24
    eliminate_data_type.cpp
25
    eliminate_identity.cpp
26
    eliminate_pad.cpp
Paul's avatar
Paul committed
27
    env.cpp
28
    file_buffer.cpp
29
    fuse_pointwise.cpp
Paul's avatar
Paul committed
30
    generate.cpp
Shucai Xiao's avatar
Shucai Xiao committed
31
    inline_module.cpp
32
    insert_pad.cpp
Paul's avatar
Paul committed
33
    instruction.cpp
34
    json.cpp
Paul's avatar
Paul committed
35
    layout_nhwc.cpp
36
    load_save.cpp
37
    make_op.cpp
38
    module.cpp
39
    msgpack.cpp
40
41
    normalize_attributes.cpp
    normalize_ops.cpp
42
    op_enums.cpp
43
    operation.cpp
44
45
46
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
    pass_manager.cpp
47
    permutation.cpp
48
    preallocate_param.cpp
49
    process.cpp
Paul's avatar
Paul committed
50
    program.cpp
51
    propagate_constant.cpp
Shucai Xiao's avatar
Shucai Xiao committed
52
    quantization.cpp
53
54
    quantize_fp16.cpp
    quantize_int8.cpp
55
    reduce_dims.cpp
56
57
    register_op.cpp
    register_target.cpp
turneram's avatar
turneram committed
58
    simplify_qdq.cpp
59
60
    rewrite_batchnorm.cpp
    rewrite_pooling.cpp
turneram's avatar
turneram committed
61
    rewrite_quantization.cpp
62
    rewrite_rnn.cpp
Paul's avatar
Paul committed
63
    schedule.cpp
64
    serialize.cpp
65
    shape.cpp
Paul's avatar
Paul committed
66
    simplify_algebra.cpp
Paul's avatar
Paul committed
67
    simplify_reshapes.cpp
Paul Fultz II's avatar
Paul Fultz II committed
68
    tmp_dir.cpp
69
    value.cpp
70
    verify_args.cpp
Paul's avatar
Paul committed
71
)
72
configure_file(version.h.in include/migraphx/version.h)
Paul's avatar
Paul committed
73
rocm_set_soversion(migraphx ${MIGRAPHX_SO_VERSION})
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
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
    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
103
    dequantizelinear
104
105
106
    div
    dot
    elu
107
    equal
108
109
110
111
112
    erf
    exp
    flatten
    floor
    gather
turneram's avatar
turneram committed
113
    gathernd
Shucai Xiao's avatar
Shucai Xiao committed
114
    get_tuple_elem
115
    greater
116
117
    gru
    identity
Shucai Xiao's avatar
Shucai Xiao committed
118
    if_op
119
    im2col
Charlie Lin's avatar
Charlie Lin committed
120
    isnan
Paul's avatar
Paul committed
121
    layout
122
    leaky_relu
123
    less
124
125
    load
    log
Shucai Xiao's avatar
Shucai Xiao committed
126
127
128
    logical_and
    logical_or
    logical_xor
129
    logsoftmax
Shucai Xiao's avatar
Shucai Xiao committed
130
    loop
131
132
133
134
135
136
    lrn
    lstm
    max
    min
    mul
    multibroadcast
turneram's avatar
turneram committed
137
    multinomial
138
    neg
139
    nonmaxsuppression
Shucai Xiao's avatar
Shucai Xiao committed
140
    nonzero
141
142
    outline
    pad
143
    pointwise
144
145
    pooling
    pow
turneram's avatar
turneram committed
146
    prefix_scan_sum
147
148
149
    prelu
    quant_convolution
    quant_dot
turneram's avatar
turneram committed
150
    quantizelinear
151
152
153
154
155
156
157
158
    recip
    reduce_max
    reduce_mean
    reduce_min
    reduce_prod
    reduce_sum
    relu
    reshape
Cagri Eryilmaz's avatar
Cagri Eryilmaz committed
159
    reverse
160
161
162
163
    rnn
    rnn_last_cell_output
    rnn_last_hs_output
    rnn_var_sl_last_output
Shucai Xiao's avatar
Shucai Xiao committed
164
    roialign
165
166
167
    round
    rsqrt
    scalar
168
169
170
    scatter_add
    scatter_mul
    scatter_none
turneram's avatar
turneram committed
171
172
    scatternd_add
    scatternd_mul
173
    scatternd_none
174
175
176
177
178
179
180
181
182
    sigmoid
    sign
    sinh
    sin
    slice
    softmax
    sqdiff
    sqrt
    squeeze
Shucai Xiao's avatar
Shucai Xiao committed
183
    step
184
185
186
    sub
    tanh
    tan
Shucai Xiao's avatar
Shucai Xiao committed
187
    topk
188
    transpose
189
    unary_not
190
191
192
    undefined
    unknown
    unsqueeze
turneram's avatar
turneram committed
193
    where
194
195
196
)
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
197
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
198
rocm_install_targets(
Paul's avatar
Paul committed
199
  TARGETS migraphx
Paul's avatar
Paul committed
200
201
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
202
    ${CMAKE_CURRENT_BINARY_DIR}/include
Paul's avatar
Paul committed
203
)
Paul Fultz II's avatar
Paul Fultz II committed
204
205
206
207
208
209
210


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

211
212
target_link_libraries(migraphx PRIVATE -ldl)

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

215
216
217
find_package(Threads)
target_link_libraries(migraphx PUBLIC Threads::Threads)

218
219
220
221
222
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>)

223
224
add_library(migraphx_all_targets INTERFACE)

Paul's avatar
Paul committed
225
set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
226

Paul Fultz II's avatar
Paul Fultz II committed
227
add_subdirectory(api)
Paul's avatar
Paul committed
228
add_subdirectory(driver)
Paul's avatar
Paul committed
229
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
230
add_subdirectory(tf)
Khalique's avatar
Khalique committed
231

Paul's avatar
Paul committed
232
add_subdirectory(py)
233
add_subdirectory(targets/ref)
234
target_link_libraries(migraphx_all_targets INTERFACE migraphx_ref)
235
236
237
238
239
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
240
if(MIGRAPHX_ENABLE_GPU)
Paul Fultz II's avatar
Paul Fultz II committed
241
list(APPEND PACKAGE_DEPENDS PACKAGE MIOpen PACKAGE rocblas)
Paul's avatar
Paul committed
242
add_subdirectory(targets/gpu)
243
244
target_link_libraries(migraphx_all_targets INTERFACE migraphx_gpu)
target_compile_definitions(migraphx_all_targets INTERFACE -DHAVE_GPU)
Paul's avatar
Paul committed
245
endif()
mei-ye's avatar
mei-ye committed
246

turneram's avatar
turneram committed
247
248
249
250
if(HAVE_HALF_EXPR)
    target_compile_definitions(migraphx PUBLIC -DHAS_HALF_V1)
endif()

Paul's avatar
Paul committed
251
rocm_export_targets(
252
  TARGETS migraphx::migraphx_c
Paul's avatar
Paul committed
253
  NAMESPACE migraphx::
Paul's avatar
Paul committed
254
  DEPENDS
255
    Threads
Paul's avatar
Paul committed
256
257
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
258
259