Skip to content

fix: turn off tensor flow oneDNN#1321

Merged
SylviaWhittle merged 1 commit intomainfrom
ns-rse/tensor-flow-precision-errors
Apr 7, 2026
Merged

fix: turn off tensor flow oneDNN#1321
SylviaWhittle merged 1 commit intomainfrom
ns-rse/tensor-flow-precision-errors

Conversation

@ns-rse
Copy link
Copy Markdown
Collaborator

@ns-rse ns-rse commented Mar 26, 2026

In #1320 some of the tests failed due to differences in precision.

We have seen this before in other places.

The log messages included...

I0000 00:00:1774527545.928435    2907 port.cc:153] 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`.
I0000 00:00:1774527545.929005    2907 cudart_stub.cc:31] Could not find cuda drivers on your machine, GPU will not be used.

This commit/pull-request therefore sets TF_ENABLE_ONEDNN_OPTS to 0 as suggested.

All tests pass locally (Python 3.11.14; tensorflow-2.18.0) will test against tensorflow-2.21.0 but have to wait for the
0.5GB download.


Before submitting a Pull Request please check the following.

  • Existing tests pass.
  • Pre-commit checks pass.

In #1320 some of the tests failed due to differences in precision.

We have seen this before in other places.

The log messages included...

```
I0000 00:00:1774527545.928435    2907 port.cc:153] 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`.
I0000 00:00:1774527545.929005    2907 cudart_stub.cc:31] Could not find cuda drivers on your machine, GPU will not be used.
```

This commit/pull-request therefore sets `TF_ENABLE_ONEDNN_OPTS` to `0` as suggested.
@ns-rse ns-rse closed this Mar 26, 2026
@ns-rse ns-rse reopened this Mar 26, 2026
@ns-rse
Copy link
Copy Markdown
Collaborator Author

ns-rse commented Mar 26, 2026

Re-running tests and they pass, failures noted in #1322 for future investigation/fixing.

Copy link
Copy Markdown
Collaborator

@SylviaWhittle SylviaWhittle left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yep. Fixes the tests. Sure.

@SylviaWhittle SylviaWhittle added this pull request to the merge queue Apr 7, 2026
Merged via the queue into main with commit b3de806 Apr 7, 2026
12 of 14 checks passed
@SylviaWhittle SylviaWhittle deleted the ns-rse/tensor-flow-precision-errors branch April 7, 2026 09:21
@ns-rse ns-rse mentioned this pull request Apr 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants