added group support for input gradient

Author: a-sidorova

lib/Conversion/TorchToLinalg/Linear.cpp

@@ -1921,6 +1921,56 @@ class ConvertAtenConvolutionBackwardOp : public OpConversionPattern<AtenConvolut
                            IT::reduction, IT::reduction, IT::reduction,
                            IT::reduction};
         }
+      } else {
+        if (numSpatialDims == 1) {
+          AffineExpr n, g, cg, o, fg, k;
+          bindDims(context, n, g, cg, o, fg, k);
+          AffineExpr d0 = rewriter.getAffineConstantExpr(dilationInts[0]);
+          SmallVector<AffineExpr> goExprs = {n, g, fg, d0 * k + o};
+          SmallVector<AffineExpr> weiExprs = {g, fg, cg, k};
+          SmallVector<AffineExpr> outExprs = {n, g, cg, o};
+          indexingMaps = {
+              AffineMap::get(6, 0, goExprs, context),
+              AffineMap::get(6, 0, weiExprs, context),
+              AffineMap::get(6, 0, outExprs, context)};
+          iteratorTypes = {IT::parallel, IT::parallel, IT::parallel, IT::parallel,
+                           IT::reduction, IT::reduction};
+        } else if (numSpatialDims == 2) {
+          AffineExpr n, g, cg, oh, ow, fg, kh, kw;
+          bindDims(context, n, g, cg, oh, ow, fg, kh, kw);
+          AffineExpr d0 = rewriter.getAffineConstantExpr(dilationInts[0]);
+          AffineExpr d1 = rewriter.getAffineConstantExpr(dilationInts[1]);
+          SmallVector<AffineExpr> goExprs = {
+              n, g, fg, d0 * kh + oh, d1 * kw + ow};
+          SmallVector<AffineExpr> weiExprs = {g, fg, cg, kh, kw};
+          SmallVector<AffineExpr> outExprs = {n, g, cg, oh, ow};
+          indexingMaps = {
+              AffineMap::get(8, 0, goExprs, context),
+              AffineMap::get(8, 0, weiExprs, context),
+              AffineMap::get(8, 0, outExprs, context)};
+          iteratorTypes = {IT::parallel, IT::parallel, IT::parallel, IT::parallel, IT::parallel,
+                           IT::reduction, IT::reduction, IT::reduction};
+        } else {
+          AffineExpr n, g, cg, od, oh, ow, fg, kd, kh, kw;
+          bindDims(context, n, g, cg, od, oh, ow, fg, kd, kh, kw);
+          AffineExpr d0 = rewriter.getAffineConstantExpr(dilationInts[0]);
+          AffineExpr d1 = rewriter.getAffineConstantExpr(dilationInts[1]);
+          AffineExpr d2 = rewriter.getAffineConstantExpr(dilationInts[2]);
+          SmallVector<AffineExpr> goExprs = {n,
+                                             g,
+                                             fg,
+                                             d0 * kd + od,
+                                             d1 * kh + oh,
+                                             d2 * kw + ow};
+          SmallVector<AffineExpr> weiExprs = {g, fg, cg, kd, kh, kw};
+          SmallVector<AffineExpr> outExprs = {n, g, cg, od, oh, ow};
+          indexingMaps = {
+              AffineMap::get(10, 0, goExprs, context),
+              AffineMap::get(10, 0, weiExprs, context),
+              AffineMap::get(10, 0, outExprs, context)};
+          iteratorTypes = {IT::parallel, IT::parallel, IT::parallel, IT::parallel, IT::parallel, IT::parallel,
+                           IT::reduction, IT::reduction, IT::reduction, IT::reduction};
+        }
       }
 
       auto genericRes = linalg::GenericOp::create(