diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index d869ac53582..d3a39390b23 100755 --- a/tensorflow/workspace.bzl +++ b/tensorflow/workspace.bzl @@ -685,8 +685,8 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""): ) # Check out LLVM and MLIR from llvm-project. - LLVM_COMMIT = "e996f1d4194bccf1c8ca984d695b848c0093bc23" - LLVM_SHA256 = "4bcb567a485791f4b49d39dc6901c7d179d592f5a427a22f05ea6b4c710103de" + LLVM_COMMIT = "71635ea5ffd62a7de91c759c0dfb7bb40c16fd94" + LLVM_SHA256 = "fcf303a228071a705f8185d66ecab4e0c4e82bdcaa4626919e302b3276b09bdd" 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), diff --git a/third_party/mlir/BUILD b/third_party/mlir/BUILD index 257d2e8e4a7..706528b695c 100644 --- a/third_party/mlir/BUILD +++ b/third_party/mlir/BUILD @@ -87,8 +87,10 @@ gentbl( "include/mlir/IR/BuiltinOps.td", "include/mlir/IR/BuiltinDialect.td", "include/mlir/Interfaces/CallInterfaces.td", + "include/mlir/Interfaces/CastInterfaces.td", "include/mlir/IR/SymbolInterfaces.td", ":OpBaseTdFiles", + ":SideEffectTdFiles", ], ) @@ -124,6 +126,8 @@ cc_library( "include/mlir/IR/*.h", ]) + [ "include/mlir/Interfaces/CallInterfaces.h", + "include/mlir/Interfaces/CastInterfaces.h", + "include/mlir/Interfaces/SideEffectInterfaces.h", "include/mlir/Interfaces/DecodeAttributesInterfaces.h", "include/mlir/Interfaces/FoldInterfaces.h", ], @@ -133,6 +137,7 @@ cc_library( ":BuiltinOpsIncGen", ":BuiltinTypesIncGen", ":CallOpInterfacesIncGen", + ":CastOpInterfacesIncGen", ":InferTypeOpInterfaceIncGen", ":OpAsmInterfaceIncGen", ":RegionKindInterfaceIncGen", @@ -1031,6 +1036,7 @@ filegroup( "include/mlir/IR/OpAsmInterface.td", "include/mlir/IR/SymbolInterfaces.td", "include/mlir/Interfaces/CallInterfaces.td", + "include/mlir/Interfaces/CastInterfaces.td", "include/mlir/Interfaces/ControlFlowInterfaces.td", "include/mlir/Interfaces/ViewLikeInterface.td", ":OpBaseTdFiles", @@ -1599,6 +1605,7 @@ cc_library( includes = ["include"], deps = [ ":CallOpInterfaces", + ":CastOpInterfaces", ":CommonFolders", ":ControlFlowInterfaces", ":EDSC", @@ -3041,6 +3048,7 @@ filegroup( srcs = [ "include/mlir/Dialect/Tensor/IR/TensorBase.td", "include/mlir/Dialect/Tensor/IR/TensorOps.td", + "include/mlir/Interfaces/CastInterfaces.td", "include/mlir/Interfaces/ControlFlowInterfaces.td", ":OpBaseTdFiles", ":SideEffectTdFiles", @@ -3098,6 +3106,7 @@ cc_library( ], includes = ["include"], deps = [ + ":CastOpInterfaces", ":ControlFlowInterfaces", ":IR", ":SideEffectInterfaces", @@ -3573,6 +3582,43 @@ cc_library( ], ) +gentbl( + name = "CastOpInterfacesIncGen", + strip_include_prefix = "include", + tbl_outs = [ + ( + "-gen-op-interface-decls", + "include/mlir/Interfaces/CastInterfaces.h.inc", + ), + ( + "-gen-op-interface-defs", + "include/mlir/Interfaces/CastInterfaces.cpp.inc", + ), + ], + tblgen = ":mlir-tblgen", + td_file = "include/mlir/Interfaces/CastInterfaces.td", + td_srcs = [ + ":OpBaseTdFiles", + ], +) + +cc_library( + name = "CastOpInterfaces", + srcs = [ + "lib/Interfaces/CastInterfaces.cpp", + ], + hdrs = [ + "include/mlir/Interfaces/CastInterfaces.h", + ], + includes = ["include"], + deps = [ + ":CastOpInterfacesIncGen", + ":IR", + ":Support", + "@llvm-project//llvm:Support", + ], +) + gentbl( name = "ControlFlowInterfacesIncGen", strip_include_prefix = "include", @@ -5289,6 +5335,8 @@ exports_files( "include/mlir/Bindings/Python/Attributes.td", "include/mlir/Interfaces/CallInterfaces.h", "include/mlir/Interfaces/CallInterfaces.td", + "include/mlir/Interfaces/CastInterfaces.h", + "include/mlir/Interfaces/CastInterfaces.td", "include/mlir/Interfaces/ControlFlowInterfaces.h", "include/mlir/Interfaces/ControlFlowInterfaces.td", "include/mlir/Interfaces/CopyOpInterface.td",