casts to avoid ambiguity

apache · masahi · Aug 31, 2021 · Aug 23, 2021 · Aug 23, 2021 · Aug 23, 2021
commit 12f7213248d5cf7083b49f0acaf6a5ab7f1bd078
diff --git a/include/tvm/topi/reduction.h b/include/tvm/topi/reduction.h
@@ -445,9 +445,9 @@ inline FCommReduce MakeArgminReducer(bool select_last_index = false) {
     // indices if not select_last_index;
     PrimExpr proper_index;
     if (select_last_index) {
-      proper_index = lhs[0] > rhs[0];
+      proper_index = PrimExpr(lhs[0]) > PrimExpr(rhs[0]);
     } else {
-      proper_index = lhs[0] < rhs[0];
+      proper_index = PrimExpr(lhs[0]) < PrimExpr(rhs[0]);
     }
 
     PrimExpr update_index = is_smaller || (is_same && proper_index);
@@ -500,9 +500,9 @@ inline FCommReduce MakeArgmaxReducer(bool select_last_index = false) {
     // indices if not select_last_index;
     PrimExpr proper_index;
     if (select_last_index) {
-      proper_index = lhs[0] > rhs[0];
+      proper_index = PrimExpr(lhs[0]) > PrimExpr(rhs[0]);
     } else {
-      proper_index = lhs[0] < rhs[0];
+      proper_index = PrimExpr(lhs[0]) < PrimExpr(rhs[0]);
     }
 
     PrimExpr update_index = is_bigger || (is_same && proper_index);