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

[mlir]the ptx compilation flow #45

Open
CONGCONGLEEE opened this issue Mar 9, 2022 · 2 comments
Open

[mlir]the ptx compilation flow #45

CONGCONGLEEE opened this issue Mar 9, 2022 · 2 comments

Comments

@CONGCONGLEEE
Copy link

Recently I'm trying to run the model compiled by onnx-mlir on GPU.
Here is my idea about the compilation flow:
affine/scf/std Dialect -> gpu Dialect ->NVVM IR -> LLVM IR -> ptx assemble.
I have no idea whether it will work. If not, could you please share me some of your points? It would be very nice to get some help with this since I am not familiar with this area myself, thanks!

@gysit
Copy link
Collaborator

gysit commented Mar 9, 2022

I think this is a good question for https://discourse.llvm.org/c/mlir/31 (I have not been working on GPU compilation recently).

The lowering path you propose sounds similar to what we have done in the open earth compiler but their have been a lot of changes since then. Having a look at something like IREE (https://github.com/google/iree) that AFAIK has an end-to-end GPU compilation workflow could help as well.

@CONGCONGLEEE
Copy link
Author

Ok, thank you very much!

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

2 participants