-
Notifications
You must be signed in to change notification settings - Fork 481
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][TORCH] Add TorchToTosa lowering for aten.where.self op #1454
Conversation
a3a605d
to
2446ece
Compare
https://gist.github.com/AmosLewis/fa8662eef03c3379bda6a3974036abd1 for test this op. |
e89a71f
to
3c5faa6
Compare
In the test, it converts torch to tosa successfully. But why after that, it converts tosa to linalg? The tosa to linalg conversion arise the bug. @sjarus @eric-k256
|
I think the code that generates the indexing maps for the |
I think this line should be using the result rank variable |
Yes. After replacing with rank, everything looks good.
|
b0a1c48
to
b854e20
Compare
llvm to use llvm/llvm-project@main...AmosLewis:llvm-project:tosa-to-linalg llvm patch waiting for review and merge : https://reviews.llvm.org/D135343 |
b854e20
to
4780d3d
Compare
4780d3d
to
5d3ec5a
Compare
Need this merged first. |
@ramiro050 Could you help uplift llvm to d325d2b? |
5d3ec5a
to
1e44fbb
Compare
1e44fbb
to
ff1d72a
Compare
Finish rebase and uplift the llvm to #1502.
|
Is this happening locally on your machine? The CI seems to be building fine. |
It was happening in CI (because of a bogus cache entry) so I deleted that cache entry and restarted the build, which then passed. All is fine now, but I'll keep an eye out for more failures. |
Nice! Thanks for the quick fix! |
Here is the link to useful torch_mlir_debug_command.txt that will be helpful for boot camp and refer later in development of torch-mlir. |
Find this issue when lowering gpt2 to tosa dialect in SHARK with python nod-ai/SHARK#338
If you are boot camp torch-mlir:
Here is useful comments to understand the debug step: #961 (comment)
Here is the link to useful torch_mlir_debug_command.txt that will be helpful for boot camp and refer later in development of torch-mlir.
After you read all the following comments, for ONNX work, refer to this dc9ea08