oneDNN version planned for Tensorflow v2.5

[ratan]

Tensorflow v2.4.1 has oneDNN version 1.6.4 integrated. Any idea what oneDNN version will be included in Tensorflow branch v2.5, so that one can know what all new feature coming into corresponding oneDNN version?

[vpirogov]

@ratan,

We plant to upgrade oneDNN to v2.2 in Tensorflow v2.5. This is of course contingent on the PR being accepted by Tensorflow maintainers.

[alenik01]

@vpirogov, thanks for the heads-up, these are great news.

This is of course contingent on the PR being accepted by TensorFlow maintainers.

By the way is there some features list or roadmap from TF side of what is planned to be included in v2.5 release?

[vpirogov]

@alenik01,

Let me bring @agramesh1, Tensorflow enabling lead, to address this question.

[alenik01]

Good morning, @vpirogov.

We plant to upgrade oneDNN to v2.2 in Tensorflow v2.5

the PR#47543 has been recently pushed with the upgrade to v2.1. May you please comment a bit on the plans/timeline for v2.2?

[agramesh1]

@alenik01 We are working on it, the plan is still to update to v2.2 before the branching of TF2.5.

[alenik01]

Good morning, @agramesh1. Many thanks for clarifying this. And for v2.2 there will be a separate PR, correct?
I am asking because we are also preparing a PR to TF at the moment.