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

fix

pull/3139/head
Xinan Jiang(姜曦楠) 2024-04-12 10:47:40 +08:00
parent f6dee64cad
commit 244bd24a07
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -819,8 +819,9 @@ public:
}); });
// Get the selected item directly from the concat operand. // Get the selected item directly from the concat operand.
Value replace; Value replace = op.getResult();
for (int64_t sum = 0, i = 0; i < elements.size(); ++i) { int64_t sum = 0;
for (unsigned i = 0; i < elements.size(); ++i) {
// Unsupported unknown dim size // Unsupported unknown dim size
if (elementWidths[i] == kUnknownSize) if (elementWidths[i] == kUnknownSize)
return failure(); return failure();