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.

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

Add limitation for BFLOAT supported ops for NNPA #3008

Merged

Conversation

Sunny-Anand
Copy link
Collaborator

As part of ONNX-1.17.0 update the below list of Ops were modified to now support BFLOAT data types. This PR updates the NNPA supported docs with the limitation for this data type for NNPA.

Update to support bfloat16:

Acos, Acosh, Asin, Asinh, Atan, Atanh, AveragePool, Bernoulli, Conv, ConvTranspose, Cos, Cosh, DeformConv, Det, Dropout, Elu, EyeLike, GRU, GlobalAveragePool, GlobalLpPool, GlobalMaxPool, GridSample, HardSigmoid, HardSwish, InstanceNormalization, LSTM, LpNormalization, LpPool, MaxPool, MaxRoiPool, MaxUnpool, Mish, Multinomial, NegativeLogLikelihoodLoss, RNN, RandomNormal, RandomNormalLike, RandomUniform, RandomUniformLike, RoiAlign, Round, Selu, Sin, Sinh, Softplus, Softsign, Tan, ThresholdedRelu

Copy link
Collaborator

@AlexandreEichenberger AlexandreEichenberger left a comment

Choose a reason for hiding this comment

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

Thanks for the quick update

@AlexandreEichenberger AlexandreEichenberger merged commit 890ebea into onnx:main Nov 13, 2024
7 checks passed
@jenkins-droid
Copy link
Collaborator

Jenkins Linux s390x Build #15984 [push] add limitation for BFLOA... started at 10:59

@jenkins-droid
Copy link
Collaborator

Jenkins Linux amd64 Build #15982 [push] add limitation for BFLOA... started at 10:00

@jenkins-droid
Copy link
Collaborator

Jenkins Linux ppc64le Build #15012 [push] add limitation for BFLOA... started at 11:15

@jenkins-droid
Copy link
Collaborator

Jenkins Linux amd64 Build #15982 [push] add limitation for BFLOA... passed after 1 hr 19 min

@jenkins-droid
Copy link
Collaborator

Jenkins Linux s390x Build #15984 [push] add limitation for BFLOA... passed after 1 hr 27 min

@jenkins-droid
Copy link
Collaborator

Jenkins Linux ppc64le Build #15012 [push] add limitation for BFLOA... failed after 1 hr 44 min

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.

3 participants