Commit 5e018774 by aswinjohn Committed by Tianqi Chen

Minor bug: causes undefined symbol error in libtvm from NNPACK (#1368)

parent 60b874e2
...@@ -7,7 +7,7 @@ if(USE_NNPACK) ...@@ -7,7 +7,7 @@ if(USE_NNPACK)
list(APPEND RUNTIME_SRCS ${NNPACK_CONTRIB_SRC}) list(APPEND RUNTIME_SRCS ${NNPACK_CONTRIB_SRC})
include_directories(${NNPACK_PATH}/include) include_directories(${NNPACK_PATH}/include)
include_directories(${PTHREAD_POOL_PATH}/include) include_directories(${PTHREAD_POOL_PATH}/include)
find_library(NNPACK_CONTRB_LIB nnpack ${NNPACK_PATH}/lib) find_library(NNPACK_CONTRIB_LIB nnpack ${NNPACK_PATH}/lib)
find_library(NNPACK_PTHREAD_CONTRIB_LIB pthreadpool ${NNPACK_PATH}/lib) find_library(NNPACK_PTHREAD_CONTRIB_LIB pthreadpool ${NNPACK_PATH}/lib)
list(APPEND TVM_RUNTIME_LINKER_LIBS ${NNPACK_CONTRIB_LIB}) list(APPEND TVM_RUNTIME_LINKER_LIBS ${NNPACK_CONTRIB_LIB})
list(APPEND TVM_RUNTIME_LINKER_LIBS ${NNPACK_PTHREAD_CONTRIB_LIB}) list(APPEND TVM_RUNTIME_LINKER_LIBS ${NNPACK_PTHREAD_CONTRIB_LIB})
......
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