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

Integrate LLVM at llvm/llvm-project@16428a8d91 

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

PiperOrigin-RevId: 357550807
This commit is contained in:
A. Unique TensorFlower 2021-02-15 04:17:20 -08:00 committed by TensorFlow MLIR Team
parent 89d81adf6d
commit 2fe0c33083
3 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
WORKSPACE
View File

@ -15,9 +15,9 @@
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive") load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
LLVM_COMMIT = "2bfe27da171e8a6dddac6c444c4bca003103941a" LLVM_COMMIT = "16428a8d91a904c841086a971a8f26c3d8cf1307"
LLVM_SHA256 = "e94b8ec0d1b854da11dbf1587701e798df97be630f7019d0658452fae5e03bcf" LLVM_SHA256 = "8c2f3945608365258148e7e1979810d99c676e0e750f174d92bdf2e0da549954"
LLVM_BAZEL_TAG = "llvm-project-{commit}".format(commit = LLVM_COMMIT) LLVM_BAZEL_TAG = "llvm-project-{commit}".format(commit = LLVM_COMMIT)

2
build_tools/llvm_version.txt
View File

@ -1,2 +1,2 @@
2bfe27da171e8a6dddac6c444c4bca003103941a 16428a8d91a904c841086a971a8f26c3d8cf1307

10
lib/Dialect/mhlo/transforms/legalize_to_linalg.cc
View File

@ -805,9 +805,8 @@ class ReshapeOpConverter : public OpConversionPattern<OpTy> {
loc, collapsed_type, args[0], collapsing_map); loc, collapsed_type, args[0], collapsing_map);
Value reshape_buffer = rewriter.create<linalg::ReshapeOp>( Value reshape_buffer = rewriter.create<linalg::ReshapeOp>(
loc, result_type, collapsed_op, expanding_map); loc, result_type, collapsed_op, expanding_map);
rewriter.replaceOpWithNewOp<linalg::CopyOp>( rewriter.replaceOpWithNewOp<linalg::CopyOp>(reshape_op, reshape_buffer,
reshape_op, reshape_buffer, args[1], /*inputPermutation =*/nullptr, args[1]);
/*outputPermutation =*/nullptr);
} else { } else {
auto collapsed_type = RankedTensorType::get({total_elems}, elem_type); auto collapsed_type = RankedTensorType::get({total_elems}, elem_type);
Value collapsed_op = rewriter.create<linalg::TensorReshapeOp>( Value collapsed_op = rewriter.create<linalg::TensorReshapeOp>(
@ -821,9 +820,8 @@ class ReshapeOpConverter : public OpConversionPattern<OpTy> {
if (isLHLO) { if (isLHLO) {
Value reshape_buffer = rewriter.create<linalg::ReshapeOp>( Value reshape_buffer = rewriter.create<linalg::ReshapeOp>(
reshape_op.getLoc(), result_type, args[0], reassociation_map); reshape_op.getLoc(), result_type, args[0], reassociation_map);
rewriter.replaceOpWithNewOp<linalg::CopyOp>( rewriter.replaceOpWithNewOp<linalg::CopyOp>(reshape_op, reshape_buffer,
reshape_op, reshape_buffer, args[1], /*inputPermutation =*/nullptr, args[1]);
/*outputPermutation =*/nullptr);
} else { } else {
rewriter.replaceOpWithNewOp<linalg::TensorReshapeOp>( rewriter.replaceOpWithNewOp<linalg::TensorReshapeOp>(
reshape_op, result_type, args[0], reassociation_map); reshape_op, result_type, args[0], reassociation_map);