experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix #10199

Browse Source 
PiperOrigin-RevId: 159030457
This commit is contained in:
Gunhan Gulsoy 2017-06-14 14:59:48 -07:00 committed by TensorFlower Gardener
parent f0a8bd95c7
commit 69bc160235
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/core/platform/cpu_feature_guard.cc
View File

@ -96,6 +96,14 @@ std::once_flag g_cpu_feature_guard_warn_once_flag;
void WarnAboutUnusedCPUFeatures() { void WarnAboutUnusedCPUFeatures() {
std::call_once(g_cpu_feature_guard_warn_once_flag, [] { std::call_once(g_cpu_feature_guard_warn_once_flag, [] {
#ifdef PLATFORM_WINDOWS
#ifndef __AVX__
WarnIfFeatureUnused(CPUFeature::AVX, "AVX");
#endif // __AVX__
#ifndef __AVX2__
WarnIfFeatureUnused(CPUFeature::AVX2, "AVX2");
#endif // __AVX2__
#else // ifdef platform windows
#ifndef __SSE__ #ifndef __SSE__
WarnIfFeatureUnused(CPUFeature::SSE, "SSE"); WarnIfFeatureUnused(CPUFeature::SSE, "SSE");
#endif // __SSE__ #endif // __SSE__
@ -123,6 +131,7 @@ void WarnAboutUnusedCPUFeatures() {
#ifndef __FMA__ #ifndef __FMA__
WarnIfFeatureUnused(CPUFeature::FMA, "FMA"); WarnIfFeatureUnused(CPUFeature::FMA, "FMA");
#endif // __FMA__ #endif // __FMA__
#endif // else of ifdef platform windows
}); });
} }