added check for out paddings

Author: a-sidorova

lib/Conversion/TorchToLinalg/Linear.cpp

@@ -1743,30 +1743,32 @@ class ConvertAtenConvolutionBackwardOp : public OpConversionPattern<AtenConvolut
 
     // Stride, padding, dilation for the backward conv. We only support constant
     // lists here, consistent with forward convolution lowering.
-    SmallVector<int64_t> strideInts;
+    SmallVector<Value> paddingIntValues;
+    SmallVector<int64_t> strideInts, dilationInts, outputPaddingInts;
+
     if (!matchPattern(op.getStride(), m_TorchListOfConstantInts(strideInts)))
       return rewriter.notifyMatchFailure(op,
                                          "only support constant int strides");
-    SmallVector<int64_t> dilationInts;
+
     if (!matchPattern(op.getDilation(),
                       m_TorchListOfConstantInts(dilationInts)))
       return rewriter.notifyMatchFailure(op,
                                          "only support constant int dilations");
-    SmallVector<Value> paddingIntValues, outputPaddingIntValues;
+
+    if (!matchPattern(op.getOutputPadding(),
+                      m_TorchListOfConstantInts(outputPaddingInts)))
+      return rewriter.notifyMatchFailure(op,
+                                         "only support constant int output paddings");
+    if (!llvm::all_of(outputPaddingInts,
+                      [](int64_t outPad) { return outPad == 0; }))
+    return rewriter.notifyMatchFailure(
+        op, "unimplemented: only output padding of 0 supported.");
+
     if (!getListConstructElements(op.getPadding(), paddingIntValues))
       return rewriter.notifyMatchFailure(
           op, "only support padding from a list construct");
     paddingIntValues = getTypeConvertedValues(rewriter, loc, getTypeConverter(),
                                               paddingIntValues);
-    if (!getListConstructElements(op.getOutputPadding(), outputPaddingIntValues))
-      return rewriter.notifyMatchFailure(
-          op, "only support output padding from a list construct");
-    outputPaddingIntValues = getTypeConvertedValues(rewriter, loc, getTypeConverter(),
-                                                    outputPaddingIntValues);
-    //if (!llvm::all_of(outputPaddingIntValues,
-    //                  [](int64_t outPad) { return outPad == 0; }))
-    //  return rewriter.notifyMatchFailure(
-    //      op, "unimplemented: only output padding of 0 supported.");
 
     // The expandGroups lambda function below is used to expand the group
     // dimension for weights and input, output tensors.