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

CPU Feature Guard MKL brand needs to change to oneDNNchange to one dnn

Browse Source
This commit is contained in:
nammbash 2020-06-09 11:25:45 -07:00
parent 8dffa4de1b
commit 15b97dd44a
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -138,8 +138,16 @@ 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()) {
LOG(INFO) << "This TensorFlow binary is optimized with Intel(R) MKL-DNN " 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 "
<< intel_library_official_name
<< "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 "