-
Notifications
You must be signed in to change notification settings - Fork 1
/
Version.cpp
156 lines (133 loc) · 3.42 KB
/
Version.cpp
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
#include <ATen/Version.h>
#include <ATen/Config.h>
#if AT_MKL_ENABLED()
#include <mkl.h>
#endif
#if AT_MKLDNN_ENABLED()
#include <mkldnn.hpp>
#include <ideep.hpp>
#endif
#include <caffe2/core/common.h>
#include <sstream>
namespace at {
std::string get_mkl_version() {
std::string version;
#if AT_MKL_ENABLED()
{
// Magic buffer number is from MKL documentation
// https://software.intel.com/en-us/mkl-developer-reference-c-mkl-get-version-string
char buf[198];
mkl_get_version_string(buf, 198);
version = buf;
}
#else
version = "MKL not found";
#endif
return version;
}
std::string get_mkldnn_version() {
std::ostringstream ss;
#if AT_MKLDNN_ENABLED()
// Cribbed from mkl-dnn/src/common/verbose.cpp
// Too bad: can't get ISA info conveniently :(
// Apparently no way to get ideep version?
// https://github.com/intel/ideep/issues/29
{
const mkldnn_version_t* ver = mkldnn_version();
ss << "Intel(R) MKL-DNN v" << ver->major << "." << ver->minor << "." << ver->patch
<< " (Git Hash " << ver->hash << ")";
}
#else
ss << "MKLDNN not found";
#endif
return ss.str();
}
std::string get_openmp_version() {
std::ostringstream ss;
#ifdef _OPENMP
{
ss << "OpenMP " << _OPENMP;
// Reference:
// https://stackoverflow.com/questions/1304363/how-to-check-the-version-of-openmp-on-linux
const char* ver_str = nullptr;
switch (_OPENMP) {
case 200505:
ver_str = "2.5";
break;
case 200805:
ver_str = "3.0";
break;
case 201107:
ver_str = "3.1";
break;
case 201307:
ver_str = "4.0";
break;
case 201511:
ver_str = "4.5";
break;
default:
ver_str = nullptr;
break;
}
if (ver_str) {
ss << " (a.k.a. OpenMP " << ver_str << ")";
}
}
#else
ss << "OpenMP not found";
#endif
return ss.str();
}
std::string show_config() {
std::ostringstream ss;
ss << "PyTorch built with:\n"; // TODO add the version of PyTorch
// Reference:
// https://blog.kowalczyk.info/article/j/guide-to-predefined-macros-in-c-compilers-gcc-clang-msvc-etc..html
#if defined(__GNUC__)
{
ss << " - GCC " << __GNUC__ << "." << __GNUC_MINOR__ << "\n";
}
#endif
#if defined(__clang_major__)
{
ss << " - clang " << __clang_major__ << "." << __clang_minor__ << "." << __clang_patchlevel__ << "\n";
}
#endif
#if defined(_MSC_VER)
{
ss << " - MSVC " << _MSC_FULL_VER << "\n";
}
#endif
#if AT_MKL_ENABLED()
ss << " - " << get_mkl_version() << "\n";
#endif
#if AT_MKLDNN_ENABLED()
ss << " - " << get_mkldnn_version() << "\n";
#endif
#ifdef _OPENMP
ss << " - " << get_openmp_version() << "\n";
#endif
#ifdef USE_LAPACK
// TODO: Actually record which one we actually picked
ss << " - LAPACK is enabled (usually provided by MKL)\n";
#endif
#if AT_NNPACK_ENABLED()
// TODO: No version; c.f. https://github.com/Maratyszcza/NNPACK/issues/165
ss << " - NNPACK is enabled\n";
#endif
if (hasCUDA()) {
ss << detail::getCUDAHooks().showConfig();
}
ss << " - Build settings: ";
for (const std::pair<std::string, std::string>& pair : caffe2::GetBuildOptions()) {
if (!pair.second.empty()) {
ss << pair.first << "=" << pair.second << ", ";
}
}
ss << "\n";
// TODO: do HIP
// TODO: do XLA
return ss.str();
}
}