diff --git a/tensorflow/compiler/mlir/hlo/BUILD b/tensorflow/compiler/mlir/hlo/BUILD index 0f3860a3175..06ee0c231b9 100644 --- a/tensorflow/compiler/mlir/hlo/BUILD +++ b/tensorflow/compiler/mlir/hlo/BUILD @@ -464,7 +464,6 @@ cc_library( ":hlo", ":lhlo", ":lhlo_gpu", - "@llvm-project//mlir:IR", ], ) @@ -500,7 +499,6 @@ cc_library( "@llvm-project//mlir:SCFDialect", "@llvm-project//mlir:StandardOps", "@llvm-project//mlir:Support", - "@llvm-project//mlir:Transforms", ], ) @@ -639,12 +637,10 @@ cc_library( ":lhlo", "@llvm-project//llvm:Support", "@llvm-project//mlir:Affine", - "@llvm-project//mlir:IR", "@llvm-project//mlir:LinalgTransforms", "@llvm-project//mlir:Pass", "@llvm-project//mlir:SCFDialect", "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:TransformUtils", "@llvm-project//mlir:ViewLikeInterface", ], @@ -702,12 +698,10 @@ cc_library( deps = [ ":cycle_detector", ":hlo", - "@llvm-project//llvm:Core", "@llvm-project//llvm:Support", "@llvm-project//mlir:IR", "@llvm-project//mlir:Pass", "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:TransformUtils", ], alwayslink = 1, @@ -738,7 +732,6 @@ cc_library( deps = [ ":hlo", "@llvm-project//llvm:Support", - "@llvm-project//mlir:Analysis", "@llvm-project//mlir:IR", "@llvm-project//mlir:Pass", "@llvm-project//mlir:StandardOps", @@ -759,7 +752,6 @@ cc_library( "@llvm-project//mlir:IR", "@llvm-project//mlir:Pass", "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:TransformUtils", ], alwayslink = 1, @@ -777,8 +769,6 @@ cc_library( "@llvm-project//llvm:Support", "@llvm-project//mlir:IR", "@llvm-project//mlir:Pass", - "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", ], alwayslink = 1, @@ -797,7 +787,6 @@ cc_library( "@llvm-project//mlir:IR", "@llvm-project//mlir:Pass", "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", ], alwayslink = 1, @@ -838,11 +827,9 @@ cc_library( ":hlo", ":lower_complex_inc_gen", "@llvm-project//llvm:Support", - "@llvm-project//mlir:Analysis", "@llvm-project//mlir:IR", "@llvm-project//mlir:Pass", "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", ], alwayslink = 1, diff --git a/tensorflow/compiler/mlir/tools/kernel_gen/BUILD b/tensorflow/compiler/mlir/tools/kernel_gen/BUILD index 05d4219147d..67dbead7269 100644 --- a/tensorflow/compiler/mlir/tools/kernel_gen/BUILD +++ b/tensorflow/compiler/mlir/tools/kernel_gen/BUILD @@ -34,7 +34,7 @@ cc_library( copts = if_cuda_is_configured(["-DGOOGLE_CUDA=1"]) + if_rocm_is_configured(["-DTENSORFLOW_USE_ROCM=1"]), deps = [ "//tensorflow/compiler/mlir/hlo", - "//tensorflow/compiler/mlir/hlo:all_passes", + "//tensorflow/compiler/mlir/hlo:all_passes", # buildcleaner: keep "//tensorflow/compiler/mlir/hlo:hlo_legalize_to_lhlo", "//tensorflow/compiler/mlir/hlo:legalize_to_linalg", "//tensorflow/compiler/mlir/hlo:legalize_trigonometric_to_approximation", @@ -44,31 +44,23 @@ cc_library( "//tensorflow/compiler/mlir/hlo:lhlo_legalize_to_gpu", "//tensorflow/compiler/mlir/hlo:transform_unranked_hlo", # buildcleaner: keep "//tensorflow/compiler/mlir/tensorflow", - "//tensorflow/compiler/mlir/tensorflow:compile_mlir_util_no_tf_dialect_passes", "//tensorflow/compiler/mlir/tensorflow:dump_mlir_util", "//tensorflow/compiler/mlir/tools/kernel_gen/transforms:passes", "//tensorflow/compiler/mlir/xla:xla_legalize_tf", - "//tensorflow/compiler/xla:debug_options_flags", - "//tensorflow/compiler/xla:status", "//tensorflow/compiler/xla:statusor", "//tensorflow/compiler/xla:util", - "//tensorflow/compiler/xla/service/gpu:stream_executor_util", - "//tensorflow/compiler/xla/service/gpu:target_constants", - "//tensorflow/compiler/xla/service/gpu/llvm_gpu_backend", "//tensorflow/compiler/xla/service/mlir_gpu:kernel_lowering", "//tensorflow/compiler/xla/service/mlir_gpu:passes", "//tensorflow/core:lib", - "//tensorflow/core/platform:cuda_libdevice_path", "@llvm-project//llvm:Support", "@llvm-project//mlir:AffineToStandard", - "@llvm-project//mlir:AllPassesAndDialectsNoRegistration", + "@llvm-project//mlir:AllPassesAndDialectsNoRegistration", # buildcleaner: keep "@llvm-project//mlir:GPUDialect", "@llvm-project//mlir:GPUToGPURuntimeTransforms", "@llvm-project//mlir:GPUToNVVMTransforms", "@llvm-project//mlir:GPUTransforms", "@llvm-project//mlir:IR", "@llvm-project//mlir:LLVMDialect", - "@llvm-project//mlir:LinalgOps", "@llvm-project//mlir:LinalgTransforms", "@llvm-project//mlir:NVVMDialect", "@llvm-project//mlir:Parser", @@ -80,7 +72,6 @@ cc_library( "@llvm-project//mlir:ShapeToStandard", "@llvm-project//mlir:ShapeTransforms", "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", ], ) diff --git a/tensorflow/compiler/mlir/tools/kernel_gen/transforms/BUILD b/tensorflow/compiler/mlir/tools/kernel_gen/transforms/BUILD index 34944514a62..18e7a3f858a 100644 --- a/tensorflow/compiler/mlir/tools/kernel_gen/transforms/BUILD +++ b/tensorflow/compiler/mlir/tools/kernel_gen/transforms/BUILD @@ -26,7 +26,6 @@ cc_library( "@llvm-project//mlir:LLVMDialect", "@llvm-project//mlir:LLVMTransforms", "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", ], ) @@ -37,9 +36,7 @@ cc_library( hdrs = ["rewriters.h"], deps = [ "@llvm-project//mlir:IR", - "@llvm-project//mlir:Pass", "@llvm-project//mlir:SCFDialect", - "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", ], ) @@ -51,10 +48,7 @@ cc_library( deps = [ "//tensorflow/compiler/mlir/tools/kernel_gen/ir:tf_framework_ops", "@llvm-project//mlir:IR", - "@llvm-project//mlir:LLVMDialect", - "@llvm-project//mlir:LLVMTransforms", "@llvm-project//mlir:StandardOps", - "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", ], ) @@ -107,7 +101,6 @@ cc_library( "@llvm-project//mlir:IR", "@llvm-project//mlir:LLVMDialect", "@llvm-project//mlir:LLVMTransforms", - "@llvm-project//mlir:LinalgTransforms", "@llvm-project//mlir:LinalgOps", "@llvm-project//mlir:Pass", "@llvm-project//mlir:SCFDialect", @@ -120,7 +113,6 @@ cc_library( "@llvm-project//mlir:ShapeTransforms", "@llvm-project//mlir:StandardOps", "@llvm-project//mlir:StandardOpsTransforms", - "@llvm-project//mlir:AllPassesAndDialects", "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", "@llvm-project//llvm:TransformUtils",