1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
#include <atomic>
#include "caffe2/core/common.h"
namespace caffe2 {
// A global variable to mark if Caffe2 has cuda linked to the current runtime.
// Do not directly use this variable, but instead use the HasCudaRuntime()
// function below.
std::atomic<bool> g_caffe2_has_cuda_linked{false};
std::atomic<bool> g_caffe2_has_hip_linked{false};
bool HasCudaRuntime() {
return g_caffe2_has_cuda_linked.load();
}
bool HasHipRuntime() {
return g_caffe2_has_hip_linked.load();
}
namespace internal {
void SetCudaRuntimeFlag() {
g_caffe2_has_cuda_linked.store(true);
}
void SetHipRuntimeFlag() {
g_caffe2_has_hip_linked.store(true);
}
} // namespace internal
const std::map<string, string>& GetBuildOptions() {
#ifndef CAFFE2_BUILD_STRINGS
#define CAFFE2_BUILD_STRINGS {}
#endif
static const std::map<string, string> kMap = CAFFE2_BUILD_STRINGS;
return kMap;
}
} // namespace caffe2
|
