experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Exposes MLIR HLO passes outside of the mlir/hlo folder.

Browse Source 
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
This commit is contained in:
Mehdi Amini 2020-07-30 10:54:32 -07:00 committed by TensorFlower Gardener
parent cb3a22b732
commit cac585bccb
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/mlir/hlo/BUILD
View File

@ -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",