Skip to content

[QNN EP] Reshape Transpose pattern fusion for 6D tensors #26296

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.

Sign up for GitHub

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

Jump to bottom
Closed
wants to merge 3 commits into from
Closed

[QNN EP] Reshape Transpose pattern fusion for 6D tensors #26296

wants to merge 3 commits into from

Conversation

quic-ashwshan
Copy link
Contributor

Description

  • Added qnn node group for pattern Reshape -> Transpose -> Reshape with reshape node giving output as rank 6.
  • Added support for QDQ pattern and added unit test to verify 6D tensor offload on NPU with this pattern

Motivation and Context

  • Ensure Reshape Transpose Reshape pattern gets offloaded to the NPU since QNN Does not have support for 6D tensors.

@quic-ashwshan
[QNN EP] Reshape Transpose pattern fusion for 6D tensors
938db14 
- Added qnn node group for pattern Reshape -> Transpose -> Reshape with
  reshape node giving output as rank 6.
- Added support for QDQ pattern and added unit test to verify 6D tensor
  offload on NPU with this pattern
@quic-ashwshan
- Relaxed condition to get unit dimension at any index
701d2c3 
- Minor fixes
@quic-ashwshan quic-ashwshan marked this pull request as ready for review October 15, 2025 19:28
@chilo-ms
Copy link
Contributor

/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 Azure Pipelines
Copy link

Azure Pipelines successfully started running 4 pipeline(s).

@chilo-ms
Copy link
Contributor

Could you help merge main to include the fix to pass some failing pipelines?
ModuleNotFoundError: No module named 'onnxscript._framework_apis.torch_2_9'

@quic-ashwshan
Merge branch 'master' of github.com:CodeLinaro/onnxruntime into dev/q…
2b83164 
…uic-ashwshan/reshape_transpose_rank6torank5_fusion
@quic-ashwshan
Copy link
Contributor Author

Cannot merge mainline with open source email. Creating new PR with latest mainline

@quic-ashwshan
Copy link
Contributor Author

Pease merge this PR - #26338

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@quic-ashwshan @chilo-ms