-
Notifications
You must be signed in to change notification settings - Fork 3.4k
[QNN EP] Add Support for Reciprocal Op in QNN EP #25035
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
[QNN EP] Add Support for Reciprocal Op in QNN EP #25035
Conversation
- Implemented ReciprocalOpBuilder to support ONNX Reciprocal Op in QNN EP. - Decomposed Reciprocal into a Div Op. - Added unit tests to run Reciprocal Op on HTP
/azp run Linux QNN CI Pipeline,Win_TRT_Minimal_CUDA_Test_CI,Windows ARM64 QNN CI Pipeline,Windows GPU Doc Gen CI Pipeline,Windows x64 QNN CI Pipeline |
Azure Pipelines successfully started running 5 pipeline(s). |
- Implemented ReciprocalOpBuilder to support ONNX Reciprocal Op in QNN EP. - Decomposed Reciprocal into a Div Op. - Added unit tests to run Reciprocal Op on HTP - Added QDQ Selector for Reciprocal Op
- Implemented ReciprocalOpBuilder to support ONNX Reciprocal Op in QNN EP. - Decomposed Reciprocal into a Div Op. - Added unit tests to run Reciprocal Op on HTP - Added QDQ Selector for Reciprocal Op
/azp run Linux QNN CI Pipeline,Win_TRT_Minimal_CUDA_Test_CI,Windows ARM64 QNN CI Pipeline,Windows GPU Doc Gen CI Pipeline,Windows x64 QNN CI Pipeline |
Azure Pipelines successfully started running 5 pipeline(s). |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There is one CPU test of Recirpocal which is failing now since we have added support in EP now. Will fix that and have another patchset. Function - Reciprocal_double is failing as we do not map DOUBLE to float in QNN. |
- Add check for input type to be float for CPU Backend
Resolved. |
/azp run Linux QNN CI Pipeline,Win_TRT_Minimal_CUDA_Test_CI,Windows ARM64 QNN CI Pipeline,Windows GPU Doc Gen CI Pipeline,Windows x64 QNN CI Pipeline |
Azure Pipelines successfully started running 5 pipeline(s). |
onnxruntime/core/providers/qnn/builder/opbuilder/reciprocal_op_builder.cc
Fixed
Show resolved
Hide resolved
- Copyright change
/azp run Linux QNN CI Pipeline,Win_TRT_Minimal_CUDA_Test_CI,Windows ARM64 QNN CI Pipeline,Windows GPU Doc Gen CI Pipeline,Windows x64 QNN CI Pipeline |
Azure Pipelines successfully started running 5 pipeline(s). |
onnxruntime/core/providers/qnn/builder/opbuilder/reciprocal_op_builder.cc
Fixed
Show resolved
Hide resolved
- Lintrunner patch
/azp run Linux QNN CI Pipeline,Win_TRT_Minimal_CUDA_Test_CI,Windows ARM64 QNN CI Pipeline,Windows GPU Doc Gen CI Pipeline,Windows x64 QNN CI Pipeline |
Azure Pipelines successfully started running 5 pipeline(s). |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Description
Adds support for the ONNX Reciprocal operator in QNN EP via Div decomposition.
Motivation and Context
Enables execution of models using Reciprocal on QNN backend, improving Op support.