fix bug

EnzymeAD · Mar 8, 2024 · 3aacc35 · 3aacc35
1 parent 10e312d
commit 3aacc35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp b/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp
@@ -202,7 +202,7 @@ struct AddPad final : OpRewritePattern<mlir::stablehlo::AddOp> {
 
           if (lhs.getEdgePaddingHigh()[idx] != 0) {
           starts[idx] = type.getShape()[idx] - lhs.getEdgePaddingHigh()[idx];
-          limits[idx] = 0;
+          limits[idx] = type.getShape()[idx];
           auto postSlice = rewriter.create<stablehlo::SliceOp>(op.getLoc(), rhs, starts, limits, strides);
           vals.push_back(postSlice);
           }