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

Add function to visualize fx graphs #84

Merged
merged 4 commits into from
Aug 16, 2024
Merged

Conversation

harsh-nod
Copy link
Contributor

This PR adds graphviz-based functionality to
visualize fx graphs. This is a useful graphical tool for debugging graph transformations.

@harsh-nod
Copy link
Contributor Author

Here is the picture you get if you have pygraphviz installed.

gemm

This PR adds graphviz-based functionality to
visualize fx graphs. This is a useful graphical tool
for debugging graph transformations.

Signed-off-by: Harsh Menon <harsh@nod-labs.com>
Signed-off-by: Harsh Menon <harsh@nod-labs.com>
shark_turbine/kernel/wave/visualization.py Outdated Show resolved Hide resolved
lit_tests/kernel/wave/expansion.py Outdated Show resolved Hide resolved
Signed-off-by: Harsh Menon <harsh@nod-labs.com>
Also add assert to check for image and xfail
test if graphviz is not available.

Signed-off-by: Harsh Menon <harsh@nod-labs.com>
@harsh-nod harsh-nod merged commit eb852f0 into iree-org:main Aug 16, 2024
5 checks passed
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.

2 participants