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

Integrate LLVM at llvm/llvm-project@c30ab6c2a3

Browse Source 
Updates LLVM usage to match
[c30ab6c2a307](https://github.com/llvm/llvm-project/commit/c30ab6c2a307)

PiperOrigin-RevId: 342309362
Change-Id: I7e966f259ba3ba79824f9575dab44ee1a8e4968b
This commit is contained in:
Tres Popp 2020-11-13 11:58:12 -08:00 committed by TensorFlower Gardener
parent a19c7cced0
commit e4a6158a07
7 changed files with 29 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/mlir-module-serialized-str-attr.mlir
View File

@ -7,4 +7,4 @@ module attributes {tf.versions = {producer = 888 : i32}} {
} loc(unknown)
} loc(unknown)
// CHECK: "module attributes {tf.versions = {producer = 888 : i32}} {\0A func @main(%arg0: tensor<?xi32>) -> tensor<?xi32> {\0A %0 = \22tf.Identity\22(%arg0) : (tensor<?xi32>) -> tensor<?xi32> loc(unknown)\0A return %0 : tensor<?xi32> loc(unknown)\0A } loc(unknown)\0A} loc(unknown)"
// CHECK: "module attributes {tf.versions = {producer = 888 : i32}} {\0A func @main(%arg0: tensor<?xi32>) -> tensor<?xi32> {\0A %0 = \22tf.Identity\22(%arg0) : (tensor<?xi32>) -> tensor<?xi32> loc(unknown)\0A return %0 : tensor<?xi32> loc(unknown)\0A } loc(unknown)\0A} loc(unknown)\0A"

4
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/debug_info.py
View File

@ -34,8 +34,8 @@ class TestModule(tf.Module):
return x + y
# Basic check that the debug info file is being correctly saved and loaded.
#
# CHECK: #[[LOC:.*]] = loc({{.*}}callsite("{{[^"]*}}/debug_info.py":{{[0-9]+}}:{{[0-9]+}}
# CHECK: "tf.AddV2"{{.*}}#[[LOC]]
# CHECK: "tf.AddV2"{{.*}}loc(#[[LOC:.*]])
# CHECK: #[[LOC]] = loc({{.*}}callsite("{{[^"]*}}/debug_info.py":{{[0-9]+}}:{{[0-9]+}}
if __name__ == '__main__':

2
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
View File

@ -366,7 +366,7 @@ def TFR_TFRFuncOp : TFR_Op<"func", [HasParent<"ModuleOp">,
Syntax:
```
op ::= `tfr.func` symbol-ref-id `(` argument-list `)` (`->`
op ::= `tfr.func` visibility? symbol-ref-id `(` argument-list `)` (`->`
function-result-list)? function-attributes? region
```

4
tensorflow/compiler/xla/service/cpu/simple_orc_jit.cc
View File

@ -163,7 +163,9 @@ llvm::Expected<std::unique_ptr<SimpleOrcJIT>> SimpleOrcJIT::Create(
LLVMCompiler::ModuleHook pre_optimization_hook,
LLVMCompiler::ModuleHook post_optimization_hook,
std::function<void(const llvm::object::ObjectFile&)> post_codegen_hook) {
auto target_process_control = llvm::orc::SelfTargetProcessControl::Create();
auto SSP = std::make_shared<llvm::orc::SymbolStringPool>();
auto target_process_control =
llvm::orc::SelfTargetProcessControl::Create(std::move(SSP));
if (!target_process_control) {
return target_process_control.takeError();
}

2
tensorflow/python/compiler/mlir/mlir.py
View File

@ -68,7 +68,7 @@ def convert_function(concrete_function, pass_pipeline='tf-standard-pipeline'):
... tf.TensorSpec(None, tf.dtypes.float32),
... tf.TensorSpec(None, tf.dtypes.float32))
>>> tf.mlir.experimental.convert_function(concrete_function)
'...module attributes {...} {...}'
'...module attributes {...} {...}...'
Args:
concrete_function: An object of type ConcreteFunction.

4
tensorflow/workspace.bzl
View File

@ -686,8 +686,8 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
)
# Check out LLVM and MLIR from llvm-project.
LLVM_COMMIT = "76b6cb515b2f385f8cdd3c41ecd31c78698f20b2"
LLVM_SHA256 = "711ce462f3c48bbc17395c160790f9221469c90f0b5a8419f5b758221fc5c1ef"
LLVM_COMMIT = "c30ab6c2a307cfdce8323ed94c3d70eb2d26bc14"
LLVM_SHA256 = "5f2ff1217222d74b347f5c9c4e11545c42818b4d1d14282532b7171334eaf190"
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),

19
third_party/mlir/BUILD vendored
View File

@ -316,6 +316,22 @@ gentbl(
],
)
gentbl(
name = "AsyncPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls -name Async",
"include/mlir/Dialect/Async/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Async/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
##---------------------------------------------------------------------------##
# AVX512 dialect.
##---------------------------------------------------------------------------##
@ -613,9 +629,11 @@ cc_library(
"lib/Dialect/Async/Transforms/*.cpp",
"lib/Dialect/Async/Transforms/*.h",
]),
hdrs = ["include/mlir/Dialect/Async/Passes.h"],
includes = ["include"],
deps = [
":Async",
":AsyncPassIncGen",
":IR",
":Pass",
":SCFDialect",
@ -3246,6 +3264,7 @@ cc_library(
":AffineToStandard",
":AffineTransforms",
":Async",
":AsyncPassIncGen",
":AsyncToLLVM",
":AsyncTransforms",
":ConversionPasses",