Skip to content

Commit a700a68

Browse files
sdasgup3tensorflower-gardener
authored andcommitted
[HLO Componentization] Create hlo/transforms/ sub-component (Phase II).
This CL takes care of 1. Migrating external projects dependencies from xla/service --> xla/hlo/transforms Phase I takes care of 1. Migrating xla/service/<passes> --> xla/hlo/transforms/<passes> 2. Setting up build aliases in xla/service ensuring external dependencies are still satisfied. PiperOrigin-RevId: 687723365
1 parent da15624 commit a700a68

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

third_party/xla/xla/service/spmd/shardy/BUILD

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,14 +36,14 @@ cc_library(
3636
"//xla:xla_data_proto_cc",
3737
"//xla/hlo/ir:hlo",
3838
"//xla/hlo/pass:hlo_pass",
39+
"//xla/hlo/transforms:hlo_dce",
40+
"//xla/hlo/transforms:tuple_simplifier",
3941
"//xla/hlo/translate/hlo_to_mhlo:hlo_to_mlir_hlo",
4042
"//xla/hlo/translate/mhlo_to_hlo:mlir_hlo_to_hlo",
4143
"//xla/hlo/utils:hlo_sharding_util",
4244
"//xla/mlir_hlo:mhlo_passes",
4345
"//xla/service:computation_layout",
44-
"//xla/service:hlo_dce",
4546
"//xla/service:hlo_proto_cc",
46-
"//xla/service:tuple_simplifier",
4747
"//xla/service/llvm_ir:llvm_util",
4848
"//xla/service/spmd/shardy/mhlo_round_trip:mhlo_export",
4949
"//xla/service/spmd/shardy/mhlo_round_trip:mhlo_import",

third_party/xla/xla/service/spmd/shardy/shardy_xla_pass.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@ limitations under the License.
4646
#include "xla/hlo/ir/hlo_computation.h"
4747
#include "xla/hlo/ir/hlo_input_output_alias_config.h"
4848
#include "xla/hlo/ir/hlo_module.h"
49+
#include "xla/hlo/transforms/simplifiers/hlo_dce.h"
50+
#include "xla/hlo/transforms/simplifiers/tuple_simplifier.h"
4951
#include "xla/hlo/translate/hlo_to_mhlo/hlo_to_mlir_hlo.h"
5052
#include "xla/hlo/translate/mhlo_to_hlo/mlir_hlo_to_hlo.h"
5153
#include "xla/hlo/utils/hlo_sharding_util.h"
@@ -54,14 +56,12 @@ limitations under the License.
5456
#include "xla/mlir_hlo/mhlo/transforms/passes.h"
5557
#include "xla/service/computation_layout.h"
5658
#include "xla/service/hlo.pb.h"
57-
#include "xla/service/hlo_dce.h"
5859
#include "xla/service/llvm_ir/llvm_util.h"
5960
#include "xla/service/spmd/shardy/constants.h"
6061
#include "xla/service/spmd/shardy/mhlo_round_trip/mhlo_export.h"
6162
#include "xla/service/spmd/shardy/mhlo_round_trip/mhlo_import.h"
6263
#include "xla/service/spmd/shardy/sdy_round_trip/pipelines.h"
6364
#include "xla/service/spmd/shardy/utils.h"
64-
#include "xla/service/tuple_simplifier.h"
6565
#include "xla/shape.h"
6666
#include "xla/shape_layout.h"
6767
#include "xla/shape_util.h"

0 commit comments

Comments
 (0)