Print different oneDNN warnings for x86 and aarch64 backends.

PiperOrigin-RevId: 447508799
tensorflow · May 9, 2022 · 1078ba3 · 1078ba3
1 parent c4a94c9
commit 1078ba3
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;