Add Check about negative uint constant (#10484)

* fix InferType bug * fix InferType related bug * check if uint variable is negative * check if uint variable is negative * check if uint variable is negative * check if uint variable is negative * check if uint variable is negative * check if uint variable is negative * check if uint variable is negative * check if uint variable is negative * check if uint variable is negative
apache · Mar 8, 2022 · f4b74ba · f4b74ba
1 parent 704ac8b
commit f4b74ba
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/arith/const_fold.h b/src/arith/const_fold.h
@@ -108,6 +108,9 @@ inline PrimExpr TryConstFold<tir::Add>(PrimExpr a, PrimExpr b) {
 template <>
 inline PrimExpr TryConstFold<tir::Sub>(PrimExpr a, PrimExpr b) {
   TVM_ARITH_CONST_PROPAGATION({
+    ICHECK(!(pa && pa->dtype.is_uint() && pa->value == 0U && b.dtype().is_uint()))
+        << "Checked failed. Minuend 's value is 0U and it's dtype is uint "
+        << "while Subtrahend's dtype is uint; which will cause a negative uint";
     const DataType& rtype = a.dtype();
     if (pa && pb) return IntImm(rtype, pa->value - pb->value);
     if (pb && pb->value == 0) return a;