From 7f442d94f78da814140e1cd7150a6dde4fd7a8f8 Mon Sep 17 00:00:00 2001 From: Sandeep Dasgupta Date: Wed, 2 Oct 2024 16:31:32 -0700 Subject: [PATCH] [HLO Componentization] Create hlo/parser sub-component (Phase II). This CL takes care of 1. Migrating external projects dependencies from xla/service --> xla/hlo/parser Phase I takes care of 1. Migrating xla/service --> xla/hlo/parser 2. Setting up build aliases in xla/service ensuring external dependencies are still satisfied. PiperOrigin-RevId: 681635267 --- xla/hlo/experimental/auto_sharding/BUILD | 4 ++-- xla/hlo/experimental/auto_sharding/auto_sharding_runner.cc | 2 +- xla/hlo/experimental/auto_sharding/auto_sharding_test.cc | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/xla/hlo/experimental/auto_sharding/BUILD b/xla/hlo/experimental/auto_sharding/BUILD index 35cd7a5f71d60..d8b80adb8140b 100644 --- a/xla/hlo/experimental/auto_sharding/BUILD +++ b/xla/hlo/experimental/auto_sharding/BUILD @@ -333,7 +333,7 @@ xla_cc_binary( deps = [ ":auto_sharding", "//xla/hlo/ir:hlo", - "//xla/service:hlo_parser", + "//xla/hlo/parser:hlo_parser", "//xla/tools:hlo_module_loader", "@com_google_absl//absl/status", "@tsl//tsl/platform:platform_port", @@ -383,12 +383,12 @@ xla_cc_test( ":auto_sharding_strategy", ":auto_sharding_util", "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", "//xla/hlo/utils:hlo_live_range", "//xla/hlo/utils:hlo_matchers", "//xla/service:buffer_value", "//xla/service:hlo_alias_analysis", "//xla/service:hlo_memory_scheduler", - "//xla/service:hlo_parser", "//xla/service:hlo_value", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", diff --git a/xla/hlo/experimental/auto_sharding/auto_sharding_runner.cc b/xla/hlo/experimental/auto_sharding/auto_sharding_runner.cc index e82bf2a5e3175..753bc8a70d54a 100644 --- a/xla/hlo/experimental/auto_sharding/auto_sharding_runner.cc +++ b/xla/hlo/experimental/auto_sharding/auto_sharding_runner.cc @@ -20,7 +20,7 @@ limitations under the License. #include "absl/status/status.h" #include "xla/hlo/experimental/auto_sharding/auto_sharding.h" #include "xla/hlo/ir/hlo_module.h" -#include "xla/service/hlo_parser.h" +#include "xla/hlo/parser/hlo_parser.h" #include "xla/tools/hlo_module_loader.h" #include "tsl/platform/init_main.h" diff --git a/xla/hlo/experimental/auto_sharding/auto_sharding_test.cc b/xla/hlo/experimental/auto_sharding/auto_sharding_test.cc index aa3e45ea54b0b..8ef96bb2564be 100644 --- a/xla/hlo/experimental/auto_sharding/auto_sharding_test.cc +++ b/xla/hlo/experimental/auto_sharding/auto_sharding_test.cc @@ -41,12 +41,12 @@ limitations under the License. #include "xla/hlo/ir/hlo_opcode.h" #include "xla/hlo/ir/hlo_schedule.h" #include "xla/hlo/ir/hlo_sharding.h" +#include "xla/hlo/parser/hlo_parser.h" #include "xla/hlo/utils/hlo_live_range.h" #include "xla/hlo/utils/hlo_matchers.h" #include "xla/service/buffer_value.h" #include "xla/service/hlo_alias_analysis.h" #include "xla/service/hlo_memory_scheduler.h" -#include "xla/service/hlo_parser.h" #include "xla/service/hlo_value.h" #include "xla/tests/hlo_test_base.h" #include "xla/tsl/lib/core/status_test_util.h"