Add UIntImm to select rewrite (#909)

2a1e1607 · Meghan Cowan · Tianqi Chen · 6d4eb2e8 · 2a1e1607
Commit 2a1e1607 authored Feb 18, 2018 by Meghan Cowan Committed by Tianqi Chen Feb 18, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 additions and 0 deletions

src/pass/rewrite_unsafe_select.cc
+1 -0

No files found.
--- a/src/pass/rewrite_unsafe_select.cc
+++ b/src/pass/rewrite_unsafe_select.cc
@@ -77,6 +77,7 @@ class UnsafeExprDetector : public ExprFunctor<bool(const Expr& n)> {
    return false;
  }
  bool VisitExpr_(const Variable* op) final { return false; }
+  bool VisitExpr_(const UIntImm* op) final { return false; }
  bool VisitExpr_(const IntImm* op) final { return false; }
  bool VisitExpr_(const FloatImm* op) final { return false; }
  bool VisitExpr_(const StringImm* op) final { return false; }