Change Torch lowering to use divSI

Signed-off-by: MaheshRavishankar <[email protected]>
iree-org · Jan 25, 2025 · ffca362 · ffca362
1 parent 96aaf87
commit ffca362
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/compiler/plugins/input/Torch/InputConversion/BindSymbolicShapes.cpp b/compiler/plugins/input/Torch/InputConversion/BindSymbolicShapes.cpp
@@ -281,11 +281,11 @@ class BindSymbolicShapesPass final
         case AffineExprKind::Mul:
           return builder.create<arith::MulIOp>(loc, lhs, rhs);
         case AffineExprKind::Mod:
-          return builder.create<arith::RemUIOp>(loc, lhs, rhs);
+          return builder.create<arith::RemSIOp>(loc, lhs, rhs);
         case AffineExprKind::FloorDiv:
-          return builder.create<arith::DivUIOp>(loc, lhs, rhs);
+          return builder.create<arith::DivSIOp>(loc, lhs, rhs);
         case AffineExprKind::CeilDiv:
-          return builder.create<arith::CeilDivUIOp>(loc, lhs, rhs);
+          return builder.create<arith::CeilDivSIOp>(loc, lhs, rhs);
         default:
           break;
         }