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") ...@@ -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_LIB "${CMAKE_INSTALL_PREFIX}/lib")
set(NGRAPH_INSTALL_INCLUDE "${CMAKE_INSTALL_PREFIX}/include") set(NGRAPH_INSTALL_INCLUDE "${CMAKE_INSTALL_PREFIX}/include")
set(NGRAPH_INSTALL_DOC "${CMAKE_INSTALL_PREFIX}/doc") set(NGRAPH_INSTALL_DOC "${CMAKE_INSTALL_PREFIX}/doc")
set(NGRAPH_INSTALL_BIN "${CMAKE_INSTALL_PREFIX}/bin")
set(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_PREFIX}/lib") set(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_PREFIX}/lib")
#----------------------------------------------------------------------------------------------- #-----------------------------------------------------------------------------------------------
......
...@@ -14,10 +14,6 @@ ...@@ -14,10 +14,6 @@
# limitations under the License. # limitations under the License.
# ****************************************************************************** # ******************************************************************************
if(MKLDNN_INCLUDE_DIR)
link_directories(${MKLDNN_LIB_DIR})
endif()
if (NGRAPH_CPU_ENABLE) if (NGRAPH_CPU_ENABLE)
set (SRC set (SRC
nbench.cpp nbench.cpp
...@@ -31,9 +27,16 @@ if (NGRAPH_CPU_ENABLE) ...@@ -31,9 +27,16 @@ if (NGRAPH_CPU_ENABLE)
"NGRAPH_HEADERS_PATH=\"${NGRAPH_INCLUDE_PATH}\"" "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") include_directories("${PROJECT_SOURCE_DIR}/test")
set_source_files_properties(nbench.cpp PROPERTIES COMPILE_DEFINITIONS "${HEADER_SEARCH_DEFINES}") set_source_files_properties(nbench.cpp PROPERTIES COMPILE_DEFINITIONS "${HEADER_SEARCH_DEFINES}")
install(TARGETS nbench RUNTIME DESTINATION ${NGRAPH_INSTALL_BIN})
endif() 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