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

[FXML-4440] Introduce a pass that annotates the type of the argument as an attribute #167

Merged
merged 4 commits into from
Apr 22, 2024

Conversation

josel-amd
Copy link
Collaborator

@josel-amd josel-amd commented Apr 17, 2024

Create a pass that annotates all func op inputs with their type as attributes.

@josel-amd josel-amd requested review from TinaAMD and cferry-AMD April 17, 2024 14:28
@mgehre-amd
Copy link
Collaborator

We will also need to remember the types of the function results. To that end, we might need to find a more generic name of this pass, maybe something like AnnotateFunctionType?

Copy link
Collaborator

@cferry-AMD cferry-AMD left a comment

Choose a reason for hiding this comment

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

Good idea! Like @mgehre-amd said, we should also save the return types.

mlir/lib/Dialect/Func/Transforms/AnnotateInputTypes.cpp Outdated Show resolved Hide resolved
mlir/include/mlir/Dialect/Func/Transforms/Passes.td Outdated Show resolved Hide resolved
mlir/test/Dialect/Func/annotate-types.mlir Outdated Show resolved Hide resolved
mlir/test/Dialect/Func/annotate-types.mlir Outdated Show resolved Hide resolved
mlir/test/Dialect/Func/annotate-types.mlir Outdated Show resolved Hide resolved
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, only some nitpicks

Copy link
Collaborator

@cferry-AMD cferry-AMD left a comment

Choose a reason for hiding this comment

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

Looks good!

mlir/test/Dialect/Func/annotate-types.mlir Show resolved Hide resolved
mlir/lib/Dialect/Func/Transforms/AnnotateFunctionType.cpp Outdated Show resolved Hide resolved
@josel-amd josel-amd merged commit f799862 into feature/fused-ops Apr 22, 2024
4 checks passed
@josel-amd josel-amd deleted the jose.FXML-4440 branch April 22, 2024 11:41
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.

4 participants