[HLO Componentization] Create pass sub-component

PiperOrigin-RevId: 674391570
openxla · Sep 17, 2024 · e7de05e · e7de05e
1 parent 8ace4ee
commit e7de05e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/xla/hlo/experimental/auto_sharding/BUILD b/xla/hlo/experimental/auto_sharding/BUILD
@@ -49,6 +49,7 @@ cc_library(
         "//xla:util",
         "//xla:xla_data_proto_cc",
         "//xla/hlo/ir:hlo",
+        "//xla/hlo/pass:hlo_pass",
         "//xla/hlo/transforms:hlo_constant_splitter",
         "//xla/hlo/utils:hlo_live_range",
         "//xla/hlo/utils:hlo_sharding_util",
@@ -62,7 +63,6 @@ cc_library(
         "//xla/service:hlo_cost_analysis",
         "//xla/service:hlo_dce",
         "//xla/service:hlo_memory_scheduler",
-        "//xla/service:hlo_pass",
         "//xla/service:hlo_value",
         "//xla/service:optimize_input_output_buffer_alias",
         "//xla/service:sharding_propagation",

diff --git a/xla/hlo/experimental/auto_sharding/auto_sharding.h b/xla/hlo/experimental/auto_sharding/auto_sharding.h
@@ -41,11 +41,11 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_schedule.h"
 #include "xla/hlo/ir/hlo_sharding.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/hlo/utils/hlo_live_range.h"
 #include "xla/service/call_graph.h"
 #include "xla/service/hlo_alias_analysis.h"
 #include "xla/service/hlo_cost_analysis.h"
-#include "xla/service/hlo_pass_interface.h"
 #include "xla/shape.h"
 
 namespace xla {