зеркало из https://github.com/microsoft/EdgeML.git
Merge branch 'edgemlCmake' of https://github.com/Microsoft/EdgeML into edgemlCmake
This commit is contained in:
Коммит
f36bd6c109
|
@ -26,20 +26,26 @@ endif()
|
|||
|
||||
#define variables for mkl include directories
|
||||
#set your MKL_ROOT here
|
||||
set(MKL_ROOT "C:/Program Files (x86)/IntelSWTools/compilers_and_libraries/windows/mkl")
|
||||
set(MKL_INCLUDE_DIR ${MKL_ROOT}/include)
|
||||
include_directories(${MKL_INCLUDE_DIR})
|
||||
link_directories(${MKL_ROOT}/lib/intel64_win)
|
||||
link_directories(${MKL_ROOT}/../compiler/lib/intel64_win)
|
||||
|
||||
#set(MKL_ROOT "/opt/intel/mkl/")
|
||||
#set(MKL_INCLUDE_DIR ${MKL_ROOT}/include)
|
||||
#include_directories(${MKL_INCLUDE_DIR})
|
||||
#link_directories(${MKL_ROOT}/lib/intel64_lin)
|
||||
#link_directories(${MKL_ROOT}/../compiler/lib/intel64_lin)
|
||||
if(MSVC)
|
||||
set(MKL_ROOT "C:/Program Files (x86)/IntelSWTools/compilers_and_libraries/windows/mkl")
|
||||
set(MKL_INCLUDE_DIR ${MKL_ROOT}/include)
|
||||
include_directories(${MKL_INCLUDE_DIR})
|
||||
link_directories(${MKL_ROOT}/lib/intel64_win)
|
||||
link_directories(${MKL_ROOT}/../compiler/lib/intel64_win)
|
||||
ENDIF(MSVC)
|
||||
|
||||
IF(CMAKE_COMPILER_IS_GNUCC)
|
||||
set(MKL_ROOT "/opt/intel/mkl/")
|
||||
set(MKL_INCLUDE_DIR ${MKL_ROOT}/include)
|
||||
include_directories(${MKL_INCLUDE_DIR})
|
||||
link_directories(${MKL_ROOT}/lib/intel64_lin)
|
||||
link_directories(${MKL_ROOT}/../compiler/lib/intel64_lin)
|
||||
ENDIF(CMAKE_COMPILER_IS_GNUCC)
|
||||
|
||||
# add debug definitions to compiler flags
|
||||
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -DLIGHT_LOGGER") #-DLOGGER #-DTIMER -DCONCISE #-DSTDERR_ONSCREEN #-DLIGHT_LOGGER -DVERBOSE #-DDUMP #-DVERIFY
|
||||
#set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -DLIGHT_LOGGER") #-DLOGGER #-DTIMER -DCONCISE #-DSTDERR_ONSCREEN #-DLIGHT_LOGGER -DVERBOSE #-DDUMP #-DVERIFY
|
||||
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -DLIGHT_LOGGER -DSTDERR_ONSCREEN -DVERBOSE -DDUMP -DVERIFY") #-DLOGGER #-DTIMER -DCONCISE #-DSTDERR_ONSCREEN #-DLIGHT_LOGGER -DVERBOSE #-DDUMP #-DVERIFY
|
||||
|
||||
set(CONFIG_FLAGS "-DSINGLE") #-DXML -DZERO_BASED_IO
|
||||
|
||||
|
|
|
@ -4,8 +4,11 @@ set (src ProtoNNTrainDriver.cpp)
|
|||
|
||||
source_group("src" FILES ${src})
|
||||
|
||||
set (CMAKE_RUNTIME_OUTPUT_DIRECTORY_DEBUG ${CMAKE_SOURCE_DIR})
|
||||
set (CMAKE_RUNTIME_OUTPUT_DIRECTORY_RELEASE ${CMAKE_SOURCE_DIR})
|
||||
#set (CMAKE_RUNTIME_OUTPUT_DIRECTORY_DEBUG ${CMAKE_SOURCE_DIR})
|
||||
#set (CMAKE_RUNTIME_OUTPUT_DIRECTORY_RELEASE ${CMAKE_SOURCE_DIR})
|
||||
|
||||
set (GLOBAL_BIN_DIR ${CMAKE_BINARY_DIR}/bin)
|
||||
set (EXECUTABLE_OUTPUT_PATH ${GLOBAL_BIN_DIR})
|
||||
|
||||
add_executable(${tool_name} ${src} ${include})
|
||||
target_include_directories(${tool_name} PRIVATE ${CMAKE_SOURCE_DIR}/src/common ${CMAKE_SOURCE_DIR}/src/ProtoNN)
|
||||
|
@ -18,4 +21,4 @@ IF(NOT CMAKE_COMPILER_IS_GNUCC)
|
|||
target_link_libraries(${tool_name} common ProtoNN mkl_intel_ilp64 mkl_intel_thread mkl_core libiomp5md)
|
||||
ENDIF (NOT CMAKE_COMPILER_IS_GNUCC)
|
||||
|
||||
set_property(TARGET ${tool_name} PROPERTY FOLDER "drivers/ProtoNN")
|
||||
set_property(TARGET ${tool_name} PROPERTY FOLDER "drivers/ProtoNN")
|
||||
|
|
|
@ -25,7 +25,7 @@ ntest="-N 2007"
|
|||
#gdb=" gdb --args"
|
||||
executable="./BonsaiPredict"
|
||||
command=$gdb" "$executable" "$input_format" "$ntest" "$input_dir" "$model_dir
|
||||
echo "Running Bonsai with following command: "
|
||||
echo "Running Bonsai predict with following command: "
|
||||
echo $command
|
||||
echo ""
|
||||
exec $command
|
||||
|
|
|
@ -48,9 +48,9 @@ iters="-I 100"
|
|||
########################################################
|
||||
|
||||
#gdb=" gdb --args"
|
||||
executable="./Bonsai"
|
||||
executable="./BonsaiTrain"
|
||||
command=$gdb" "$executable" "$input_format" "$num_features" "$num_labels" "$ntrain" "$ntest" "$projection_dimension" "$tree_depth" "$sigma" "$reg_W" "$reg_Z" "$reg_Theta" "$reg_V" "$sparse_Z" "$sparse_Theta" "$sparse_V" "$sparse_W" "$batch_factor" "$iters" "$input_dir
|
||||
echo "Running Bonsai with following command: "
|
||||
echo "Running Bonsai train with following command: "
|
||||
echo $command
|
||||
echo ""
|
||||
exec $command
|
||||
|
|
Загрузка…
Ссылка в новой задаче