CMakeLists.txt 1.33 KB
Newer Older
Paul's avatar
Paul committed
1
2
3
find_package(Protobuf REQUIRED)

protobuf_generate_cpp(PROTO_SRCS PROTO_HDRS onnx.proto)
Paul's avatar
Paul committed
4
5
6
7
add_library(onnx-proto STATIC ${PROTO_SRCS})
target_include_directories(onnx-proto SYSTEM PUBLIC ${CMAKE_CURRENT_BINARY_DIR} ${PROTOBUF_INCLUDE_DIR})
target_compile_options(onnx-proto PRIVATE -w)
target_link_libraries(onnx-proto PRIVATE ${PROTOBUF_LIBRARY})
Paul's avatar
Use pic  
Paul committed
8
set_target_properties(onnx-proto PROPERTIES POSITION_INDEPENDENT_CODE On)
Paul's avatar
Paul committed
9

Paul's avatar
Paul committed
10
11
12
13
add_library(migraph_onnx onnx.cpp)
rocm_clang_tidy_check(migraph_onnx)
target_link_libraries(migraph_onnx PRIVATE onnx-proto)
target_link_libraries(migraph_onnx PUBLIC migraph)
Paul's avatar
Paul committed
14

Paul's avatar
Paul committed
15
16
17
18
rocm_install_targets(
  TARGETS migraph_onnx
)

Paul's avatar
Paul committed
19
20
add_executable(read_onnx read_onnx.cpp)
rocm_clang_tidy_check(read_onnx)
Paul's avatar
Paul committed
21
target_link_libraries(read_onnx migraph_onnx)
Scott Thornton's avatar
Scott Thornton committed
22

Paul's avatar
Paul committed
23

24
if(MIGRAPH_ENABLE_GPU)
Scott Thornton's avatar
Scott Thornton committed
25
26
add_executable(mnist mnist.cpp)
rocm_clang_tidy_check(mnist)
27
target_link_libraries(mnist migraph_cpu migraph_gpu migraph_onnx)
28

29
30
31
add_executable(cifar10 cifar10.cpp)
rocm_clang_tidy_check(cifar10)
target_link_libraries(cifar10 migraph_cpu migraph_gpu migraph_onnx)
32

Paul's avatar
Paul committed
33
34
add_executable(verify_onnx verify_onnx.cpp)
rocm_clang_tidy_check(verify_onnx)
Paul's avatar
Paul committed
35
target_link_libraries(verify_onnx migraph_onnx migraph_cpu migraph_gpu)
Paul's avatar
Paul committed
36
37
38
39

add_executable(perf_onnx perf_onnx.cpp)
rocm_clang_tidy_check(perf_onnx)
target_link_libraries(perf_onnx migraph_onnx migraph_cpu migraph_gpu)
Paul's avatar
Paul committed
40
endif()