From ce78c49c6965fb9a1c305ac1b349c0087e856a8d Mon Sep 17 00:00:00 2001
From: Balint Cristian <cristian.balint@gmail.com>
Date: Thu, 17 Oct 2024 13:20:26 +0300
Subject: [PATCH] [LLVM][Arith] Presburger compile fix for MLIR/LLVM 19.x

---
 src/arith/presburger_set.cc | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/arith/presburger_set.cc b/src/arith/presburger_set.cc
index 3798ba190446..4f4d7e18578f 100644
--- a/src/arith/presburger_set.cc
+++ b/src/arith/presburger_set.cc
@@ -215,7 +215,9 @@ PresburgerSet Intersect(const Array<PresburgerSet>& sets) {
 
 IntSet EvalSet(const PrimExpr& e, const PresburgerSet& set) {
   Array<PrimExpr> tvm_coeffs = DetectLinearEquation(e, set->GetVars());
-#if TVM_MLIR_VERSION >= 160
+#if TVM_MLIR_VERSION >= 190
+  SmallVector<llvm::DynamicAPInt> coeffs;
+#elif TVM_MLIR_VERSION >= 160
   SmallVector<mlir::presburger::MPInt> coeffs;
 #else
   SmallVector<int64_t> coeffs;
@@ -223,7 +225,9 @@ IntSet EvalSet(const PrimExpr& e, const PresburgerSet& set) {
 
   coeffs.reserve(tvm_coeffs.size());
   for (const PrimExpr& it : tvm_coeffs) {
-#if TVM_MLIR_VERSION >= 160
+#if TVM_MLIR_VERSION >= 190
+    coeffs.push_back(llvm::DynamicAPInt(*as_const_int(it)));
+#elif TVM_MLIR_VERSION >= 160
     coeffs.push_back(mlir::presburger::MPInt(*as_const_int(it)));
 #else
     coeffs.push_back(*as_const_int(it));