diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index 42e960144e0..783a57bea77 100755 --- a/tensorflow/workspace.bzl +++ b/tensorflow/workspace.bzl @@ -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), diff --git a/third_party/mlir/BUILD b/third_party/mlir/BUILD index c0d6573b22e..df0a2cc74b8 100644 --- a/third_party/mlir/BUILD +++ b/third_party/mlir/BUILD @@ -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",