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

TosaLayerwiseConstantFoldPass: speed up folding of transpose for bf16 #75

Merged
merged 1 commit into from
Sep 4, 2023

Conversation

mgehre-amd
Copy link
Collaborator

This speeds up the TosaLayerwiseConstantFoldPass on an important benchmark from 86 seconds to 7 seconds.

@mgehre-amd mgehre-amd requested review from ljfitz and TinaAMD August 30, 2023 13:51
mlir/lib/Dialect/Tosa/Transforms/TosaFolders.cpp Outdated Show resolved Hide resolved
mlir/lib/Dialect/Tosa/Transforms/TosaFolders.cpp Outdated Show resolved Hide resolved
mlir/lib/Dialect/Tosa/Transforms/TosaFolders.cpp Outdated Show resolved Hide resolved
mlir/include/mlir/IR/BuiltinAttributes.h Outdated Show resolved Hide resolved
@mgehre-amd mgehre-amd force-pushed the matthias.tosa_layerwise_raw branch 2 times, most recently from 4d7c18b to 8ca3aab Compare August 31, 2023 06:57
@mgehre-amd mgehre-amd force-pushed the matthias.tosa_layerwise_raw branch from 8ca3aab to 6793e8c Compare August 31, 2023 15:45
Copy link

@TinaAMD TinaAMD left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@mgehre-amd mgehre-amd merged commit acaf1f1 into feature/fused-ops Sep 4, 2023
1 check passed
@mgehre-amd mgehre-amd deleted the matthias.tosa_layerwise_raw branch September 4, 2023 08:01
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.

3 participants