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

update llvm-project to d13da154a7c7eff77df8686b2de1cfdfa7cc7029 (#2483)

pull/2487/head
Daniel Garvey 2023-09-26 16:15:55 -05:00 committed by GitHub
parent c9fd78988e
commit ff7f8b21dc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
externals/llvm-project vendored

@ -1 +1 @@
Subproject commit f66cd9e9556a53142a26a5c21a72e21f1579217c
Subproject commit d13da154a7c7eff77df8686b2de1cfdfa7cc7029

10
lib/Conversion/TorchToSCF/TorchToSCF.cpp
View File

@ -237,17 +237,17 @@ public:
SmallVector<Type> regionArgTypes;
SmallVector<Location> regionArgLocs;
for (Value value : scfForOp.getLoopBody().front().getArguments()) {
for (Value value : scfForOp.getRegion().front().getArguments()) {
regionArgTypes.push_back(value.getType());
regionArgLocs.push_back(value.getLoc());
}
// Populate the loop body region.
if (!scfForOp.getLoopBody().empty())
rewriter.eraseBlock(&scfForOp.getLoopBody().back());
if (!scfForOp.getRegion().empty())
rewriter.eraseBlock(&scfForOp.getRegion().back());
auto *block = rewriter.createBlock(&scfForOp.getLoopBody(),
scfForOp.getLoopBody().begin(),
auto *block = rewriter.createBlock(&scfForOp.getRegion(),
scfForOp.getRegion().begin(),
regionArgTypes, regionArgLocs);
// Rewrite uses of the torch loop block arguments to the new for-loop