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 conv fp16 kernel in xnnpack EP #22301

Merged
merged 25 commits into from
Oct 10, 2024
Merged

Add conv fp16 kernel in xnnpack EP #22301

merged 25 commits into from
Oct 10, 2024

Conversation

mszhanyi
Copy link
Contributor

@mszhanyi mszhanyi commented Oct 3, 2024

Description

Add FP16 kernels of Conv and ConvTranspose
AB#50186

Motivation and Context

@mszhanyi mszhanyi marked this pull request as draft October 3, 2024 13:13
@mszhanyi mszhanyi marked this pull request as ready for review October 3, 2024 14:28
@mszhanyi mszhanyi requested review from skottmckay and wejoncy October 4, 2024 05:40
onnxruntime/core/providers/xnnpack/detail/utils.cc Outdated Show resolved Hide resolved
onnxruntime/core/providers/xnnpack/detail/utils.cc Outdated Show resolved Hide resolved
onnxruntime/core/providers/xnnpack/detail/utils.cc Outdated Show resolved Hide resolved
onnxruntime/core/providers/xnnpack/nn/conv.cc Outdated Show resolved Hide resolved
onnxruntime/core/providers/xnnpack/nn/conv.cc Outdated Show resolved Hide resolved
onnxruntime/core/providers/xnnpack/nn/conv.cc Outdated Show resolved Hide resolved
onnxruntime/core/providers/xnnpack/nn/conv_base.cc Outdated Show resolved Hide resolved
onnxruntime/core/providers/xnnpack/nn/conv_transpose.cc Outdated Show resolved Hide resolved
onnxruntime/test/providers/checkers.cc Outdated Show resolved Hide resolved
@mszhanyi mszhanyi requested a review from skottmckay October 7, 2024 02:17
@mszhanyi mszhanyi marked this pull request as draft October 7, 2024 03:50
@mszhanyi mszhanyi marked this pull request as ready for review October 7, 2024 04:02
@mszhanyi mszhanyi requested review from skottmckay and wejoncy October 9, 2024 11:34
@mszhanyi mszhanyi merged commit 25b1c38 into main Oct 10, 2024
89 of 91 checks passed
@mszhanyi mszhanyi deleted the zhanyi/xnnconvfp16 branch October 10, 2024 00:48
ishwar-raut1 pushed a commit to ishwar-raut1/onnxruntime that referenced this pull request Nov 19, 2024
### Description
Add FP16 kernels of Conv and ConvTranspose

[AB#50186](https://aiinfra.visualstudio.com/6a833879-cd9b-44a4-a9de-adc2d818f13c/_workitems/edit/50186)



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->

---------
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