colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Integrate LLVM at llvm/llvm-project@26750a1264 

Updates LLVM usage to match
[26750a1264b3](https://github.com/llvm/llvm-project/commit/26750a1264b3)

PiperOrigin-RevId: 339127601
This commit is contained in:
Benjamin Kramer 2020-10-26 15:10:29 -07:00 committed by TensorFlow MLIR Team
parent 6eda9ed273
commit b7aa01dbe0
3 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
build_tools/llvm_version.txt
View File

@ -1,2 +1,2 @@
c89447b65984c97145f63be21e42cfa98da60dd2 26750a1264b3df114a1efae7cde6f0784206b2ce

7
lib/Dialect/mhlo/transforms/hlo_legalize_to_lhlo.cc
View File

@ -541,10 +541,6 @@ struct HloLegalizeToLhlo
return std::all_of(op.operand_type_begin(), op.operand_type_end(), return std::all_of(op.operand_type_begin(), op.operand_type_end(),
isMemRefType); isMemRefType);
}); });
target.addDynamicallyLegalOp<shape::AssumingOp>([&](shape::AssumingOp op) {
return std::all_of(op.result_type_begin(), op.result_type_end(),
isMemRefType);
});
auto kind = results_escape_function auto kind = results_escape_function
? BufferizeTypeConverter::KeepAsFunctionResult ? BufferizeTypeConverter::KeepAsFunctionResult
@ -557,7 +553,8 @@ struct HloLegalizeToLhlo
populateWithBufferizeOpConversionPatterns<mlir::ReturnOp, mlir::ReturnOp, populateWithBufferizeOpConversionPatterns<mlir::ReturnOp, mlir::ReturnOp,
lmhlo::CopyOp>( lmhlo::CopyOp>(
&context, converter, patterns); &context, converter, patterns);
populateShapeTypeConversionPatterns(&context, converter, patterns); populateShapeStructuralTypeConversionsAndLegality(&context, converter,
patterns, target);
if (failed(applyPartialConversion(getOperation(), target, patterns))) if (failed(applyPartialConversion(getOperation(), target, patterns)))
signalPassFailure(); signalPassFailure();
} }

2
lib/Dialect/mhlo/transforms/lhlo_fuse_linalg.cc
View File

@ -121,7 +121,7 @@ class LhloFuseLinalgPass
for (unsigned id = 0, e = LinalgOp(op).getNumInputs(); id < e; ++id) { for (unsigned id = 0, e = LinalgOp(op).getNumInputs(); id < e; ++id) {
linalg::Aliases aliases; linalg::Aliases aliases;
linalg::LinalgDependenceGraph graph(aliases, linalg_ops); linalg::LinalgDependenceGraph graph(aliases, linalg_ops);
if (auto info = fuseProducerOf(b, op, id, graph, &folder)) { if (auto info = fuseProducerOfBuffer(b, op, id, graph, &folder)) {
auto originalOp = info->originalProducer.getOperation(); auto originalOp = info->originalProducer.getOperation();
erase_set.insert(originalOp); erase_set.insert(originalOp);
auto originalOpInLinalgOpsVector = std::find_if( auto originalOpInLinalgOpsVector = std::find_if(