Skip to content

[CPU BF16] BF16 support was added to the SoftMax primitive.#4279

Merged
dmitry-gorokhov merged 2 commits intoopenvinotoolkit:masterfrom
maxnick:feature/softmax_bf16
Feb 16, 2021
Merged

[CPU BF16] BF16 support was added to the SoftMax primitive.#4279
dmitry-gorokhov merged 2 commits intoopenvinotoolkit:masterfrom
maxnick:feature/softmax_bf16

Conversation

@maxnick
Copy link
Copy Markdown
Contributor

@maxnick maxnick commented Feb 10, 2021

OneDNN fork SoftMax primitive was extended with the bfloat16 support. See ticket 32817.
Corresponding oneDNN PR openvinotoolkit/oneDNN#34.

@maxnick maxnick marked this pull request as ready for review February 11, 2021 08:31
@maxnick maxnick requested review from a team and removed request for a team February 11, 2021 08:31
@maxnick maxnick added the category: CPU OpenVINO CPU plugin label Feb 11, 2021
@dmitry-gorokhov dmitry-gorokhov added this to the 2021.3 milestone Feb 14, 2021
@dmitry-gorokhov
Copy link
Copy Markdown

@maxnick please provide the link to the PR in oneDNN fork.

@maxnick
Copy link
Copy Markdown
Contributor Author

maxnick commented Feb 15, 2021

@maxnick please provide the link to the PR in oneDNN fork.

Done.

Copy link
Copy Markdown

@dmitry-gorokhov dmitry-gorokhov left a comment

Choose a reason for hiding this comment

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

In general looks good to me. Will merge after internal performance validation

@dmitry-gorokhov dmitry-gorokhov merged commit 274a70b into openvinotoolkit:master Feb 16, 2021
@maxnick maxnick deleted the feature/softmax_bf16 branch March 2, 2021 14:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

category: CPU OpenVINO CPU plugin

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants