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

Create flag to print final mlir module #3535

Open
causten opened this issue Oct 16, 2024 · 1 comment
Open

Create flag to print final mlir module #3535

causten opened this issue Oct 16, 2024 · 1 comment
Assignees

Comments

@causten
Copy link
Collaborator

causten commented Oct 16, 2024

Add a parameter for migraphx-driver to print the mlir fusion graphs per model. MIGraphX today already has the ability to print MLIR code objects so this will be to identify the winning one since today; using an environment variable, it prints them all during the tuning phase. Do not print all, just the ones that will be run in the model. The driver should export each chosen fusion graph via in a well structured manner for rocmlir to import the file(s) (.mlir?).

rocmlir will compile them (ensuring that a single code object is generated), verify them, and benchmark them.

@pfultz2
Copy link
Collaborator

pfultz2 commented Oct 16, 2024

We already have a flag to print the mlir modules. Instead, we should have env variable to a path to dump each mlir program to an individual file. #3182 dumps them to mxrs, we just need to another variable to dump the mlir instead. Either way, #3182 needs to be merged in first.

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

No branches or pull requests

3 participants