[Arith] Support dtype promotion in TIR comparison expr creation

masahi · masahi · commit d399ada5f219 · 2022-03-12T08:29:50.000+09:00
diff --git a/src/arith/int_constraints.cc b/src/arith/int_constraints.cc
@@ -49,13 +49,13 @@ Array<PrimExpr> AsConditions(const Array<Var>& variables, const Map<Var, IntGrou
     const auto& bnds = bounds[v];
     PrimExpr lhs = bnds->coef * v;
     for (const PrimExpr& rhs : bnds->equal) {
-      res.push_back(tir::EQ(lhs, rhs));
+      res.push_back(lhs == rhs);
     }
     for (const PrimExpr& rhs : bnds->lower) {
-      res.push_back(tir::GE(lhs, rhs));
+      res.push_back(lhs >= rhs);
     }
     for (const PrimExpr& rhs : bnds->upper) {
-      res.push_back(tir::LE(lhs, rhs));
+      res.push_back(lhs <= rhs);
     }
   }
   for (const PrimExpr& e : relations) {

Original file line number	Diff line number	Diff line change
`@@ -49,13 +49,13 @@ Array<PrimExpr> AsConditions(const Array<Var>& variables, const Map<Var, IntGrou`
`49`	`49`	`const auto& bnds = bounds[v];`
`50`	`50`	`PrimExpr lhs = bnds->coef * v;`
`51`	`51`	`for (const PrimExpr& rhs : bnds->equal) {`
`52`		`- res.push_back(tir::EQ(lhs, rhs));`
	`52`	`+ res.push_back(lhs == rhs);`
`53`	`53`	`}`
`54`	`54`	`for (const PrimExpr& rhs : bnds->lower) {`
`55`		`- res.push_back(tir::GE(lhs, rhs));`
	`55`	`+ res.push_back(lhs >= rhs);`
`56`	`56`	`}`
`57`	`57`	`for (const PrimExpr& rhs : bnds->upper) {`
`58`		`- res.push_back(tir::LE(lhs, rhs));`
	`58`	`+ res.push_back(lhs <= rhs);`
`59`	`59`	`}`
`60`	`60`	`}`
`61`	`61`	`for (const PrimExpr& e : relations) {`