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
|
/* Copyright 2016 The TensorFlow Authors. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include "tensorflow/core/platform/cpu_feature_guard.h"
#include <mutex>
#include <string>
#include "tensorflow/core/platform/cpu_info.h"
#include "tensorflow/core/platform/logging.h"
namespace tensorflow {
namespace port {
namespace {
// If the CPU feature isn't present, log a fatal error.
void CheckFeatureOrDie(CPUFeature feature, const string& feature_name) {
if (!TestCPUFeature(feature)) {
#ifdef __ANDROID__
// Some Android emulators seem to indicate they don't support SSE, so to
// avoid crashes when testing, switch this to a warning.
LOG(WARNING)
#else
LOG(FATAL)
#endif
<< "The TensorFlow library was compiled to use " << feature_name
<< " instructions, but these aren't available on your machine.";
}
}
// Check if CPU feature is inclued in the TensorFlow binary.
void CheckIfFeatureUnused(CPUFeature feature, const string& feature_name,
string& missing_instructions) {
if (TestCPUFeature(feature)) {
missing_instructions.append(" ");
missing_instructions.append(feature_name);
}
}
// Raises an error if the binary has been compiled for a CPU feature (like AVX)
// that isn't available on the current machine. It also warns of performance
// loss if there's a feature available that's not being used.
// Depending on the compiler and initialization order, a SIGILL exception may
// occur before this code is reached, but this at least offers a chance to give
// a more meaningful error message.
class CPUFeatureGuard {
public:
CPUFeatureGuard() {
#ifdef __SSE__
CheckFeatureOrDie(CPUFeature::SSE, "SSE");
#endif // __SSE__
#ifdef __SSE2__
CheckFeatureOrDie(CPUFeature::SSE2, "SSE2");
#endif // __SSE2__
#ifdef __SSE3__
CheckFeatureOrDie(CPUFeature::SSE3, "SSE3");
#endif // __SSE3__
#ifdef __SSE4_1__
CheckFeatureOrDie(CPUFeature::SSE4_1, "SSE4.1");
#endif // __SSE4_1__
#ifdef __SSE4_2__
CheckFeatureOrDie(CPUFeature::SSE4_2, "SSE4.2");
#endif // __SSE4_2__
#ifdef __AVX__
CheckFeatureOrDie(CPUFeature::AVX, "AVX");
#endif // __AVX__
#ifdef __AVX2__
CheckFeatureOrDie(CPUFeature::AVX2, "AVX2");
#endif // __AVX2__
#ifdef __AVX512F__
CheckFeatureOrDie(CPUFeature::AVX512F, "AVX512F");
#endif // __AVX512F__
#ifdef __FMA__
CheckFeatureOrDie(CPUFeature::FMA, "FMA");
#endif // __FMA__
}
};
CPUFeatureGuard g_cpu_feature_guard_singleton;
std::once_flag g_cpu_feature_guard_warn_once_flag;
} // namespace
void InfoAboutUnusedCPUFeatures() {
std::call_once(g_cpu_feature_guard_warn_once_flag, [] {
string missing_instructions;
#ifdef PLATFORM_WINDOWS
#ifndef __AVX__
CheckIfFeatureUnused(CPUFeature::AVX, "AVX", missing_instructions);
#endif // __AVX__
#ifndef __AVX2__
CheckIfFeatureUnused(CPUFeature::AVX2, "AVX2", missing_instructions);
#endif // __AVX2__
#else // ifdef platform windows
#ifndef __SSE__
CheckIfFeatureUnused(CPUFeature::SSE, "SSE", missing_instructions);
#endif // __SSE__
#ifndef __SSE2__
CheckIfFeatureUnused(CPUFeature::SSE2, "SSE2", missing_instructions);
#endif // __SSE2__
#ifndef __SSE3__
CheckIfFeatureUnused(CPUFeature::SSE3, "SSE3", missing_instructions);
#endif // __SSE3__
#ifndef __SSE4_1__
CheckIfFeatureUnused(CPUFeature::SSE4_1, "SSE4.1", missing_instructions);
#endif // __SSE4_1__
#ifndef __SSE4_2__
CheckIfFeatureUnused(CPUFeature::SSE4_2, "SSE4.2", missing_instructions);
#endif // __SSE4_2__
#ifndef __AVX__
CheckIfFeatureUnused(CPUFeature::AVX, "AVX", missing_instructions);
#endif // __AVX__
#ifndef __AVX2__
CheckIfFeatureUnused(CPUFeature::AVX2, "AVX2", missing_instructions);
#endif // __AVX2__
#ifndef __AVX512F__
CheckIfFeatureUnused(CPUFeature::AVX512F, "AVX512F", missing_instructions);
#endif // __AVX512F__
#ifndef __FMA__
CheckIfFeatureUnused(CPUFeature::FMA, "FMA", missing_instructions);
#endif // __FMA__
#endif // else of ifdef platform windows
if (!missing_instructions.empty()) {
LOG(INFO) << "Your CPU supports instructions that this TensorFlow "
<< "binary was not compiled to use:" << missing_instructions;
}
});
}
} // namespace port
} // namespace tensorflow
|
