experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bump OSS LLVM to 448b777b864a312e3b2ceae1e0cd59752a1fe90e

Browse Source 
PiperOrigin-RevId: 305329788
Change-Id: I30e62190811430d9e1103e879096ac6444b0caed
This commit is contained in:
Benjamin Kramer 2020-04-07 13:34:07 -07:00 committed by TensorFlower Gardener
parent c57bb8bdc6
commit 9327fd74f6
5 changed files with 7 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/mlir/BUILD
View File

@ -170,7 +170,6 @@ tf_cc_binary(
"@llvm-project//mlir:AllPassesAndDialects", "@llvm-project//mlir:AllPassesAndDialects",
"@llvm-project//mlir:IR", "@llvm-project//mlir:IR",
"@llvm-project//mlir:Support", "@llvm-project//mlir:Support",
"@llvm-project//mlir:TranslateClParser",
"@llvm-project//mlir:Translation", "@llvm-project//mlir:Translation",
], ],
) )

2
tensorflow/compiler/mlir/tf_mlir_translate_main.cc
View File

@ -25,7 +25,7 @@ limitations under the License.
#include "mlir/Support/FileUtilities.h" // from @llvm-project #include "mlir/Support/FileUtilities.h" // from @llvm-project
#include "mlir/Support/LogicalResult.h" // from @llvm-project #include "mlir/Support/LogicalResult.h" // from @llvm-project
#include "mlir/Support/ToolUtilities.h" // from @llvm-project #include "mlir/Support/ToolUtilities.h" // from @llvm-project
#include "mlir/Support/TranslateClParser.h" // from @llvm-project #include "mlir/Translation.h" // from @llvm-project
#include "tensorflow/compiler/mlir/init_mlir.h" #include "tensorflow/compiler/mlir/init_mlir.h"
#include "tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.h" #include "tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.h"
#include "tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.h" #include "tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.h"

5
tensorflow/compiler/mlir/xla/transforms/lhlo_legalize_to_parallel_loops.cc
View File

@ -287,7 +287,7 @@ class ReduceOpConverter : public OpConversionPattern<xla_lhlo::ReduceOp> {
: *outer_ivs_it++); : *outer_ivs_it++);
} }
} else { } else {
indices = ValueRange(inner.getInductionVars()); indices = inner.getInductionVars();
} }
rewriter->setInsertionPointToStart(inner.getBody()); rewriter->setInsertionPointToStart(inner.getBody());
@ -409,8 +409,7 @@ class ReduceWindowOpConverter
Value reduction_result = *window_loop.getResults().begin(); Value reduction_result = *window_loop.getResults().begin();
auto output_ivs = output_loop.getInductionVars(); auto output_ivs = output_loop.getInductionVars();
rewriter->create<StoreOp>(loc, reduction_result, xla_output, rewriter->create<StoreOp>(loc, reduction_result, xla_output, output_ivs);
ValueRange{output_ivs});
return std::make_pair(output_loop, window_loop); return std::make_pair(output_loop, window_loop);
} }

4
tensorflow/workspace.bzl
View File

@ -591,8 +591,8 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
) )
# Check out LLVM and MLIR from llvm-project. # Check out LLVM and MLIR from llvm-project.
LLVM_COMMIT = "71305033d11564fe9c19e8e40200680daae41e89" LLVM_COMMIT = "448b777b864a312e3b2ceae1e0cd59752a1fe90e"
LLVM_SHA256 = "b3deb570e34de7850793e17a15e9576fbc6bc0229150a7cf0ff26a2cdb010ad0" LLVM_SHA256 = "852a7e4a8ce5e3385368665ae3091548e58fde0c012b66950c0151fb7942a118"
LLVM_URLS = [ LLVM_URLS = [
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/llvm/llvm-project/archive/{commit}.tar.gz".format(commit = LLVM_COMMIT), "https://storage.googleapis.com/mirror.tensorflow.org/github.com/llvm/llvm-project/archive/{commit}.tar.gz".format(commit = LLVM_COMMIT),
"https://github.com/llvm/llvm-project/archive/{commit}.tar.gz".format(commit = LLVM_COMMIT), "https://github.com/llvm/llvm-project/archive/{commit}.tar.gz".format(commit = LLVM_COMMIT),

24
third_party/mlir/BUILD vendored
View File

@ -722,16 +722,12 @@ cc_library(
"lib/Support/JitRunner.cpp", "lib/Support/JitRunner.cpp",
# TODO(jpienaar): Move this out, else Support depends on Analysis/ # TODO(jpienaar): Move this out, else Support depends on Analysis/
"lib/Support/MlirOptMain.cpp", "lib/Support/MlirOptMain.cpp",
# TODO(jpienaar): Move this out, else Support depends on Analysis/
"lib/Support/TranslateClParser.cpp",
], ],
), ),
hdrs = glob([ hdrs = glob([
"include/mlir/ADT/*.h", "include/mlir/ADT/*.h",
"include/mlir/Support/*.h", "include/mlir/Support/*.h",
]) + [ ]),
"include/mlir/Translation.h",
],
includes = ["include"], includes = ["include"],
deps = [ deps = [
"@llvm-project//llvm:support", "@llvm-project//llvm:support",
@ -2184,9 +2180,9 @@ cc_library(
], ],
includes = ["include"], includes = ["include"],
deps = [ deps = [
":Analysis",
":IR", ":IR",
":Parser", ":Parser",
":StandardOps",
":Support", ":Support",
"@llvm-project//llvm:support", "@llvm-project//llvm:support",
], ],
@ -2377,21 +2373,6 @@ cc_library(
], ],
) )
cc_library(
name = "TranslateClParser",
srcs = ["lib/Support/TranslateClParser.cpp"],
hdrs = ["include/mlir/Support/TranslateClParser.h"],
includes = ["include"],
deps = [
":Analysis",
":IR",
":Parser",
":Support",
":Translation",
"@llvm-project//llvm:support",
],
)
cc_library( cc_library(
name = "MlirTranslateMain", name = "MlirTranslateMain",
srcs = ["tools/mlir-translate/mlir-translate.cpp"], srcs = ["tools/mlir-translate/mlir-translate.cpp"],
@ -2400,7 +2381,6 @@ cc_library(
":IR", ":IR",
":Parser", ":Parser",
":Support", ":Support",
":TranslateClParser",
":Translation", ":Translation",
"@llvm-project//llvm:support", "@llvm-project//llvm:support",
], ],