llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix multiline comment.

pull/3883/head
Sayan Saha 2024-11-19 12:22:22 -05:00
parent d09d568553
commit cec8af37be
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp
View File

@ -5186,8 +5186,7 @@ public:
return rewriter.notifyMatchFailure(op, "expected weight to have a rank"); return rewriter.notifyMatchFailure(op, "expected weight to have a rank");
} }
unsigned rank = *maybeRank; unsigned rank = *maybeRank;
// first 2 dimensions of weight corresponds to out_channels and in_channels / \ // first 2 dimensions of weight are out_channels and in_channels / groups
groups
if (rank < 3) if (rank < 3)
return rewriter.notifyMatchFailure( return rewriter.notifyMatchFailure(
op, "ConvPaddingOp weight must be at least 3 dimensional."); op, "ConvPaddingOp weight must be at least 3 dimensional.");