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

llvm: update tag to 02b3a35 (#1124) 

As part of this update, this patch also updates MHLO tag to ad54b43.
pull/1125/head
Ashay Rane 2022-07-31 15:26:24 -07:00 committed by GitHub
parent 2c3b3606d0
commit 55bbbec144
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
externals/llvm-external-projects/torch-mlir-dialects/lib/Dialect/TMTensor/Transforms/ConvertToLoops.cpp vendored
View File

@ -8,6 +8,7 @@
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/Arithmetic/IR/Arithmetic.h" #include "mlir/Dialect/Arithmetic/IR/Arithmetic.h"
#include "mlir/Dialect/Arithmetic/Utils/Utils.h"
#include "mlir/Dialect/Func/IR/FuncOps.h" #include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h" #include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Math/IR/Math.h" #include "mlir/Dialect/Math/IR/Math.h"
@ -40,9 +41,14 @@ static LogicalResult lowerToLoopsImpl(OpBuilder &builder,
return scalarLoopOp.generateScalarImplementation(builder, loc, ivs); return scalarLoopOp.generateScalarImplementation(builder, loc, ivs);
} }
LogicalResult status = success(); LogicalResult status = success();
Value offset = getValueOrCreateConstantIndexOp(builder, loc,
loopRanges[loopDepth].offset);
Value size =
getValueOrCreateConstantIndexOp(builder, loc, loopRanges[loopDepth].size);
Value stride = getValueOrCreateConstantIndexOp(builder, loc,
loopRanges[loopDepth].stride);
builder.create<scf::ForOp>( builder.create<scf::ForOp>(
loc, loopRanges[loopDepth].offset, loopRanges[loopDepth].size, loc, offset, size, stride, ValueRange{},
loopRanges[loopDepth].stride, ValueRange{},
[&](OpBuilder &b, Location loc, Value iv, ValueRange args) { [&](OpBuilder &b, Location loc, Value iv, ValueRange args) {
ivs.push_back(iv); ivs.push_back(iv);
status = status =

2
externals/llvm-project vendored

@ -1 +1 @@
Subproject commit 3580daacf3ecb9bc628d05a4037266df6532ad6e Subproject commit 02b3a358926e7bbcac9226cbecbfc3067c2ad61b

2
externals/mlir-hlo vendored

@ -1 +1 @@
Subproject commit eb1042390d39131fe7e330b54dc5f29a79c9a072 Subproject commit ad54b43c623cc5ae69b0e90f395b3fba13ffa55a