[HLO Componentization] Create pass sub-component

PiperOrigin-RevId: 675287627
openxla · Sep 16, 2024 · dc284c4 · dc284c4
1 parent 6653168
commit dc284c4
Show file tree

Hide file tree

Showing 117 changed files with 237 additions and 241 deletions.
diff --git a/xla/service/BUILD b/xla/service/BUILD
diff --git a/xla/service/add_original_value.h b/xla/service/add_original_value.h
@@ -17,7 +17,7 @@ limitations under the License.
 #define XLA_SERVICE_ADD_ORIGINAL_VALUE_H_
 
 #include "absl/status/statusor.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/algebraic_simplifier.h b/xla/service/algebraic_simplifier.h
@@ -35,8 +35,8 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_instructions.h"
 #include "xla/hlo/ir/hlo_module.h"
 #include "xla/hlo/ir/hlo_opcode.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/literal.h"
-#include "xla/service/hlo_pass_interface.h"
 #include "xla/shape.h"
 #include "xla/shape_util.h"
 #include "xla/util.h"

diff --git a/xla/service/algebraic_simplifier_test.cc b/xla/service/algebraic_simplifier_test.cc
@@ -42,13 +42,13 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_instructions.h"
 #include "xla/hlo/ir/hlo_opcode.h"
+#include "xla/hlo/pass/hlo_pass_fix.h"
 #include "xla/layout_util.h"
 #include "xla/literal.h"
 #include "xla/literal_util.h"
 #include "xla/primitive_util.h"
 #include "xla/service/hlo_creation_utils.h"
 #include "xla/service/hlo_parser.h"
-#include "xla/service/hlo_pass_fix.h"
 #include "xla/service/host_memory_offload_annotations.h"
 #include "xla/service/layout_assignment.h"
 #include "xla/service/pattern_matcher.h"

diff --git a/xla/service/all_gather_broadcast_reorder.h b/xla/service/all_gather_broadcast_reorder.h
@@ -18,7 +18,7 @@ limitations under the License.
 
 #include "absl/status/statusor.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/all_gather_combiner.h b/xla/service/all_gather_combiner.h
