PiperOrigin-RevId: 319656927
Change-Id: I0abcec5162696bbd890f3151ee016c6ff5668317
This commit is contained in:
Alexander Belyaev 2020-07-05 03:08:07 -07:00 committed by TensorFlower Gardener
parent af7fd02ca4
commit f25f8ff7cd
3 changed files with 39 additions and 8 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 = "01c4574a129e987cd89daf94969bda2e929fe5a8"
LLVM_SHA256 = "33637bbb8329f722852b5031cdf560ebfdd94253fb4448645f0892a58cacce67"
LLVM_COMMIT = "91c320e9d852120f011028e15ff0e25cdf762aad"
LLVM_SHA256 = "06793de12600306d63b8731ea7b0a84b7c9adc3bac823ea082e151de48199a7f"
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

@ -665,6 +665,17 @@ cc_library(
],
)
cc_library(
name = "CopyOpInterface",
srcs = ["lib/Interfaces/CopyOpInterface.cpp"],
hdrs = ["include/mlir/Interfaces/CopyOpInterface.h"],
includes = ["include"],
deps = [
":CopyOpInterfaceIncGen",
":IR",
],
)
gentbl(
name = "ShapeOpsIncGen",
strip_include_prefix = "include",
@ -2130,6 +2141,26 @@ gentbl(
],
)
gentbl(
name = "CopyOpInterfaceIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/CopyOpInterface.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/CopyOpInterface.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/CopyOpInterface.td",
td_srcs = [
":OpBaseTdFiles",
],
)
gentbl(
name = "TransformsPassIncGen",
strip_include_prefix = "include",
@ -2152,14 +2183,13 @@ cc_library(
"lib/Transforms/*.cpp",
"lib/Transforms/*.h",
]),
hdrs = glob([
"include/mlir/Transforms/*.h",
]),
hdrs = glob(["include/mlir/Transforms/*.h"]),
includes = ["include"],
deps = [
":Affine",
":Analysis",
":ControlFlowInterfaces",
":CopyOpInterface",
":IR",
":LinalgOps",
":LoopLikeInterface",
@ -3223,6 +3253,7 @@ filegroup(
srcs = [
"include/mlir/Dialect/Linalg/IR/LinalgBase.td",
"include/mlir/Dialect/Linalg/IR/LinalgOps.td",
"include/mlir/Interfaces/CopyOpInterface.td",
"include/mlir/Interfaces/ViewLikeInterface.td",
":AffineOpsTdFiles",
":OpBaseTdFiles",
@ -3275,6 +3306,7 @@ filegroup(
"include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.td",
"include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td",
"include/mlir/Dialect/Linalg/IR/LinalgStructuredOpsInterface.td",
"include/mlir/Interfaces/CopyOpInterface.td",
"include/mlir/Interfaces/ViewLikeInterface.td",
":AffineOpsTdFiles",
":LinalgOpsTdFiles",
@ -3431,6 +3463,7 @@ cc_library(
includes = ["include"],
deps = [
":Affine",
":CopyOpInterface",
":DialectUtils",
":EDSC",
":IR",

View File

@ -169,9 +169,7 @@ cc_library(
cc_library(
name = "TestTransforms",
srcs = glob([
"lib/Transforms/*.cpp",
]),
srcs = glob(["lib/Transforms/*.cpp"]),
defines = ["MLIR_CUDA_CONVERSIONS_ENABLED"],
includes = ["lib/Dialect/Test"],
deps = [