Skip to content

Commit 7cb5ee6

Browse files
sdasgup3tensorflower-gardener
authored andcommitted
[HLO Componentization] Create hlo/builder sub-component (Phase II).
This CL takes care of 1. Migrating external projects dependencies from xla/client --> xla/hlo/builder Phase I takes care of 1. Migrating xla/translate --> xla/hlo/translate 2. Setting up build aliases in xla/translate ensuring external dependencies are still satisfied. PiperOrigin-RevId: 685539579
1 parent f2a0407 commit 7cb5ee6

File tree

22 files changed

+33
-33
lines changed

22 files changed

+33
-33
lines changed

tensorflow/compiler/mlir/tensorflow/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -769,7 +769,7 @@ cc_library(
769769
"@llvm-project//mlir:Support",
770770
"@local_xla//xla:status_macros",
771771
"@local_xla//xla:xla_data_proto_cc",
772-
"@local_xla//xla/client:sharding_builder",
772+
"@local_xla//xla/hlo/builder:sharding_builder",
773773
"@local_xla//xla/hlo/ir:hlo",
774774
"@local_xla//xla/hlo/parser:hlo_parser",
775775
],

tensorflow/compiler/mlir/tensorflow/transforms/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -686,7 +686,7 @@ cc_library(
686686
"@local_xla//xla:window_util",
687687
"@local_xla//xla:xla_data_proto_cc",
688688
"@local_xla//xla:xla_proto_cc",
689-
"@local_xla//xla/client:sharding_builder",
689+
"@local_xla//xla/hlo/builder:sharding_builder",
690690
"@local_xla//xla/mlir_hlo",
691691
"@stablehlo//:chlo_ops",
692692
"@stablehlo//:stablehlo_ops",

tensorflow/compiler/mlir/tf2xla/internal/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ tf_cc_test(
5656
"@com_google_absl//absl/status",
5757
"@com_google_googletest//:gtest_main",
5858
"@local_tsl//tsl/platform:statusor",
59-
"@local_xla//xla/client:xla_computation",
59+
"@local_xla//xla/hlo/builder:xla_computation",
6060
"@local_xla//xla/service:hlo_proto_cc",
6161
],
6262
)

tensorflow/compiler/mlir/tf2xla/internal/passes/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -361,7 +361,7 @@ cc_library(
361361
"@llvm-project//mlir:Support",
362362
"@llvm-project//mlir:TransformUtils",
363363
"@local_xla//xla:xla_data_proto_cc",
364-
"@local_xla//xla/client:sharding_builder",
364+
"@local_xla//xla/hlo/builder:sharding_builder",
365365
],
366366
)
367367

tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ limitations under the License.
4949
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
5050
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h"
5151
#include "tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h"
52-
#include "xla/client/sharding_builder.h"
52+
#include "xla/hlo/builder/sharding_builder.h"
5353
#include "xla/xla_data.pb.h"
5454
#include "tsl/platform/errors.h"
5555

tensorflow/compiler/mlir/tf2xla/internal/test_matchers_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ limitations under the License.
2323
#include "absl/status/status.h"
2424
#include "tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h"
2525
#include "tensorflow/compiler/tf2xla/xla_helpers.h"
26-
#include "xla/client/xla_computation.h"
26+
#include "xla/hlo/builder/xla_computation.h"
2727
#include "xla/service/hlo.pb.h"
2828
#include "tensorflow/core/lib/monitoring/cell_reader.h"
2929
#include "tensorflow/core/lib/monitoring/counter.h"

third_party/xla/xla/hlo/evaluator/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ xla_cc_test(
134134
"//xla:types",
135135
"//xla:util",
136136
"//xla:xla_data_proto_cc",
137-
"//xla/client:xla_builder",
137+
"//xla/hlo/builder:xla_builder",
138138
"//xla/hlo/ir:hlo",
139139
"//xla/service:call_graph",
140140
"//xla/service:dynamic_dimension_inference",

third_party/xla/xla/hlo/evaluator/hlo_evaluator_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,10 +38,10 @@ limitations under the License.
3838
#include "xla/array2d.h"
3939
#include "xla/array3d.h"
4040
#include "xla/array4d.h"
41-
#include "xla/client/xla_builder.h"
4241
#include "xla/comparison_util.h"
4342
#include "xla/debug_options_flags.h"
4443
#include "xla/error_spec.h"
44+
#include "xla/hlo/builder/xla_builder.h"
4545
#include "xla/hlo/ir/hlo_computation.h"
4646
#include "xla/hlo/ir/hlo_instruction.h"
4747
#include "xla/hlo/ir/hlo_opcode.h"

third_party/xla/xla/pjrt/c/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ cc_library(
133133
"//xla:util",
134134
"//xla:xla_data_proto_cc",
135135
"//xla:xla_proto_cc",
136-
"//xla/client:xla_computation",
136+
"//xla/hlo/builder:xla_computation",
137137
"//xla/hlo/ir:hlo",
138138
"//xla/pjrt:compile_options_proto_cc",
139139
"//xla/pjrt:mlir_to_hlo",

third_party/xla/xla/pjrt/c/pjrt_c_api_wrapper_impl.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ limitations under the License.
4141
#include "mlir/IR/BuiltinOps.h"
4242
#include "mlir/IR/MLIRContext.h"
4343
#include "mlir/IR/OwningOpRef.h"
44-
#include "xla/client/xla_computation.h"
44+
#include "xla/hlo/builder/xla_computation.h"
4545
#include "xla/hlo/ir/hlo_module.h"
4646
#include "xla/layout.h"
4747
#include "xla/literal.h"

0 commit comments

Comments
 (0)