Skip to content

Commit

Permalink
Generate compute graph in json format
Browse files Browse the repository at this point in the history
  • Loading branch information
RCoeurjoly committed May 24, 2024
1 parent 596f6de commit df69954
Show file tree
Hide file tree
Showing 3 changed files with 290 additions and 227 deletions.
1 change: 1 addition & 0 deletions kernel/functional.h
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,7 @@ struct ComputeGraph
dict<int, SparseAttr> sparse_attrs;

public:
const std::vector<Node>& get_nodes() const { return nodes; }
template<typename Graph>
struct BaseRef
{
Expand Down
Loading

0 comments on commit df69954

Please sign in to comment.