зеркало из https://github.com/microsoft/caffe.git
73 строки
3.0 KiB
CMake
73 строки
3.0 KiB
CMake
# Builds Matlab (or Octave) interface. In case of Matlab caffe must be
|
|
# compield as shared library. Octave can link static or shared caffe library
|
|
# To install octave run: sudo apt-get install liboctave-dev
|
|
|
|
if(NOT BUILD_matlab)
|
|
return()
|
|
endif()
|
|
|
|
if(HAVE_MATLAB AND Octave_compiler)
|
|
set(build_using ${Matlab_build_mex_using})
|
|
elseif(HAVE_MATLAB AND NOT Octave_compiler)
|
|
set(build_using "Matlab")
|
|
elseif(NOT HAVE_MATLAB AND Octave_compiler)
|
|
set(build_using "Octave")
|
|
else()
|
|
return()
|
|
endif()
|
|
|
|
if(NOT BUILD_SHARED_LIBS AND build_using MATCHES Matlab)
|
|
message(FATAL_ERROR "Matlab MEX interface (with default mex options file) can only be built if caffe is compiled as shared library. Please enable 'BUILD_SHARED_LIBS' in CMake. Aternativelly you can switch to Octave compiler.")
|
|
endif()
|
|
|
|
# helper function to set proper mex file extention
|
|
function(caffe_fetch_and_set_proper_mexext mexfile_variable)
|
|
execute_process(COMMAND ${Matlab_mexext} OUTPUT_STRIP_TRAILING_WHITESPACE RESULT_VARIABLE res OUTPUT_VARIABLE ext)
|
|
if(res MATCHES 0)
|
|
get_filename_component(folder ${${mexfile_variable}} PATH)
|
|
get_filename_component(name_we ${${mexfile_variable}} NAME_WE)
|
|
set(${mexfile_variable} ${folder}/${name_we}.${ext} PARENT_SCOPE)
|
|
endif()
|
|
endfunction()
|
|
|
|
# global settings
|
|
file(GLOB Matlab_srcs +caffe/private/caffe_.cpp)
|
|
set(Matlab_caffe_mex ${PROJECT_SOURCE_DIR}/matlab/+caffe/private/caffe_.mex)
|
|
|
|
caffe_get_current_cflags(cflags)
|
|
caffe_parse_linker_libs(Caffe_LINKER_LIBS folders libflags macos_frameworks)
|
|
set(folders $<TARGET_LINKER_FILE_DIR:caffe> ${folders})
|
|
|
|
# prepare linker flag lists
|
|
string(REPLACE ";" ";-L" link_folders "-L${folders}")
|
|
string(REPLACE ";" ":" rpath_folders "${folders}")
|
|
|
|
if(build_using MATCHES "Matlab")
|
|
set(libflags -lcaffe${CAffe_POSTFIX} ${libflags}) # Matlab R2014a complans for -Wl,--whole-archive
|
|
|
|
caffe_fetch_and_set_proper_mexext(Matlab_caffe_mex)
|
|
add_custom_command(OUTPUT ${Matlab_caffe_mex} COMMAND ${Matlab_mex}
|
|
ARGS -output ${Matlab_caffe_mex} ${Matlab_srcs} ${cflags} ${link_folders} ${libflags}
|
|
DEPENDS caffe COMMENT "Building Matlab interface: ${Matlab_caffe_mex}" VERBATIM)
|
|
add_custom_target(matlab ALL DEPENDS ${Matlab_caffe_mex} SOURCES ${Matlab_srcs})
|
|
|
|
elseif(build_using MATCHES "Octave")
|
|
|
|
if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang")
|
|
set(libflags -Wl,-force_load,$<TARGET_LINKER_FILE:caffe> ${libflags})
|
|
elseif("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU")
|
|
set(libflags -Wl,--whole-archive -lcaffe${CAffe_POSTFIX} -Wl,--no-whole-archive ${libflags})
|
|
endif()
|
|
|
|
add_custom_command(OUTPUT ${Matlab_caffe_mex} COMMAND ${Octave_compiler}
|
|
ARGS --mex -o ${Matlab_caffe_mex} ${Matlab_srcs} ${cflags} ${link_folders} ${libflags} -Wl,-rpath,${rpath_folders}
|
|
DEPENDS caffe COMMENT "Building Octave interface: ${Matlab_caffe_mex}" VERBATIM)
|
|
|
|
add_custom_target(octave ALL DEPENDS ${Matlab_caffe_mex} SOURCES ${Matlab_srcs})
|
|
endif()
|
|
|
|
# ---[ Install
|
|
file(GLOB mfiles caffe/*.m)
|
|
install(FILES ${mfiles} ${Matlab_caffe_mex} DESTINATION matlab)
|
|
|