@@ -19,8 +19,8 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/service/hlo_domain_map.h"
-#include "xla/service/hlo_pass_interface.h"
 #include "xla/xla_data.pb.h"
 
 namespace xla {

diff --git a/xla/service/all_gather_decomposer.h b/xla/service/all_gather_decomposer.h
@@ -28,8 +28,8 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_instructions.h"
 #include "xla/hlo/ir/hlo_module.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/service/collective_ops_utils.h"
-#include "xla/service/hlo_pass_interface.h"
 #include "xla/shape.h"
 
 namespace xla {

diff --git a/xla/service/all_reduce_combiner.h b/xla/service/all_reduce_combiner.h
@@ -20,7 +20,7 @@ limitations under the License.
 #include "absl/strings/string_view.h"
 #include "xla/array2d.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/xla_data.pb.h"
 
 namespace xla {

diff --git a/xla/service/all_reduce_contiguous.h b/xla/service/all_reduce_contiguous.h
@@ -18,7 +18,7 @@ limitations under the License.
 
 #include "absl/status/statusor.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/all_reduce_folder.h b/xla/service/all_reduce_folder.h
@@ -18,7 +18,7 @@ limitations under the License.
 
 #include "absl/status/statusor.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/all_reduce_reassociate.h b/xla/service/all_reduce_reassociate.h
@@ -18,7 +18,7 @@ limitations under the License.
 
 #include "absl/status/statusor.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/all_reduce_simplifier.h b/xla/service/all_reduce_simplifier.h
@@ -18,7 +18,7 @@ limitations under the License.
 
 #include "absl/status/statusor.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/ar_crs_combiner.h b/xla/service/ar_crs_combiner.h
@@ -30,8 +30,8 @@ limitations under the License.
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_module.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/service/call_graph.h"
-#include "xla/service/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/async_collective_creator.h b/xla/service/async_collective_creator.h
@@ -20,7 +20,7 @@ limitations under the License.
 #include <utility>
 #include <vector>
 
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/batch_dot_simplification.h b/xla/service/batch_dot_simplification.h
@@ -17,7 +17,7 @@ limitations under the License.
 #define XLA_SERVICE_BATCH_DOT_SIMPLIFICATION_H_
 
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 // Simplifies batch dot operations.

diff --git a/xla/service/batchnorm_expander.h b/xla/service/batchnorm_expander.h
@@ -19,7 +19,7 @@ limitations under the License.
 #include <utility>
 
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/batchnorm_expander_test.cc b/xla/service/batchnorm_expander_test.cc
@@ -25,7 +25,6 @@ limitations under the License.
 #include "xla/layout_util.h"
 #include "xla/literal.h"
 #include "xla/service/hlo_parser.h"
-#include "xla/service/hlo_pass_fix.h"
 #include "xla/shape_util.h"
 #include "xla/test.h"
 #include "xla/tests/hlo_test_base.h"

diff --git a/xla/service/bfloat16_conversion_folding.h b/xla/service/bfloat16_conversion_folding.h
@@ -17,8 +17,8 @@ limitations under the License.
 #define XLA_SERVICE_BFLOAT16_CONVERSION_FOLDING_H_
 
 #include "xla/hlo/ir/hlo_module.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/service/float_support.h"
-#include "xla/service/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/bfloat16_propagation.h b/xla/service/bfloat16_propagation.h
@@ -23,9 +23,9 @@ limitations under the License.
 #include "absl/container/flat_hash_set.h"
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_module.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/service/float_support.h"
 #include "xla/service/hlo_dataflow_analysis.h"
-#include "xla/service/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/broadcast_canonicalizer.h b/xla/service/broadcast_canonicalizer.h
@@ -18,7 +18,7 @@ limitations under the License.
 
 #include <optional>
 
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/call_inliner.h b/xla/service/call_inliner.h
@@ -22,7 +22,7 @@ limitations under the License.
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/change_op_data_type.h b/xla/service/change_op_data_type.h
@@ -20,7 +20,7 @@ limitations under the License.
 #include <memory>
 #include <utility>
 
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/collective_permute_decomposer.h b/xla/service/collective_permute_decomposer.h
@@ -17,7 +17,7 @@ limitations under the License.
 #define XLA_SERVICE_COLLECTIVE_PERMUTE_DECOMPOSER_H_
 
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/collective_pipeliner.h b/xla/service/collective_pipeliner.h
@@ -20,7 +20,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/collective_pipeliner_test.cc b/xla/service/collective_pipeliner_test.cc
@@ -36,9 +36,9 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_instructions.h"
 #include "xla/hlo/ir/hlo_module.h"
 #include "xla/hlo/ir/hlo_opcode.h"
+#include "xla/hlo/pass/hlo_pass_pipeline.h"
 #include "xla/hlo/utils/hlo_matchers.h"
 #include "xla/service/hlo_parser.h"
-#include "xla/service/hlo_pass_pipeline.h"
 #include "xla/service/host_memory_offload_annotations.h"
 #include "xla/tests/filecheck.h"
 #include "xla/tests/hlo_test_base.h"

diff --git a/xla/service/collective_quantizer.h b/xla/service/collective_quantizer.h
@@ -20,7 +20,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/collective_transformation_reorderer.h b/xla/service/collective_transformation_reorderer.h
@@ -22,7 +22,7 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_instructions.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/collectives_schedule_linearizer.h b/xla/service/collectives_schedule_linearizer.h
@@ -19,7 +19,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/util.h"
 
 namespace xla {

diff --git a/xla/service/conditional_canonicalizer.h b/xla/service/conditional_canonicalizer.h
@@ -18,7 +18,7 @@ limitations under the License.
 #include <utility>
 
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/conditional_code_motion.cc b/xla/service/conditional_code_motion.cc
@@ -38,11 +38,11 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_instructions.h"
 #include "xla/hlo/ir/hlo_opcode.h"
+#include "xla/hlo/pass/hlo_pass_pipeline.h"
 #include "xla/literal.h"
 #include "xla/map_util.h"
 #include "xla/service/hlo_cse.h"
 #include "xla/service/hlo_dce.h"
-#include "xla/service/hlo_pass_pipeline.h"
 #include "xla/service/hlo_verifier.h"
 #include "xla/service/tuple_simplifier.h"
 #include "xla/shape.h"

diff --git a/xla/service/conditional_code_motion.h b/xla/service/conditional_code_motion.h
@@ -21,7 +21,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/conditional_simplifier.h b/xla/service/conditional_simplifier.h
@@ -19,7 +19,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/conditional_to_select.h b/xla/service/conditional_to_select.h
@@ -17,7 +17,7 @@ limitations under the License.
 #define XLA_SERVICE_CONDITIONAL_TO_SELECT_H_
 
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/convert_async_collectives_to_sync.h b/xla/service/convert_async_collectives_to_sync.h
@@ -25,7 +25,7 @@ limitations under the License.
 #include "absl/types/span.h"
 #include "xla/hlo/ir/hlo_computation.h"
 #include "xla/hlo/ir/hlo_instruction.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/util.h"
 
 namespace xla {

diff --git a/xla/service/convert_memory_placement_to_internal_annotations.h b/xla/service/convert_memory_placement_to_internal_annotations.h
@@ -24,7 +24,7 @@
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/convert_mover.h b/xla/service/convert_mover.h
@@ -23,7 +23,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/convolution_group_converter.h b/xla/service/convolution_group_converter.h
@@ -20,7 +20,7 @@ limitations under the License.
 
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/status_macros.h"
 
 namespace xla {

diff --git a/xla/service/copy_insertion.h b/xla/service/copy_insertion.h
@@ -23,10 +23,10 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_computation.h"
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_module.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/service/call_graph.h"
 #include "xla/service/hlo_alias_analysis.h"
 #include "xla/service/hlo_dataflow_analysis.h"
-#include "xla/service/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/defuser.h b/xla/service/defuser.h
@@ -19,7 +19,7 @@ limitations under the License.
 #include <utility>
 
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/despecializer.h b/xla/service/despecializer.h
@@ -25,8 +25,8 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_casting_utils.h"
 #include "xla/hlo/ir/hlo_instructions.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
-#include "xla/service/hlo_pass_pipeline.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_pipeline.h"
 
 namespace xla {
 

diff --git a/xla/service/dot_decomposer.h b/xla/service/dot_decomposer.h
@@ -20,7 +20,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 

diff --git a/xla/service/dot_dimension_merger.h b/xla/service/dot_dimension_merger.h
@@ -20,7 +20,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {