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

graph: backend: dnnl: fix some issue about fuse_src_tranpose to_matmul and mqa decompose kernel #2847

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

gyhintel
Copy link
Contributor

Description

  1. Fix the pass fuse_src_transpose_to_matmul, it is aimed to optimize pattern like "softmax+transpose+reshape/reorder+matmul". But the order in the in_md.permute_axes needs to be converted.
  2. fuse_src_transpose_to_matmul is not used in mqa_decomp and sdp_primitive kernels. So removed.
  3. Fix the kernel the dst pointer issue in the mqa_decomp kernel.

@gyhintel gyhintel added the component:graph-api Codeowner: @oneapi-src/onednn-graph label Mar 10, 2025
@gyhintel gyhintel self-assigned this Mar 10, 2025
@gyhintel gyhintel requested a review from a team as a code owner March 10, 2025 14:41
@gyhintel gyhintel force-pushed the gyh/main/fix_fuse_transpose branch from 504d8cd to cb8b163 Compare March 10, 2025 14:52
@gyhintel
Copy link
Contributor Author

make test
set test_scope=NIGHTLY
disable benchdnn_all
enable benchdnn_graph

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
component:graph-api Codeowner: @oneapi-src/onednn-graph
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants