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

Bump OSS LLVM to 6bfc45cf60d2a82e659d3911c26169ad02cf78c0.

Browse Source 
Add StripDebugInfoPass to MLIR GPU kernel lowering.

PiperOrigin-RevId: 293572178
Change-Id: Id86a25b613ebe441b1386102985c7ad8bab19a8b
This commit is contained in:
Thomas Joerg 2020-02-06 05:28:57 -08:00 committed by TensorFlower Gardener
parent a1c59c73da
commit bc28d49ce2
3 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc
View File

@ -359,6 +359,7 @@ Status LowerKernelBodiesToNVVM(mlir::ModuleOp module) {
// Some basic cleanup.
kernelPm.addNestedPass<::mlir::FuncOp>(::mlir::createCanonicalizerPass());
kernelPm.addNestedPass<::mlir::FuncOp>(::mlir::createCSEPass());
kernelPm.addPass(::mlir::createStripDebugInfoPass());
if (failed(pm.run(module))) {
return InternalError("Lowering to NVVM IR failed.");

4
tensorflow/workspace.bzl
View File

@ -595,8 +595,8 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
)
# Check out LLVM and MLIR from llvm-project.
LLVM_COMMIT = "7b6e49a2f022288f010bc600323f1d0aafca6241"
LLVM_SHA256 = "2a87423459f1f28421b81c713baf30dd3361c6cd1f0ca35d6479fb5609e2cf97"
LLVM_COMMIT = "6bfc45cf60d2a82e659d3911c26169ad02cf78c0"
LLVM_SHA256 = "c19ca04ac647f903ed56e7c23c5dc031e4ecba2c9a9860dc6592a2f55f2ae2ce"
LLVM_URLS = [
"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),

2
third_party/mlir/BUILD vendored
View File

@ -1554,6 +1554,8 @@ cc_library(
cc_library(
name = "LLVMIRModuleTranslation",
srcs = [
"lib/Target/LLVMIR/DebugTranslation.cpp",
"lib/Target/LLVMIR/DebugTranslation.h",
"lib/Target/LLVMIR/ModuleTranslation.cpp",
],
hdrs = [