Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
 into main
  • Loading branch information
parthmannan committed Jan 24, 2023
2 parents 384cf76 + e0beab5 commit 9ed6a91
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ torch.compile(net, backend='nvprims_nvfuser')
```
I am using `nvprims_nvfuser` backend here to get FX Graph with higher level semantics. You can use TorchInductor (default backend) and you'll get a graph with lower level ops. For example, Softmax would be broken down into primitive ops.
Though I don't anticipate any issues, this code has not been tested with TorchInductor.

3. Install pydot package
```
pip install pydot
Expand Down Expand Up @@ -82,4 +83,4 @@ edge_shapes = [nx_g.edges[x]['shape'] for x in nx.edge_bfs(nx_g, source=source_n

2. Graph drawing of the extracted graph
Example below
![Example Graph][examples/fx_graph_extracted_id_count(1).png]
![Example Graph](examples/fx_graph_extracted_id_count(1).png)

0 comments on commit 9ed6a91

Please sign in to comment.