From c5a7e10be062cd58291d78c16507e08fda155119 Mon Sep 17 00:00:00 2001
From: Bob Huang <huangrb@pkueda.org.cn>
Date: Thu, 18 Jul 2024 04:01:03 -0400
Subject: [PATCH] [421]after applying the fix method, re-evaluating DivUIOp
 satisfies the condition.

---
 lib/polygeist/Passes/AffineCFG.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/polygeist/Passes/AffineCFG.cpp b/lib/polygeist/Passes/AffineCFG.cpp
index 7cf74c2a022c..3cdd46baa04d 100644
--- a/lib/polygeist/Passes/AffineCFG.cpp
+++ b/lib/polygeist/Passes/AffineCFG.cpp
@@ -332,7 +332,7 @@ AffineApplyNormalizer::AffineApplyNormalizer(AffineMap map,
              isValidSymbolInt(t.getDefiningOp()->getOperand(1))) &&
             (!(fix(t.getDefiningOp()->getOperand(0), false) &&
                fix(t.getDefiningOp()->getOperand(1), false)))) ||
-           (t.getDefiningOp<DivSIOp>() &&
+           ((t.getDefiningOp<DivUIOp>() || t.getDefiningOp<DivSIOp>()) &&
             (isValidIndex(t.getDefiningOp()->getOperand(0)) &&
              isValidSymbolInt(t.getDefiningOp()->getOperand(1)))) ||
            (t.getDefiningOp<RemUIOp>() &&