Skip to content

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

Closed
quic-ashwshan wants to merge 3 commits into
microsoft:mainfrom
CodeLinaro:dev/quic-ashwshan/reshape_transpose_rank6torank5_fusion
Closed

[QNN EP] Reshape Transpose pattern fusion for 6D tensors#26296
quic-ashwshan wants to merge 3 commits into
microsoft:mainfrom
CodeLinaro:dev/quic-ashwshan/reshape_transpose_rank6torank5_fusion

Conversation

@quic-ashwshan

Copy link
Copy Markdown
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.

- 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
quic-ashwshan marked this pull request as ready for review October 15, 2025 19:28
@chilo-ms

Copy link
Copy Markdown
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

Copy link
Copy Markdown
Azure Pipelines successfully started running 4 pipeline(s).

@chilo-ms

Copy link
Copy Markdown
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'

…uic-ashwshan/reshape_transpose_rank6torank5_fusion
@quic-ashwshan

Copy link
Copy Markdown
Contributor Author

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

@quic-ashwshan

Copy link
Copy Markdown
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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants