frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

test existence of confu and ninja before installing nnpack.

Browse Source 
Summary:
TSIA
Closes https://github.com/caffe2/caffe2/pull/229

Differential Revision: D4795563

Pulled By: aaronmarkham

fbshipit-source-id: 4df871760a1124bb7a2ef226d01b4ced12d21ab1
This commit is contained in:
Yangqing Jia
2017-03-29 10:08:47 -07:00
committed by Facebook Github Bot
parent 04210ad531
commit afe3df32f5
1 changed files with 30 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
cmake/External/nnpack.cmake vendored
View File

@ -48,28 +48,37 @@ endif()
if (${CMAKE_SYSTEM_NAME} STREQUAL "Linux" OR ${CMAKE_SYSTEM_NAME} MATCHES "Darwin")
message(STATUS "Will try to build NNPACK from source. If anything fails, "
"follow the NNPACK prerequisite installation steps.")
# Note: per Marat, there is no support for fPIC right now so we will need to
# manually change it in build.ninja
ExternalProject_Add(nnpack_external
SOURCE_DIR ${NNPACK_PREFIX}
BUILD_IN_SOURCE 1
CONFIGURE_COMMAND ""
BUILD_COMMAND confu setup
COMMAND python ./configure.py
COMMAND sed -ibuild.ninja.bak "s/cflags = /cflags = -fPIC /" build.ninja
COMMAND sed -ibuild.ninja.bak "s/cxxflags = /cxxflags = -fPIC /" build.ninja
COMMAND ninja nnpack
INSTALL_COMMAND ""
)
find_program(CAFFE2_CONFU_COMMAND confu)
find_program(CAFFE2_NINJA_COMMAND ninja)
if (CAFFE2_CONFU_COMMAND AND CAFFE2_NINJA_COMMAND)
# Note: per Marat, there is no support for fPIC right now so we will need to
# manually change it in build.ninja
ExternalProject_Add(nnpack_external
SOURCE_DIR ${NNPACK_PREFIX}
BUILD_IN_SOURCE 1
CONFIGURE_COMMAND ""
BUILD_COMMAND confu setup
COMMAND python ./configure.py
COMMAND sed -ibuild.ninja.bak "s/cflags = /cflags = -fPIC /" build.ninja
COMMAND sed -ibuild.ninja.bak "s/cxxflags = /cxxflags = -fPIC /" build.ninja
COMMAND ninja nnpack
INSTALL_COMMAND ""
)
set(NNPACK_FOUND TRUE)
set(NNPACK_INCLUDE_DIRS
${NNPACK_PREFIX}/include
${NNPACK_PREFIX}/deps/pthreadpool/include)
set(NNPACK_LIBRARIES ${NNPACK_PREFIX}/lib/libnnpack.a ${NNPACK_PREFIX}/lib/libpthreadpool.a)
set(NNPACK_LIBRARY_DIRS ${NNPACK_PREFIX}/lib)
list(APPEND external_project_dependencies nnpack_external)
set(NNPACK_FOUND TRUE)
set(NNPACK_INCLUDE_DIRS
${NNPACK_PREFIX}/include
${NNPACK_PREFIX}/deps/pthreadpool/include)
set(NNPACK_LIBRARIES ${NNPACK_PREFIX}/lib/libnnpack.a ${NNPACK_PREFIX}/lib/libpthreadpool.a)
set(NNPACK_LIBRARY_DIRS ${NNPACK_PREFIX}/lib)
list(APPEND external_project_dependencies nnpack_external)
else()
message(WARNING "NNPACK is chosen to be installed, but confu and ninja "
"that are needed by it are not installed. As a result "
"we won't build with NNPACK.")
set(USE_NNPACK OFF)
endif()
return()
endif()