Partition fix with rfactor, simplify and likely predicates. (#3444)

5d068357 · Christian Sarofeen · Lianmin Zheng · eb056bd4 · 5d068357
Commit 5d068357 authored Jun 29, 2019 by Christian Sarofeen Committed by Lianmin Zheng Jun 29, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

src/schedule/schedule_dataflow_rewrite.cc
+1 -1

No files found.
--- a/src/schedule/schedule_dataflow_rewrite.cc
+++ b/src/schedule/schedule_dataflow_rewrite.cc
@@ -735,7 +735,7 @@ Array<Tensor> Schedule::rfactor(const Tensor& tensor,
  const Reduce* reduce = compute_op->body[idx].as<Reduce>();
  CHECK(reduce) << "Can only rfactor non-inline reductions";
  predicates.push_back(reduce->condition);
-  Expr predicate = arith::ComputeReduce<ir::And>(predicates, Expr());
+  Expr predicate = likely(simplify(arith::ComputeReduce<ir::And>(predicates, Expr())));

  std::unordered_map<const Variable*, Expr> vsub;