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

create a single oneDNN string

Browse Source
This commit is contained in:
nammbash 2020-06-16 12:02:14 -07:00
parent 15b97dd44a
commit 266387ecb4
1 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -138,16 +138,9 @@ void InfoAboutUnusedCPUFeatures() {
CheckIfFeatureUnused(CPUFeature::FMA, "FMA", missing_instructions); CheckIfFeatureUnused(CPUFeature::FMA, "FMA", missing_instructions);
#endif // __FMA__ #endif // __FMA__
#endif // else of if defined(_MSC_VER) && !defined(__clang__) #endif // else of if defined(_MSC_VER) && !defined(__clang__)
if (!missing_instructions.empty()) {
string intel_library_official_name(
"Intel(R) oneAPI Deep Neural Network Library (oneDNN) ");
#ifndef INTEL_MKL
intel_library_official_name = "oneAPI Deep Neural Network Library (oneDNN) ";
#endif
if (!missing_instructions.empty()) {
LOG(INFO) << "This TensorFlow binary is optimized with " LOG(INFO) << "This TensorFlow binary is optimized with "
<< intel_library_official_name << "oneAPI Deep Neural Network Library (oneDNN)"
<< "to use the following CPU instructions in performance-" << "to use the following CPU instructions in performance-"
<< "critical operations: " << missing_instructions << std::endl << "critical operations: " << missing_instructions << std::endl
<< "To enable them in other operations, rebuild TensorFlow " << "To enable them in other operations, rebuild TensorFlow "