Merge pull request #56032 from penpornk/cherrypicks_1HAY1

r2.9 cherry-pick: Warn that oneDNN optimizations are experimental on aarch64
tensorflow · May 9, 2022 · 83b37ad · 83b37ad
2 parents c4a94c9 + 1078ba3
commit 83b37ad
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tensorflow/core/util/util.cc b/tensorflow/core/util/util.cc
@@ -165,11 +165,17 @@ bool IsMKLEnabled() {
                    << oneDNN_enabled;
     }
     if (oneDNN_enabled) {
+#ifndef DNNL_AARCH64_USE_ACL
       LOG(INFO) << "oneDNN custom operations are on. "
                 << "You may see slightly different numerical results due to "
                 << "floating-point round-off errors from different computation "
                 << "orders. To turn them off, set the environment variable "
                 << "`TF_ENABLE_ONEDNN_OPTS=0`.";
+#else
+      LOG(INFO) << "Experimental oneDNN custom operations are on. "
+                << "If you experience issues, please turn them off by setting "
+                << "the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.";
+#endif  // !DNNL_AARCH64_USE_ACL
     }
   });
   return oneDNN_enabled;