"official/recommendation/movielens.py" did not exist on "139dd8e937a036f2094482dc4ddae2678cb97f9d"
CMakeLists.txt 868 Bytes
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
add_library(rtg_onnx onnx.cpp)
rocm_clang_tidy_check(rtg_onnx)
Paul's avatar
Use pic  
Paul committed
12
13
target_link_libraries(rtg_onnx PRIVATE onnx-proto)
target_link_libraries(rtg_onnx PUBLIC rtg)
Paul's avatar
Paul committed
14

Paul's avatar
Paul committed
15
16
add_executable(read_onnx read_onnx.cpp)
rocm_clang_tidy_check(read_onnx)
Paul's avatar
Paul committed
17
18
target_link_libraries(read_onnx rtg_onnx)

Paul's avatar
Paul committed
19
if(RTG_ENABLE_MIOPEN)
Paul's avatar
Paul committed
20
21
22
add_executable(verify_onnx verify_onnx.cpp)
rocm_clang_tidy_check(verify_onnx)
target_link_libraries(verify_onnx rtg_onnx rtg_cpu rtg_miopen)
Paul's avatar
Paul committed
23
endif()