[Relay] RelayViz graphviz renderer#10400
Conversation
Following apache#10085, this PR adds a graphviz backend. It requires python `graphviz` package and `dot` executable in the PATH, similar to `tedd.py`. This implementation is much like a porting of `visualize` function in https://tvm.apache.org/2020/07/14/bert-pytorch-tvm, except that `node_attr_dict` is replaced with a callback `get_node_attr`. `get_node_attr` can be somehow used to emphasize a set of nodes. It might be useful if we encounter problems in inferences and want to find nodes with certain types and attributes. An example is provided in https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/test_viz.py Its outputs are (conv2d with NCHW layout is green-colored): https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/mod_with_subgraph.pdf https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/mod_wo_subgraph.pdf
chiwwang
requested review from
a team,
areusch,
comaniac,
icemelon,
jroesch,
junrushao,
merrymercy,
tqchen,
yzhliu and
zhiics
as code owners
comaniac
left a comment
comaniac
left a comment
There was a problem hiding this comment.
Great work! btw would you add this rendering approach to the document as well?
echuraev
left a comment
echuraev
left a comment
There was a problem hiding this comment.
Thank you! LGTM. Nice work!
@comaniac Yes, it is already added in contrib.rst. But I didn't write this in tutorial, only added one comment. |
|
Ok let's defer doc change to a follow-up and let @chiwwang continue. |
|
I just realized that we haven't updated the doc website for a while so this tutorial is even not there yet 😂\ UPDATE: I just updated the TVM site based on the main branch commit b55997d |
Following apache#10085, this PR adds a graphviz backend. It requires python `graphviz` package and `dot` executable in the PATH, similar to `tedd.py`. This implementation is much like a porting of `visualize` function in https://tvm.apache.org/2020/07/14/bert-pytorch-tvm, except that `node_attr_dict` is replaced with a callback `get_node_attr`. `get_node_attr` can be somehow used to emphasize a set of nodes. It might be useful if we encounter problems in inferences and want to find nodes with certain types and attributes. An example is provided in https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/test_viz.py Its outputs are (conv2d with NCHW layout is green-colored): https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/mod_with_subgraph.pdf https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/mod_wo_subgraph.pdf
5 participants
Following #10085, this PR adds a graphviz backend.
It requires python
graphvizpackage anddotexecutable in the PATH, similar totedd.py.This implementation is a porting of
visualizefunction inhttps://tvm.apache.org/2020/07/14/bert-pytorch-tvm, plus a callback
get_node_attr.get_node_attrreturn specific graphviz-attributes for the node.It might be useful if we want to emphasize some nodes with certain types and attributes.
An example is provided in
https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/test_viz.py
Its outputs are:
https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/mod_with_subgraph.pdf
https://github.com/chiwwang/tvm/blob/graphviz_renderer_example/mod_wo_subgraph.pdf
Note that
nn.conv2dwith NCHW layout is green-colored.Thanks for contributing to TVM! Please refer to guideline https://tvm.apache.org/docs/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers by @ them in the pull request thread.