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

Map ops from FFE emit of TTIR to MLIR if it is supported #549

Merged
merged 1 commit into from
Oct 28, 2024

Conversation

kamalrajkannan78
Copy link
Contributor

@kamalrajkannan78 kamalrajkannan78 commented Oct 28, 2024

Mapping done for below ops

  • abs
  • exp
  • maximum
  • less
  • greater( both inputs should have same number of dimensions .if not it will face #1081 or #1080
  • not_equal ( both inputs should have same number of dimensions. if not it will face #1080

Ops and its required unique test cases

op model shapes
abs bart {1, 1, 256, 256}
abs distilbert {1, 1, 1, 128}
abs distilbert {1, 1, 1, 384}
abs gptneo {1, 1, 32, 32}
abs qwen {1, 1, 6, 6}
abs qwen {1, 1, 29, 29}
greater bart [{1, 1, 256, 256}, {1}, ]
greater distilbert [{1, 12, 384, 384}, {1}, ]
greater distilbert [{1, 1, 384, 384}, {1}, ]
greater distilbert [{1, 12, 128, 128}, {1}, ]
greater distilbert [{1, 1, 128, 128}, {1}, ]
greater opt [{1, 1, 32, 32}, {1}, ]
greater openpose [{1, 128, 28, 28}, {1, 128, 28, 28}, ]
exp openpose [{1, 128, 28, 28}, ]
maximum opt [{1, 12, 256, 256}, {1}, ]
maximum opt [{1, 16, 256, 256}, {1}, ]
maximum opt [{1, 32, 256, 256}, {1}, ]
maximum opt [{1, 12, 32, 32}, {1}, ]
maximum opt [{1, 16, 32, 32}, {1}, ]
maximum opt [{1, 32, 32, 32}, {1}, ]
not equal roberta [{1, 128}, {1}, ]
less openpose [{1, 128, 28, 28}, {1, 128, 28, 28}, ]

@kamalrajkannan78 kamalrajkannan78 force-pushed the kkannan/mapping_supported_mlir_ops branch from 9b289af to cc62b8f Compare October 28, 2024 12:52
@kamalrajkannan78 kamalrajkannan78 merged commit 8079c41 into main Oct 28, 2024
5 checks passed
@kamalrajkannan78 kamalrajkannan78 deleted the kkannan/mapping_supported_mlir_ops branch October 28, 2024 13:38
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