mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
57b51db8d7
Summary: This helps diagnosing issues like #346 Closes https://github.com/caffe2/caffe2/pull/354 Differential Revision: D4928347 Pulled By: Yangqing fbshipit-source-id: b45685f1da18cbc49be293260b1fc2268fe5cd4c
44 lines
1.5 KiB
C++
44 lines
1.5 KiB
C++
#include "caffe2/core/init.h"
|
|
#include "caffe2/core/operator.h" // for StaticLinkingProtector
|
|
|
|
#ifndef CAFFE2_BUILD_STRING
|
|
#define CAFFE2_BUILD_STRING "build_version_not_set"
|
|
#endif
|
|
namespace caffe2 {
|
|
|
|
namespace internal {
|
|
Caffe2InitializeRegistry* Caffe2InitializeRegistry::Registry() {
|
|
static Caffe2InitializeRegistry gRegistry;
|
|
return &gRegistry;
|
|
}
|
|
}
|
|
|
|
bool GlobalInit(int* pargc, char*** pargv) {
|
|
static bool global_init_was_already_run = false;
|
|
static StaticLinkingProtector g_protector;
|
|
if (global_init_was_already_run) {
|
|
VLOG(1) << "GlobalInit has already been called: did you double-call?";
|
|
return true;
|
|
}
|
|
global_init_was_already_run = true;
|
|
bool success = true;
|
|
success &= internal::Caffe2InitializeRegistry::Registry()
|
|
->RunRegisteredEarlyInitFunctions(pargc, pargv);
|
|
CAFFE_ENFORCE(success,
|
|
"Failed to run some early init functions for caffe2.");
|
|
success &= ParseCaffeCommandLineFlags(pargc, pargv);
|
|
success &= InitCaffeLogging(pargc, *pargv);
|
|
// Print out the current build version.
|
|
VLOG(1) << "Caffe2 build version: " << CAFFE2_BUILD_STRING;
|
|
// All other initialization functions.
|
|
success &= internal::Caffe2InitializeRegistry::Registry()
|
|
->RunRegisteredInitFunctions(pargc, pargv);
|
|
if (!success) {
|
|
global_init_was_already_run = false;
|
|
}
|
|
CAFFE_ENFORCE(success,
|
|
"Failed to run some init functions for caffe2.");
|
|
// TODO: if we fail GlobalInit(), should we continue?
|
|
return success;
|
|
}
|
|
} // namespace caffe2
|