This commit is contained in:
Suresh Parthasarathy 2017-11-16 15:45:24 +05:30
Родитель c1fc8922ea
Коммит 6875c3f2c3
4 изменённых файлов: 26 добавлений и 17 удалений

Просмотреть файл

@ -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