Display CUB and CUDNN paths (if defined) in BuildInfo
Print BuildInfo at the very begining of the program. convenient for checking build type.
This commit is contained in:
Родитель
b54cfccc37
Коммит
6777bbe0b7
|
@ -345,6 +345,9 @@ void PrintBuiltInfo()
|
|||
#ifdef _CUB_PATH_
|
||||
fprintf(stderr, "\t\tCUB_PATH: %s\n", _CUB_PATH_);
|
||||
#endif
|
||||
#ifdef _CUDNN_PATH_
|
||||
fprintf(stderr, "\t\tCUDNN_PATH: %s\n", _CUDNN_PATH_);
|
||||
#endif
|
||||
#ifdef _GIT_EXIST
|
||||
fprintf(stderr, "\t\tBuild Branch: %s\n", _BUILDBRANCH_);
|
||||
fprintf(stderr, "\t\tBuild SHA1: %s\n", _BUILDSHA1_);
|
||||
|
@ -568,7 +571,7 @@ int wmainOldCNTKConfig(int argc, wchar_t* argv[]) // called from wmain which i
|
|||
RedirectStdErr(logpath);
|
||||
}
|
||||
|
||||
PrintBuiltInfo();
|
||||
PrintBuiltInfo(); // this one goes to log file
|
||||
std::string timestamp = TimeDateStamp();
|
||||
|
||||
//dump config info
|
||||
|
@ -643,10 +646,11 @@ int wmainOldCNTKConfig(int argc, wchar_t* argv[]) // called from wmain which i
|
|||
// main wrapper that catches C++ exceptions and prints them
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
int wmain1(int argc, wchar_t* argv[]) // called from wmain which is a wrapper that catches & repots Win32 exceptions
|
||||
int wmain1(int argc, wchar_t* argv[]) // called from wmain which is a wrapper that catches & reports Win32 exceptions
|
||||
{
|
||||
try
|
||||
{
|
||||
PrintBuiltInfo(); // print build info directly in case that user provides zero argument (convenient for checking build type)
|
||||
if (argc <= 1)
|
||||
InvalidArgument("No command-line argument given.");
|
||||
// detect legacy CNTK configuration
|
||||
|
|
|
@ -33,6 +33,16 @@ if "%cuda_path%" == "" (
|
|||
echo #define _CUDA_PATH_ "%cuda_path:\=\\%" >> buildinfo.h$$
|
||||
)
|
||||
|
||||
if not "%cudnn_path%" == "" (
|
||||
echo #define _CUDNN_PATH_ "%cudnn_path:\=\\%" >> buildinfo.h$$
|
||||
)
|
||||
|
||||
if not "%cub_path%" == "" (
|
||||
echo #define _CUB_PATH_ "%cub_path:\=\\%" >> buildinfo.h$$
|
||||
)
|
||||
|
||||
|
||||
|
||||
echo #endif >> buildinfo.h$$
|
||||
|
||||
::: update file only if it changed (otherwise CNTK.cpp will get rebuilt each time)
|
||||
|
|
|
@ -56,6 +56,9 @@ makebuildinfo()
|
|||
if [ ! -z "$CUB_PATH" ]; then
|
||||
printf "#define _CUB_PATH_ \"%s\"\n" $CUB_PATH >> $target
|
||||
fi
|
||||
if [ ! -z "$CUDNN_PATH" ]; then
|
||||
printf "#define _CUDNN_PATH_ \"%s\"\n" $CUDNN_PATH >> $target
|
||||
fi
|
||||
printf "#define _BUILDTYPE_ \"%s\"\n" $BUILDTYPE >> $target
|
||||
printf "#endif\n" >> $target
|
||||
}
|
||||
|
|
Загрузка…
Ссылка в новой задаче