diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index 2535941d668..5d83478e39a 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 = "943660fd15f193dc6961597c25541fee2e01ebbb" - LLVM_SHA256 = "72a3f845eb1839b32bccaffa317517cca910511896b68f5c18959a579d57d4f2" + LLVM_COMMIT = "e808cab824488af137b62902e65dec3827b83b46" + LLVM_SHA256 = "99c30723f2b066bc9145bc43f762cdbde10fada818004dd0cc56abc5761606ac" 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 5627067ac5e..18a7c4df7d7 100644 --- a/third_party/mlir/BUILD +++ b/third_party/mlir/BUILD @@ -545,6 +545,7 @@ cc_library( deps = [ ":Affine", ":AffinePassIncGen", + ":AffineUtils", ":Analysis", ":IR", ":Pass", @@ -2768,6 +2769,7 @@ cc_library( "@llvm-project//mlir/test:TestDialect", "@llvm-project//mlir/test:TestIR", "@llvm-project//mlir/test:TestPass", + "@llvm-project//mlir/test:TestReducer", "@llvm-project//mlir/test:TestSPIRV", "@llvm-project//mlir/test:TestTransforms", ], @@ -2916,6 +2918,7 @@ cc_binary( "@llvm-project//mlir/test:TestDialect", "@llvm-project//mlir/test:TestIR", "@llvm-project//mlir/test:TestPass", + "@llvm-project//mlir/test:TestReducer", "@llvm-project//mlir/test:TestSPIRV", "@llvm-project//mlir/test:TestTransforms", ], diff --git a/third_party/mlir/test.BUILD b/third_party/mlir/test.BUILD index 4b999bfa466..e0966054542 100644 --- a/third_party/mlir/test.BUILD +++ b/third_party/mlir/test.BUILD @@ -167,6 +167,18 @@ cc_library( ], ) +cc_library( + name = "TestReducer", + srcs = [ + "lib/Reducer/MLIRTestReducer.cpp", + ], + deps = [ + "@llvm-project//mlir:IR", + "@llvm-project//mlir:Pass", + "@llvm-project//mlir:Support", + ], +) + cc_library( name = "TestTransforms", srcs = glob(["lib/Transforms/*.cpp"]),