fix ci

apache · Dec 12, 2021 · c5c3ba4 · c5c3ba4
1 parent 8f27523
commit c5c3ba4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/relay/ir/indexed_graph.cc b/src/relay/ir/indexed_graph.cc
@@ -61,7 +61,11 @@ IndexedGraph<Expr> CreateIndexedGraph(const Expr& expr) {
       auto post_visit = [&](const LetNode* op) {
         this->VisitExpr(op->body);
         if (let != op) {
+          Expr expr = GetRef<Expr>(op);
           visit_counter_[op]++;
+          auto node = std::make_shared<IndexedGraph<Expr>::Node>(expr, index_++);
+          graph_.node_map_[expr] = node;
+          graph_.topological_order_.push_back(node);
         }
       };
       ExpandANormalForm(let, pre_visit, post_visit);