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

[WIP]benchdnn: graph: enhance input displace and shape rewrite for linked attribute and shapes #2354

Draft
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

wzt1997
Copy link
Contributor

@wzt1997 wzt1997 commented Jan 8, 2025

Description

The PR enhances input displacement and rewrite functionality in benchdnn graph for the following aspects:

  1. Support mb rewrite on SRC1 of MatMul and scale and zp of DynamicDequantize to support SDPA patterns rewriting.
  2. Support shape rewrite for linked attributes and shapes, such as group_shape and scale/zp input of DynamicDequantize. If user provides shapes for one of the attributes or input shapes, benchdnn graph will update the other accordingly after performing some checks.
  3. Fix the data type setting for input displacement in case the primitive cannot be created.

@wzt1997 wzt1997 self-assigned this Jan 8, 2025
@github-actions github-actions bot added component:graph-api Codeowner: @oneapi-src/onednn-graph component:tests Codeowner: @oneapi-src/onednn-arch labels Jan 8, 2025
@wzt1997 wzt1997 force-pushed the zhitao/enhance-shape-rewrite branch from 4af44a1 to d65e846 Compare January 8, 2025 06:09
@wzt1997 wzt1997 changed the title [WIP]benchdnn: graph: enhance mb and shape rewrite [WIP]benchdnn: graph: enhance input displace and shape rewrite for linked attribute and shapes Jan 8, 2025
@wzt1997 wzt1997 force-pushed the zhitao/enhance-shape-rewrite branch from 50714b2 to c82a899 Compare January 8, 2025 08:02
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 component:tests Codeowner: @oneapi-src/onednn-arch
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant