3 #include "caffe2/core/common.h" 10 std::atomic<bool> g_caffe2_has_cuda_linked{
false};
12 bool HasCudaRuntime() {
13 return g_caffe2_has_cuda_linked.load();
17 void SetCudaRuntimeFlag() {
18 g_caffe2_has_cuda_linked.store(
true);
22 const std::map<string, string>& GetBuildOptions() {
23 #ifndef CAFFE2_BUILD_STRINGS 24 #define CAFFE2_BUILD_STRINGS {} 26 static const std::map<string, string> kMap = CAFFE2_BUILD_STRINGS;
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...