PiperOrigin-RevId: 319914621
Change-Id: I5436113a6d2dea9094651b78fb2ebeb8624372fd
This commit is contained in:
A. Unique TensorFlower 2020-07-06 21:27:26 -07:00 committed by TensorFlower Gardener
parent 6300825449
commit f6ab4daebc
2 changed files with 38 additions and 2 deletions

View File

@ -710,8 +710,8 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
)
# Check out LLVM and MLIR from llvm-project.
LLVM_COMMIT = "0607c8df7faf71bd726e9d18bafd2f7566984e35"
LLVM_SHA256 = "bbc2f49b016d8a5938a1396b8177498c318753fdd768fd4c169349c9a7a3f221"
LLVM_COMMIT = "f47b8851318d5ec2fa1e7867f3fdb86101cdc1da"
LLVM_SHA256 = "6174045de3ffacda49e7edbd98c84cff0a528ce832dde4183a2c0ff54469dd6a"
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),

View File

@ -397,6 +397,7 @@ filegroup(
"include/mlir/Interfaces/CallInterfaces.td",
"include/mlir/Interfaces/ControlFlowInterfaces.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
"include/mlir/Interfaces/VectorUnrollInterface.td",
"include/mlir/Interfaces/ViewLikeInterface.td",
":OpBaseTdFiles",
],
@ -654,6 +655,17 @@ cc_library(
],
)
cc_library(
name = "VectorUnrollInterface",
srcs = ["lib/Interfaces/VectorUnrollInterface.cpp"],
hdrs = ["include/mlir/Interfaces/VectorUnrollInterface.h"],
includes = ["include"],
deps = [
":IR",
":VectorUnrollInterfaceIncGen",
],
)
cc_library(
name = "ViewLikeInterface",
srcs = ["lib/Interfaces/ViewLikeInterface.cpp"],
@ -852,6 +864,7 @@ cc_library(
":SideEffectInterfaces",
":StandardOpsIncGen",
":Support",
":VectorUnrollInterface",
":ViewLikeInterface",
"@llvm-project//llvm:Support",
],
@ -916,6 +929,7 @@ cc_library(
":Support",
":VectorOpsIncGen",
":VectorTransformPatternsIncGen",
":VectorUnrollInterface",
"@llvm-project//llvm:Support",
],
)
@ -2121,6 +2135,26 @@ gentbl(
],
)
gentbl(
name = "VectorUnrollInterfaceIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/VectorUnrollInterface.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/VectorUnrollInterface.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/VectorUnrollInterface.td",
td_srcs = [
":OpBaseTdFiles",
],
)
gentbl(
name = "ViewLikeInterfaceIncGen",
strip_include_prefix = "include",
@ -3546,6 +3580,7 @@ filegroup(
name = "VectorOpsTdFiles",
srcs = [
"include/mlir/Dialect/Vector/VectorOps.td",
"include/mlir/Interfaces/VectorUnrollInterface.td",
":AffineOpsTdFiles",
":OpBaseTdFiles",
],
@ -3675,6 +3710,7 @@ exports_files(
"include/mlir/Interfaces/ControlFlowInterfaces.h",
"include/mlir/Interfaces/ControlFlowInterfaces.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
"include/mlir/Interfaces/VectorUnrollInterface.td",
"include/mlir/Interfaces/ViewLikeInterface.td",
"include/mlir/Dialect/LLVMIR/LLVMOpBase.td",
"include/mlir/Dialect/StandardOps/IR/Ops.td",