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

[WIP] ONNX demo #2114

Closed
wants to merge 152 commits into from
Closed

[WIP] ONNX demo #2114

wants to merge 152 commits into from

Conversation

superlopuh
Copy link
Member

@superlopuh superlopuh commented Feb 8, 2024

This Draft PR tracks all the changes necessary to lower an ONNX graph that adds two tensors to linalg.generic representation. The objective is to bring the diff in xDSL down to 0 and then close this pull request. I'll comment on the bits of code with suggestions for how to get things in, would love some help from @kayode-gif and @alecerio who already have experience with ONNX.

pip install -e ".[extras,onnx]"
marimo edit onnx-frontend.py

Remaining tasks:

  • add callable support in the interpreter and handle registers in calling convention to fix failing test
  • merge sub lowering
  • merge alloc lowering

onnx-frontend.py Outdated Show resolved Hide resolved
hello.mlir Outdated Show resolved Hide resolved
xdsl/dialects/linalg.py Outdated Show resolved Hide resolved
xdsl/dialects/linalg.py Outdated Show resolved Hide resolved
xdsl/dialects/linalg.py Outdated Show resolved Hide resolved
xdsl/dialects/onnx.py Outdated Show resolved Hide resolved
@superlopuh
Copy link
Member Author

superlopuh commented Jul 26, 2024

All the functionality is now merged to either the ONNX notebook or the one going from linalg to snitch, closing!

@superlopuh superlopuh closed this Jul 26, 2024
@tobiasgrosser
Copy link
Contributor

Lovely. Thank you!

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