re-add the accidentally deleted code

XJDKC · XJDKC · commit 373dac1cf816 · 2020-04-08T15:40:34.000Z
diff --git a/src/core/scheduler/scheduler.cc b/src/core/scheduler/scheduler.cc
@@ -396,7 +396,7 @@ void Graph::Analysis() {
         BlockType type = blkInfo->type_;
         // if the block belongs to a inter tensor
         // and isn't refered on the Python Side
-        if ((type == BlockType::kInter) &&
+        if ((type == BlockType::kInter || type == BlockType::kEnd) &&
             blkInfo->graph_ref_ >= blk->ref_count()) {
           free_blocks_[curIndex].push_back(blk);
         }

Original file line number	Diff line number	Diff line change
`@@ -396,7 +396,7 @@ void Graph::Analysis() {`
`396`	`396`	`BlockType type = blkInfo->type_;`
`397`	`397`	`// if the block belongs to a inter tensor`
`398`	`398`	`// and isn't refered on the Python Side`
`399`		`- if ((type == BlockType::kInter) &&`
	`399`	`+ if ((type == BlockType::kInter \|\| type == BlockType::kEnd) &&`
`400`	`400`	`blkInfo->graph_ref_ >= blk->ref_count()) {`
`401`	`401`	`free_blocks_[curIndex].push_back(blk);`
`402`	`402`	`}`