CMakeLists.txt 4.13 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
Paul's avatar
Paul committed
10
    auto_contiguous.cpp
Paul's avatar
Paul committed
11
    eliminate_common_subexpression.cpp
12
    decompose.cpp
Paul's avatar
Paul committed
13
    propagate_constant.cpp
Paul's avatar
Paul committed
14
    dead_code_elimination.cpp
Paul's avatar
Paul committed
15
    eliminate_allocation.cpp
16
    eliminate_contiguous.cpp
17
    eliminate_concat.cpp
18
    eliminate_data_type.cpp
19
    eliminate_identity.cpp
20
    eliminate_pad.cpp
Paul Fultz II's avatar
Paul Fultz II committed
21
    file_buffer.cpp
Paul's avatar
Paul committed
22
    rewrite_batchnorm.cpp
Shucai Xiao's avatar
Shucai Xiao committed
23
    rewrite_rnn.cpp
24
    rewrite_pooling.cpp
Paul's avatar
Paul committed
25
    env.cpp
Paul's avatar
Paul committed
26
    generate.cpp
Paul's avatar
Paul committed
27
    instruction.cpp
28
    load_save.cpp
29
    make_op.cpp
30
    msgpack.cpp
31
    operation.cpp
32
    permutation.cpp
Paul's avatar
Paul committed
33
    program.cpp
Shucai Xiao's avatar
Shucai Xiao committed
34
    module.cpp
Shucai Xiao's avatar
Shucai Xiao committed
35
    quantization.cpp
36
    reduce_dims.cpp
37
    remap.cpp
Paul's avatar
Paul committed
38
    shape.cpp
Paul's avatar
Paul committed
39
    schedule.cpp
40
    serialize.cpp
41
    pass_manager.cpp
42
    register_op.cpp
43
    register_target.cpp
Paul's avatar
Paul committed
44
    simplify_algebra.cpp
Paul's avatar
Paul committed
45
    simplify_reshapes.cpp
Paul Fultz II's avatar
Paul Fultz II committed
46
    tmp_dir.cpp
47
    value.cpp
48
    verify_args.cpp
Shucai Xiao's avatar
Shucai Xiao committed
49
    json.cpp
50
    convert_to_json.cpp
mei-ye's avatar
mei-ye committed
51
52
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Shucai Xiao's avatar
Shucai Xiao committed
53
54
    normalize_attributes.cpp
    normalize_ops.cpp
Paul's avatar
Paul committed
55
)
Paul's avatar
Paul committed
56
rocm_set_soversion(migraphx ${MIGRAPHX_SO_VERSION})
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
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
    div
    dot
    elu
89
    equal
90
91
92
93
94
    erf
    exp
    flatten
    floor
    gather
95
    greater
96
97
    gru
    identity
Shucai Xiao's avatar
Shucai Xiao committed
98
    if_op
99
100
    im2col
    leaky_relu
101
    less
102
103
    load
    log
Shucai Xiao's avatar
Shucai Xiao committed
104
105
106
    logical_and
    logical_or
    logical_xor
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
    logsoftmax
    lrn
    lstm
    max
    min
    mul
    multibroadcast
    neg
    outline
    pad
    pooling
    pow
    prelu
    quant_convolution
    quant_dot
    recip
    reduce_max
    reduce_mean
    reduce_min
    reduce_prod
    reduce_sum
    relu
    reshape
    rnn
    rnn_last_cell_output
    rnn_last_hs_output
    rnn_var_sl_last_output
    round
    rsqrt
    scalar
    sigmoid
    sign
    sinh
    sin
    slice
    softmax
    sqdiff
    sqrt
    squeeze
    sub
    tanh
    tan
    transpose
150
    unary_not
151
152
153
154
155
156
    undefined
    unknown
    unsqueeze
)
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
157
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
158
rocm_install_targets(
Paul's avatar
Paul committed
159
  TARGETS migraphx
Paul's avatar
Paul committed
160
161
162
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
Paul Fultz II's avatar
Paul Fultz II committed
163
164
165
166
167
168
169


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

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

172
173
174
175
176
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>)

177
178
179
add_library(migraphx_all_targets INTERFACE)
target_link_libraries(migraphx_all_targets INTERFACE migraphx_ref)

Paul's avatar
Paul committed
180
set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
181

Paul Fultz II's avatar
Paul Fultz II committed
182
add_subdirectory(api)
Paul's avatar
Paul committed
183
add_subdirectory(driver)
Paul's avatar
Paul committed
184
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
185
add_subdirectory(tf)
Khalique's avatar
Khalique committed
186

Paul's avatar
Paul committed
187
add_subdirectory(py)
188
add_subdirectory(targets/ref)
189
190
191
192
193
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
194
if(MIGRAPHX_ENABLE_GPU)
Paul Fultz II's avatar
Paul Fultz II committed
195
list(APPEND PACKAGE_DEPENDS PACKAGE MIOpen PACKAGE rocblas)
Paul's avatar
Paul committed
196
add_subdirectory(targets/gpu)
197
198
target_link_libraries(migraphx_all_targets INTERFACE migraphx_gpu)
target_compile_definitions(migraphx_all_targets INTERFACE -DHAVE_GPU)
Paul's avatar
Paul committed
199
endif()
mei-ye's avatar
mei-ye committed
200

turneram's avatar
turneram committed
201
202
203
204
if(HAVE_HALF_EXPR)
    target_compile_definitions(migraphx PUBLIC -DHAS_HALF_V1)
endif()

Paul's avatar
Paul committed
205
rocm_export_targets(
206
  TARGETS migraphx::migraphx migraphx_all_targets
Paul's avatar
Paul committed
207
  NAMESPACE migraphx::
Paul's avatar
Paul committed
208
209
210
  DEPENDS
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
211
212