graph: bugfix: compute a node many times (#88)

microsoft · Feb 20, 2019 · 90643cb · 90643cb
1 parent 3079787
commit 90643cb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/execution-plan.ts b/lib/execution-plan.ts
@@ -121,6 +121,7 @@ export class ExecutionPlan {
         });
 
         // resolve downstream nodes
+        const downstreamNodes = new Set<number>();
         outputList.forEach((output, i) => {
           const j = thisOp.node.outputs[i];
           for (const currentDownstreamNodeIndex of graphValues[j].to) {
@@ -133,10 +134,11 @@ export class ExecutionPlan {
               }
             }
             if (resolved) {
-              sequence.push(currentDownstreamNodeIndex);
+              downstreamNodes.add(currentDownstreamNodeIndex);
             }
           }
         });
+        sequence.push(...downstreamNodes);
       }
 
       const output: Tensor[] = [];