@@ -2588,16 +2588,16 @@ class ConvertAtenGridSamplerOp : public OpConversionPattern<AtenGridSamplerOp> {
2588
2588
};
2589
2589
2590
2590
auto lambdaBorder = [&](OpBuilder &b, Location loc, Value x,
2591
- Value SizeSubOne ) -> Value {
2591
+ Value sizeSubOne ) -> Value {
2592
2592
Value xMaxZero = b.create <arith::MaximumFOp>(loc, x, zeroFloat);
2593
- return b.create <arith::MinimumFOp>(loc, xMaxZero, SizeSubOne );
2593
+ return b.create <arith::MinimumFOp>(loc, xMaxZero, sizeSubOne );
2594
2594
};
2595
2595
2596
2596
auto lambdaPadding = [&](OpBuilder &b, Location loc, int64_t paddingMode,
2597
- Value x, Value SizeSubOne ) -> Value {
2597
+ Value x, Value sizeSubOne ) -> Value {
2598
2598
// Border
2599
2599
if (paddingMode == 1 ) {
2600
- return lambdaBorder (b, loc, x, SizeSubOne );
2600
+ return lambdaBorder (b, loc, x, sizeSubOne );
2601
2601
}
2602
2602
2603
2603
return x;
@@ -2609,7 +2609,10 @@ class ConvertAtenGridSamplerOp : public OpConversionPattern<AtenGridSamplerOp> {
2609
2609
Value interMode = adaptor.getInterpolationMode ();
2610
2610
2611
2611
int64_t paddingModeInt;
2612
- matchPattern (op.getPaddingMode (), m_TorchConstantInt (&paddingModeInt));
2612
+ if (!matchPattern (op.getPaddingMode (), m_TorchConstantInt (&paddingModeInt))) {
2613
+ return failure ();
2614
+ }
2615
+
2613
2616
2614
2617
SmallVector<Value> dynamicSizes{};
2615
2618
if (resultType.isDynamicDim (0 ))
0 commit comments