Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

r2.9 cherry-pick: Warn that oneDNN optimizations are experimental on aarch64 #56032

Merged
merged 1 commit into from May 9, 2022
+6 −0
Merged

r2.9 cherry-pick: Warn that oneDNN optimizations are experimental on aarch64 #56032

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
6 changes: 6 additions & 0 deletions tensorflow/core/util/util.cc
View file
Open in desktop
Expand Up @@ -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;
Expand Down