Commit 1f5b690d authored by Robert Kimball's avatar Robert Kimball Committed by Scott Cyphers

add nbench to install and setup nbench lib paths to work when running in tree (#1081)

parent 52313f9e
......@@ -113,6 +113,7 @@ message (STATUS "To Override use: cmake -DNGRAPH_INSTALL_PREFIX=/foo")
set(NGRAPH_INSTALL_LIB "${CMAKE_INSTALL_PREFIX}/lib")
set(NGRAPH_INSTALL_INCLUDE "${CMAKE_INSTALL_PREFIX}/include")
set(NGRAPH_INSTALL_DOC "${CMAKE_INSTALL_PREFIX}/doc")
set(NGRAPH_INSTALL_BIN "${CMAKE_INSTALL_PREFIX}/bin")
set(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_PREFIX}/lib")
#-----------------------------------------------------------------------------------------------
......
......@@ -14,10 +14,6 @@
# limitations under the License.
# ******************************************************************************
if(MKLDNN_INCLUDE_DIR)
link_directories(${MKLDNN_LIB_DIR})
endif()
if (NGRAPH_CPU_ENABLE)
set (SRC
nbench.cpp
......@@ -31,9 +27,16 @@ if (NGRAPH_CPU_ENABLE)
"NGRAPH_HEADERS_PATH=\"${NGRAPH_INCLUDE_PATH}\""
)
target_link_libraries(nbench ngraph)
target_link_libraries(nbench ngraph cpu_backend)
if (NGRAPH_GPU_ENABLE)
target_link_libraries(nbench gpu_backend)
endif()
if (NGRAPH_INTERPRETER_ENABLE)
target_link_libraries(nbench interpreter_backend)
endif()
include_directories("${PROJECT_SOURCE_DIR}/test")
set_source_files_properties(nbench.cpp PROPERTIES COMPILE_DEFINITIONS "${HEADER_SEARCH_DEFINES}")
install(TARGETS nbench RUNTIME DESTINATION ${NGRAPH_INSTALL_BIN})
endif()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment