Handle Select in IntSetEvaluator (apache#2687)

neo-ai · Mar 12, 2019 · 6fee9f6 · 6fee9f6
1 parent fad2597
commit 6fee9f6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/arithmetic/int_set.cc b/src/arithmetic/int_set.cc
@@ -531,6 +531,11 @@ class IntSetEvaluator :
     CHECK(eval_vec_);
     return Eval(op->value);
   }
+  IntSet VisitExpr_(const Select* op, const Expr& e) final {
+    IntSet true_set = this->Eval(op->true_value);
+    IntSet false_set = this->Eval(op->false_value);
+    return Union({false_set, true_set});
+  }
   IntSet VisitExprDefault_(const Node* op, const Expr& e) final {
     LOG(WARNING) << "cannot evaluate set type " << e->type_key();
     return IntSet::everything();