From cac585bccbafb212d545dbe29184ecfedb5913bf Mon Sep 17 00:00:00 2001 From: Mehdi Amini Date: Thu, 30 Jul 2020 10:54:32 -0700 Subject: [PATCH] Exposes MLIR HLO passes outside of the mlir/hlo folder. This allows external mlir-xyz-opt tool to register them more conveniently. It was private when it was relying on static registration and "always_link" which has a tendency to bloat binaries even when the passes aren't used. PiperOrigin-RevId: 324037010 Change-Id: I8b8aa4ff0efa46c232cfdf2bbe63bbd1833dd524 --- tensorflow/compiler/mlir/hlo/BUILD | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/compiler/mlir/hlo/BUILD b/tensorflow/compiler/mlir/hlo/BUILD index c6909079113..273bc4ebaeb 100644 --- a/tensorflow/compiler/mlir/hlo/BUILD +++ b/tensorflow/compiler/mlir/hlo/BUILD @@ -774,12 +774,12 @@ cc_library( ) cc_library( - name = "all_passes_for_testing", + name = "all_passes", hdrs = [ "include/mlir-hlo/Dialect/mhlo/transforms/register_passes.h", ], visibility = [ - "//tensorflow/compiler/mlir:__subpackages__", + ":friends", ], deps = [ ":LmhloPassIncGen", @@ -812,7 +812,7 @@ cc_binary( "tools/mlir-hlo-opt/mlir-hlo-opt.cpp", ], deps = [ - ":all_passes_for_testing", + ":all_passes", ":hlo_dialect_registration", "@llvm-project//llvm:Support", "@llvm-project//mlir:AllPassesAndDialectsNoRegistration",