diff --git a/xla/service/BUILD b/xla/service/BUILD index d37fa909723f6..36011384a5e10 100644 --- a/xla/service/BUILD +++ b/xla/service/BUILD @@ -116,12 +116,12 @@ cc_library( srcs = ["async_collective_creator.cc"], hdrs = ["async_collective_creator.h"], deps = [ - ":hlo_pass", ":shape_inference", "//xla:frontend_attributes", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/log", "@tsl//tsl/platform:errors", @@ -180,11 +180,11 @@ cc_library( deps = [ ":all_reduce_key", ":collective_ops_utils", - ":hlo_pass", ":pattern_matcher", "//xla:literal", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", @@ -222,8 +222,8 @@ cc_library( hdrs = ["all_reduce_folder.h"], deps = [ ":all_reduce_key", - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -269,8 +269,8 @@ cc_library( hdrs = ["broadcast_canonicalizer.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", @@ -300,11 +300,11 @@ cc_library( deps = [ ":float_support", ":hlo_dataflow_analysis", - ":hlo_pass", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", "@com_google_absl//absl/strings:string_view", @@ -339,12 +339,12 @@ cc_library( ":call_graph", ":float_support", ":hlo_dce", - ":hlo_pass", ":tuple_simplifier", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", @@ -385,13 +385,13 @@ cc_library( ":float_support", ":hlo_dataflow_analysis", ":hlo_dce", - ":hlo_pass", ":tuple_simplifier", "//xla:literal", "//xla:shape_tree", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/cleanup", "@com_google_absl//absl/container:flat_hash_map", @@ -430,10 +430,10 @@ cc_library( hdrs = ["collective_permute_decomposer.h"], deps = [ ":collective_ops_utils", - ":hlo_pass", "//xla:shape_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/service/gpu:backend_configs_cc", "//xla/service/graphcycles", "@com_google_absl//absl/status", @@ -485,11 +485,11 @@ cc_library( srcs = ["convert_async_collectives_to_sync.cc"], hdrs = ["convert_async_collectives_to_sync.h"], deps = [ - ":hlo_pass", "//xla:status_macros", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -550,10 +550,12 @@ cc_library( srcs = ["collective_pipeliner.cc"], hdrs = ["collective_pipeliner.h"], deps = [ + ":call_graph", ":collective_ops_utils", ":constant_value", ":hlo_dce", - ":hlo_pass", + ":hlo_parser", + ":tuple_points_to_analysis", ":value_range", "//xla:comparison_util", "//xla:literal", @@ -564,10 +566,8 @@ cc_library( "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_instruction_utils", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", - "//xla/service:call_graph", - "//xla/service:hlo_parser", - "//xla/service:tuple_points_to_analysis", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -590,10 +590,10 @@ xla_cc_test( deps = [ ":collective_pipeliner", ":hlo_parser", - ":hlo_pass_pipeline", ":host_memory_offload_annotations_hdr", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/hlo/utils:hlo_matchers", "//xla/tests:filecheck", "//xla/tests:hlo_test_base", @@ -612,10 +612,10 @@ cc_library( srcs = ["collective_quantizer.cc"], hdrs = ["collective_quantizer.h"], deps = [ - ":hlo_pass", ":pattern_matcher", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -770,7 +770,6 @@ cc_library( ":custom_call_sharding_helper", ":dot_as_convolution_util", ":hlo_graph_dumper", - ":hlo_pass", ":host_memory_offload_annotations_hdr", "//xla:array", "//xla:protobuf_util", @@ -781,6 +780,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_sharding_util", "//xla/service/spmd:shard_barrier_partitioner", "@com_google_absl//absl/algorithm:container", @@ -834,8 +834,8 @@ cc_library( "sharding_remover.h", ], deps = [ - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/service/spmd/shardy:constants", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -1142,9 +1142,9 @@ cc_library( hdrs = ["flatten_call_graph.h"], deps = [ ":call_graph", - ":hlo_pass", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", @@ -1163,10 +1163,10 @@ cc_library( ":call_graph", ":hlo_dce", ":hlo_domain_isolator", - ":hlo_pass", "//xla:status_macros", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", @@ -1207,11 +1207,11 @@ cc_library( srcs = ["hlo_computation_deduplicator.cc"], hdrs = ["hlo_computation_deduplicator.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla:status_macros", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", @@ -1227,7 +1227,6 @@ xla_cc_test( srcs = ["hlo_computation_deduplicator_test.cc"], deps = [ ":hlo_computation_deduplicator", - ":hlo_pass", "//xla:literal", "//xla:literal_util", "//xla:shape_util", @@ -1235,6 +1234,7 @@ xla_cc_test( "//xla:types", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_matchers", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", @@ -1436,7 +1436,6 @@ cc_library( ":hlo_alias_analysis", ":hlo_buffer", ":hlo_cost_analysis", - ":hlo_pass", ":hlo_value", "//xla:debug_options_flags", "//xla:shape_util", @@ -1444,6 +1443,7 @@ cc_library( "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_reachability", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -1490,10 +1490,10 @@ cc_library( hdrs = ["p2p_schedule_preparation.h"], deps = [ ":collective_ops_utils", - ":hlo_pass", "//xla:util", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_reachability", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -2150,7 +2150,6 @@ cc_library( deps = [ ":buffer_value", ":hlo_alias_analysis", - ":hlo_pass", ":logical_buffer", ":tuple_points_to_analysis", "//xla:shape_util", @@ -2158,6 +2157,7 @@ cc_library( "//xla:types", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/service/heap_simulator", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -2223,13 +2223,13 @@ cc_library( ":hlo_dataflow_analysis", ":hlo_graph_dumper", ":hlo_module_config", - ":hlo_pass", ":pattern_matcher", "//xla:debug_options_flags", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_reachability", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -2269,12 +2269,12 @@ cc_library( deps = [ ":hlo_dataflow_analysis", ":hlo_dce", - ":hlo_pass", "//xla:debug_options_flags", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_reachability", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", @@ -2364,13 +2364,13 @@ cc_library( srcs = ["batchnorm_expander.cc"], hdrs = ["batchnorm_expander.h"], deps = [ - ":hlo_pass", "//xla:literal", "//xla:literal_util", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/functional:function_ref", "@com_google_absl//absl/log:check", @@ -2388,9 +2388,9 @@ cc_library( srcs = ["op_expander_pass.cc"], hdrs = ["op_expander_pass.h"], deps = [ - ":hlo_pass", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log:check", @@ -2695,7 +2695,6 @@ xla_test( deps = [ ":batchnorm_expander", ":hlo_parser", - ":hlo_pass", "//xla:literal", "//xla:shape_util", "//xla:test", @@ -2717,7 +2716,6 @@ cc_library( ":hlo_cost_analysis", ":hlo_creation_utils", ":hlo_module_config", - ":hlo_pass", ":host_memory_offload_annotations_hdr", ":pattern_matcher", ":shape_inference", @@ -2733,6 +2731,7 @@ cc_library( "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_instruction_utils", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_sharding_util", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -2756,13 +2755,13 @@ cc_library( srcs = ["tree_reduction_rewriter.cc"], hdrs = ["tree_reduction_rewriter.h"], deps = [ - ":hlo_pass", ":shape_inference", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/client:padding", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -2778,7 +2777,6 @@ xla_cc_test( ":algebraic_simplifier", ":hlo_creation_utils", ":hlo_parser", - ":hlo_pass", ":host_memory_offload_annotations_hdr", ":layout_assignment", ":pattern_matcher", @@ -2793,6 +2791,7 @@ xla_cc_test( "//xla:window_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", # fixdeps: keep "//xla/tsl/lib/core:status_test_util", @@ -2821,10 +2820,10 @@ cc_library( srcs = ["simplify_fp_conversions.cc"], hdrs = ["simplify_fp_conversions.h"], deps = [ - ":hlo_pass", "//xla:comparison_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -2855,12 +2854,12 @@ cc_library( hdrs = ["logistic_expander.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", ":op_expander_pass", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:logging", @@ -2890,10 +2889,10 @@ cc_library( srcs = ["collectives_schedule_linearizer.cc"], hdrs = ["collectives_schedule_linearizer.h"], deps = [ - ":hlo_pass", "//xla:util", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_reachability", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", @@ -2956,10 +2955,10 @@ cc_library( srcs = ["all_gather_broadcast_reorder.cc"], hdrs = ["all_gather_broadcast_reorder.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", @@ -3026,11 +3025,11 @@ cc_library( deps = [ ":collective_combiner_utils", ":hlo_domain_map", - ":hlo_pass", "//xla:shape_util", "//xla:status_macros", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "//xla/hlo/utils:hlo_sharding_util", "@com_google_absl//absl/container:flat_hash_set", @@ -3064,12 +3063,12 @@ cc_library( ":all_reduce_key", ":collective_combiner_utils", ":hlo_domain_map", - ":hlo_pass", "//xla:array2d", "//xla:shape_util", "//xla:status_macros", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "//xla/hlo/utils:hlo_sharding_util", "@com_google_absl//absl/container:flat_hash_map", @@ -3101,10 +3100,10 @@ cc_library( srcs = ["all_reduce_contiguous.cc"], hdrs = ["all_reduce_contiguous.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla:status_macros", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", @@ -3133,13 +3132,13 @@ cc_library( ":collective_combiner_utils", ":collective_ops_utils", ":hlo_domain_map", - ":hlo_pass", ":shape_inference", "//xla:shape_util", "//xla:status_macros", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_reachability", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -3170,11 +3169,11 @@ cc_library( deps = [ ":collective_ops_utils", ":hlo_module_config", - ":hlo_pass", ":hlo_replication_analysis", "//xla:literal_util", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", @@ -3213,10 +3212,10 @@ cc_library( ":collective_decomposer_utils", ":collective_ops_utils", ":hlo_module_config", - ":hlo_pass", "//xla:literal_util", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/status:statusor", ], @@ -3245,8 +3244,8 @@ cc_library( ":all_reduce_key", ":collective_ops_utils", ":hlo_domain_map", - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/status:statusor", "@tsl//tsl/platform:errors", @@ -3271,8 +3270,8 @@ cc_library( hdrs = ["batch_dot_simplification.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", ], ) @@ -3309,13 +3308,13 @@ cc_library( deps = [ ":call_graph", ":call_inliner", - ":hlo_pass", "//xla:literal", "//xla:shape_util", "//xla:status_macros", "//xla:types", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -3351,8 +3350,6 @@ cc_library( deps = [ ":hlo_cse", ":hlo_dce", - ":hlo_pass", - ":hlo_pass_pipeline", ":hlo_verifier", ":tuple_simplifier", "//xla:debug_options_flags", @@ -3362,6 +3359,8 @@ cc_library( "//xla:types", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:inlined_vector", @@ -3400,7 +3399,6 @@ cc_library( hdrs = ["convolution_group_converter.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", "//xla:literal", "//xla:literal_util", "//xla:shape_util", @@ -3409,6 +3407,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/strings", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", @@ -3437,7 +3436,6 @@ cc_library( hdrs = ["space_to_batch_converter.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", ":pattern_matcher", ":shape_inference", "//xla:debug_options_flags", @@ -3449,6 +3447,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/tsl/lib/core:bitmap", "@com_google_absl//absl/algorithm", "@com_google_absl//absl/algorithm:container", @@ -3487,7 +3486,6 @@ cc_library( ":call_graph", ":hlo_alias_analysis", ":hlo_dataflow_analysis", - ":hlo_pass", ":pattern_matcher", ":tuple_simplifier", ":while_loop_simplifier", @@ -3496,6 +3494,7 @@ cc_library( "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", "@com_google_absl//absl/status:statusor", @@ -3531,7 +3530,6 @@ cc_library( deps = [ ":algebraic_simplifier", ":hlo_creation_utils", - ":hlo_pass", ":pattern_matcher", ":tuple_util", ":while_loop_unroller", @@ -3539,6 +3537,7 @@ cc_library( "//xla:util", "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -3578,7 +3577,6 @@ cc_library( ":hlo_buffer", ":hlo_creation_utils", ":hlo_cse", - ":hlo_pass", ":hlo_value", ":pattern_matcher", ":tuple_simplifier", @@ -3593,6 +3591,7 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm", "@com_google_absl//absl/algorithm:container", @@ -3673,7 +3672,6 @@ cc_library( ":call_inliner", ":hlo_creation_utils", ":hlo_dce", - ":hlo_pass", ":pattern_matcher", ":while_loop_analysis", "//xla:comparison_util", @@ -3684,6 +3682,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -3727,10 +3726,10 @@ cc_library( srcs = ["while_loop_trip_count_annotator.cc"], hdrs = ["while_loop_trip_count_annotator.h"], deps = [ - ":hlo_pass", ":while_loop_analysis", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -3757,11 +3756,11 @@ cc_library( hdrs = ["defuser.h"], deps = [ ":call_graph", - ":hlo_pass", "//xla:status_macros", "//xla:types", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", @@ -3802,10 +3801,10 @@ cc_library( srcs = ["dot_decomposer.cc"], hdrs = ["dot_decomposer.h"], deps = [ - ":hlo_pass", ":shape_inference", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", @@ -3841,11 +3840,11 @@ cc_library( hdrs = ["dot_dimension_merger.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", @@ -3874,12 +3873,12 @@ cc_library( srcs = ["dot_merger.cc"], hdrs = ["dot_merger.h"], deps = [ - ":hlo_pass", ":shape_inference", "//xla:protobuf_util", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/service/graphcycles", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -3918,11 +3917,11 @@ cc_library( hdrs = ["convert_mover.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", "//xla:literal", "//xla:shape_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/container:inlined_vector", @@ -3970,11 +3969,11 @@ cc_library( deps = [ ":collective_decomposer_utils", ":collective_ops_utils", - ":hlo_pass", "//xla:literal_util", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", @@ -4005,9 +4004,9 @@ cc_library( srcs = ["tuple_simplifier.cc"], hdrs = ["tuple_simplifier.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -4039,11 +4038,11 @@ cc_library( hdrs = ["reshape_mover.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", "//xla:permutation_util", "//xla:shape_util", "//xla:status_macros", "//xla:util", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@tsl//tsl/platform:errors", ], @@ -4055,8 +4054,8 @@ cc_library( hdrs = ["reshape_decomposer.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/status", ], ) @@ -4067,8 +4066,8 @@ cc_library( hdrs = ["reduce_decomposer.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/status", ], ) @@ -4161,9 +4160,9 @@ cc_library( srcs = ["dynamic_dimension_simplifier.cc"], hdrs = ["dynamic_dimension_simplifier.h"], deps = [ - ":hlo_pass", "//xla:status_macros", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", ], ) @@ -4174,8 +4173,6 @@ xla_cc_test( ":dynamic_dimension_simplifier", ":hlo_creation_utils", ":hlo_parser", - ":hlo_pass", - ":hlo_pass_pipeline", ":pattern_matcher", ":pattern_matcher_gmock", ":shape_inference", @@ -4186,6 +4183,7 @@ xla_cc_test( "//xla:window_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", # fixdeps: keep "//xla/tsl/lib/core:status_test_util", @@ -4203,7 +4201,6 @@ cc_library( ":dynamic_window_utils", ":hlo_creation_utils", ":hlo_dce", - ":hlo_pass", ":pattern_matcher", ":shape_inference", ":tuple_util", @@ -4216,6 +4213,7 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/client:xla_builder", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/functional:function_ref", @@ -4307,12 +4305,12 @@ xla_cc_test( srcs = ["reshape_mover_test.cc"], deps = [ ":algebraic_simplifier", - ":hlo_pass", ":hlo_verifier", ":pattern_matcher", ":pattern_matcher_gmock", ":reshape_mover", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", @@ -5028,7 +5026,6 @@ cc_library( ":computation_layout", ":hlo_dce", ":hlo_graph_dumper", - ":hlo_pass", ":logical_buffer", ":tuple_points_to_analysis", ":tuple_simplifier", @@ -5040,6 +5037,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -5075,7 +5073,6 @@ cc_library( ":hlo_dce", ":hlo_graph_dumper", ":hlo_ordering", - ":hlo_pass", ":hlo_value", ":tuple_simplifier", "//xla:frontend_attributes", @@ -5085,6 +5082,7 @@ cc_library( "//xla:util", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_reachability", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -5110,11 +5108,11 @@ cc_library( ":hlo_alias_analysis", ":hlo_dataflow_analysis", ":hlo_graph_dumper", - ":hlo_pass", ":hlo_value", "//xla:shape_tree", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "//xla/service/graphcycles", "@com_google_absl//absl/algorithm:container", @@ -5180,9 +5178,9 @@ cc_library( hdrs = ["memory_space_propagation.h"], deps = [ ":hlo_dataflow_analysis", - ":hlo_pass", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", ], ) @@ -5203,10 +5201,10 @@ cc_library( srcs = ["hlo_dce.cc"], hdrs = ["hlo_dce.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log:check", @@ -5227,13 +5225,13 @@ cc_library( deps = [ ":hlo_dce", ":hlo_liveness_analysis", - ":hlo_pass", ":tuple_simplifier", ":while_loop_simplifier", "//xla:status_macros", "//xla:types", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@tsl//tsl/platform:errors", @@ -5248,7 +5246,6 @@ cc_library( deps = [ ":collective_ops_utils", ":hlo_module_config", - ":hlo_pass", ":shape_inference", "//xla:comparison_util", "//xla:permutation_util", @@ -5258,6 +5255,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -5342,13 +5340,13 @@ cc_library( ":hlo_cost_analysis", ":hlo_dataflow_analysis", ":hlo_dce", - ":hlo_pass", ":logical_buffer", ":tuple_points_to_analysis", "//xla:shape_util", "//xla:status_macros", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -5523,10 +5521,10 @@ cc_library( hdrs = ["hlo_cse.h"], deps = [ ":hlo_domain_map", - ":hlo_pass", "//xla:literal", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", @@ -5562,12 +5560,12 @@ cc_library( srcs = ["hlo_constant_folding.cc"], hdrs = ["hlo_constant_folding.h"], deps = [ - ":hlo_pass", ":slow_operation_alarm", "//xla:literal", "//xla:shape_util", "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", @@ -5626,9 +5624,9 @@ cc_library( deps = [ ":hlo_domain_map", ":hlo_graph_dumper", - ":hlo_pass", "//xla:types", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@tsl//tsl/platform:status", ], ) @@ -5639,8 +5637,8 @@ cc_library( hdrs = ["hlo_domain_isolator.h"], deps = [ ":hlo_domain_remover", - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@tsl//tsl/platform:statusor", ], ) @@ -5653,9 +5651,9 @@ cc_library( ":hlo_domain_map", ":hlo_domain_verifier", ":hlo_graph_dumper", - ":hlo_pass", "//xla:types", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@tsl//tsl/platform:status", ], ) @@ -5684,12 +5682,12 @@ cc_library( srcs = ["hlo_element_type_converter.cc"], hdrs = ["hlo_element_type_converter.h"], deps = [ - ":hlo_pass", "//xla:literal", "//xla:shape_util", "//xla:types", "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@tsl//tsl/platform:errors", ], @@ -5711,9 +5709,9 @@ cc_library( srcs = ["conditional_canonicalizer.cc"], hdrs = ["conditional_canonicalizer.h"], deps = [ - ":hlo_pass", "//xla:status_macros", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", ], ) @@ -5956,12 +5954,12 @@ cc_library( srcs = ["transpose_folding.cc"], hdrs = ["transpose_folding.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla:status_macros", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/types:span", "@tsl//tsl/platform:errors", @@ -5999,11 +5997,11 @@ cc_library( srcs = ["zero_sized_hlo_elimination.cc"], hdrs = ["zero_sized_hlo_elimination.h"], deps = [ - ":hlo_pass", "//xla:literal", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -6176,8 +6174,8 @@ cc_library( srcs = ["sort_simplifier.cc"], hdrs = ["sort_simplifier.h"], deps = [ - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", @@ -6271,9 +6269,9 @@ cc_library( srcs = ["root_instruction_sinker.cc"], hdrs = ["root_instruction_sinker.h"], deps = [ - ":hlo_pass", ":tuple_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", ], ) @@ -6306,7 +6304,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/service:hlo_pass", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", "@com_google_absl//absl/status:statusor", @@ -6337,10 +6335,10 @@ cc_library( srcs = ["host_memory_transfer_asyncifier.cc"], hdrs = ["host_memory_transfer_asyncifier.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", "@com_google_absl//absl/status", @@ -6379,12 +6377,12 @@ cc_library( deps = [ ":call_graph", ":hlo_alias_analysis", - ":hlo_pass", ":hlo_value", ":host_memory_offload_annotations_hdr", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/container:inlined_vector", @@ -6482,7 +6480,6 @@ cc_library( ":hlo_alias_analysis", ":hlo_buffer", ":hlo_cse", - ":hlo_pass", ":hlo_value", ":host_memory_offload_annotations_hdr", ":host_offload_utils", @@ -6493,6 +6490,7 @@ cc_library( "//xla:status_macros", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -6542,9 +6540,9 @@ cc_library( hdrs = ["host_offloading_prepare.h"], deps = [ ":call_graph", - ":hlo_pass", ":host_memory_offload_annotations_hdr", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", @@ -6624,12 +6622,12 @@ cc_library( deps = [ ":call_graph", ":collective_ops_utils", - ":hlo_pass", ":hlo_replication_analysis", "//xla:literal_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -6662,8 +6660,6 @@ cc_library( hdrs = ["while_loop_concat_code_motion.h"], deps = [ ":hlo_dce", - ":hlo_pass", - ":hlo_pass_pipeline", ":tuple_simplifier", ":while_loop_simplifier", "//xla:shape_util", @@ -6671,6 +6667,8 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -6708,12 +6706,12 @@ cc_library( ], deps = [ ":hlo_dce", - ":hlo_pass", ":while_loop_analysis", ":while_util", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -6752,12 +6750,12 @@ cc_library( srcs = ["while_loop_expensive_invariant_code_motion.cc"], hdrs = ["while_loop_expensive_invariant_code_motion.h"], deps = [ - ":hlo_pass", ":while_loop_analysis", ":while_util", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -6789,10 +6787,10 @@ cc_library( hdrs = ["fusion_constant_sinking.h"], deps = [ ":hlo_dce", - ":hlo_pass", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", @@ -6822,11 +6820,11 @@ cc_library( srcs = ["while_loop_constant_sinking.cc"], hdrs = ["while_loop_constant_sinking.h"], deps = [ - ":hlo_pass", ":while_util", "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status:statusor", @@ -6851,10 +6849,10 @@ cc_library( srcs = ["while_loop_fusible_sinking.cc"], hdrs = ["while_loop_fusible_sinking.h"], deps = [ - ":hlo_pass", ":while_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -6889,10 +6887,10 @@ cc_library( ":defuser", ":float_normalization", ":hlo_memory_scheduler", - ":hlo_pass", - ":hlo_pass_pipeline", ":sub_byte_normalization", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status:statusor", ], @@ -6914,13 +6912,13 @@ cc_library( srcs = ["indexed_array_analysis.cc"], hdrs = ["indexed_array_analysis.h"], deps = [ - ":hlo_pass", "//xla:literal", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -7048,8 +7046,8 @@ cc_library( srcs = ["map_inliner.cc"], hdrs = ["map_inliner.h"], deps = [ - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", @@ -7067,10 +7065,10 @@ cc_library( srcs = ["optimize_input_output_buffer_alias.cc"], hdrs = ["optimize_input_output_buffer_alias.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla:status_macros", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -7106,7 +7104,6 @@ cc_library( hdrs = ["ar_crs_combiner.h"], deps = [ ":call_graph", - ":hlo_pass", ":hlo_replication_analysis", ":pattern_matcher", "//xla:literal", @@ -7114,6 +7111,7 @@ cc_library( "//xla:shape_util", "//xla:status_macros", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -7146,9 +7144,9 @@ cc_library( srcs = ["dynamic_index_splitter.cc"], hdrs = ["dynamic_index_splitter.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/container:inlined_vector", @@ -7223,10 +7221,10 @@ cc_library( ":call_graph", ":call_inliner", ":hlo_creation_utils", - ":hlo_pass", "//xla:status_macros", "//xla:types", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:status", @@ -7252,8 +7250,8 @@ cc_library( srcs = ["slice_sinker.cc"], hdrs = ["slice_sinker.h"], deps = [ - ":hlo_pass", "//xla:shape_util", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/types:span", ], @@ -7452,8 +7450,8 @@ cc_library( hdrs = ["collective_transformation_reorderer.h"], deps = [ ":hlo_dce", - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log:check", @@ -7507,13 +7505,13 @@ cc_library( srcs = ["topk_rewriter.cc"], hdrs = ["topk_rewriter.h"], deps = [ - ":hlo_pass", ":pattern_matcher", "//xla:shape_util", "//xla:util", "//xla/client:xla_builder", "//xla/client/lib:comparators", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@tsl//tsl/platform:errors", @@ -7864,7 +7862,6 @@ cc_library( hdrs = ["layout_normalization.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", ":shape_inference", "//xla:literal", "//xla:permutation_util", @@ -7873,6 +7870,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", @@ -7890,8 +7888,8 @@ cc_library( srcs = ["instruction_hoister.cc"], hdrs = ["instruction_hoister.h"], deps = [ - ":hlo_pass", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log:check", @@ -7927,10 +7925,10 @@ xla_cc_test( srcs = ["scatter_simplifier_test.cc"], deps = [ ":hlo_parser", - ":hlo_pass", - ":hlo_pass_pipeline", ":scatter_simplifier", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", ], @@ -7983,7 +7981,7 @@ cc_library( copts = tsl_copts(), deps = [ ":hlo_creation_utils", - ":hlo_pass", + "//xla/hlo/pass:hlo_pass", "//xla/service/cpu:onednn_contraction_rewriter", ], ) @@ -8042,13 +8040,13 @@ cc_library( srcs = ["reduce_window_rewriter.cc"], hdrs = ["reduce_window_rewriter.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla:status_macros", "//xla:util", "//xla:window_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -8082,12 +8080,12 @@ cc_library( hdrs = ["stochastic_convert_decomposer.h"], deps = [ ":hlo_creation_utils", - ":hlo_pass", ":shape_inference", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@tsl//tsl/platform:errors", @@ -8124,10 +8122,10 @@ cc_library( srcs = ["sub_byte_normalization.cc"], hdrs = ["sub_byte_normalization.h"], deps = [ - ":hlo_pass", "//xla:shape_layout", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -8142,9 +8140,9 @@ cc_library( srcs = ["sharding_format_picker.cc"], hdrs = ["sharding_format_picker.h"], deps = [ - ":hlo_pass", "//xla/hlo/ir:hlo", "//xla/hlo/ir:tile_assignment", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", @@ -8477,9 +8475,9 @@ cc_library( srcs = ["add_original_value.cc"], hdrs = ["add_original_value.h"], deps = [ - ":hlo_pass", "//xla:shape_util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", diff --git a/xla/service/add_original_value.h b/xla/service/add_original_value.h index b4fb093fca2d9..8dd1655e0eb0e 100644 --- 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 index 185792b336cfa..9ff96b248d398 100644 --- 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 index 8028526987d82..1af8721066b8b 100644 --- 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 index 1bfb70371a68a..0759f8ebfbbc7 100644 --- 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 index 001e19b2e682b..79bf388322081 100644 --- 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 index 4c5579ea9ca41..cf2713243e3c6 100644 --- 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 index 8c88aabc48167..bd1aa811f9716 100644 --- 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 index 2f5ec2a27d619..102245cd2ee36 100644 --- 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 index 921160d59ce7d..77706bbff34d2 100644 --- 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 index 1c92d961f2dbc..f2ff998b4b6f0 100644 --- 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 index d54912b7644d1..1c44b945bdf69 100644 --- 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 index 3b7dfe809928b..ea3acd95a24df 100644 --- 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 index 9ca015fb23311..5a542cf1e48c5 100644 --- a/xla/service/async_collective_creator.h +++ b/xla/service/async_collective_creator.h @@ -20,7 +20,7 @@ limitations under the License. #include #include -#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 index 32c386fe11db4..6ba3cf13e69f2 100644 --- 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 index ab2c13f56bc2c..0ae50afe13eb3 100644 --- a/xla/service/batchnorm_expander.h +++ b/xla/service/batchnorm_expander.h @@ -19,7 +19,7 @@ limitations under the License. #include #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 index 9922dd601940d..e4bb01e9f486d 100644 --- 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 index 707738dd8491c..c8bc39a98c4f7 100644 --- 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 index b7d1fdabfac70..d998cad08b2e7 100644 --- 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 index 0206d187942d8..0cf3bc71c750e 100644 --- a/xla/service/broadcast_canonicalizer.h +++ b/xla/service/broadcast_canonicalizer.h @@ -18,7 +18,7 @@ limitations under the License. #include -#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 index 5698d0bc17d3d..7fd584ad5eeba 100644 --- 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 index 5aa5042535e87..43e929d6e8e7c 100644 --- 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 #include -#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 index f0d4c0c0df9ae..208053abe47f8 100644 --- 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 index 8df3de487027d..f01cce9cb77fb 100644 --- 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 index c7d404d73c91e..73efde7ce5309 100644 --- 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 index 642a8c0f238d8..cae79de4a62ce 100644 --- 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 index 76f509ed5be0a..1cb07b13c6a98 100644 --- 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 index 526ed64efe980..1f3ca55b9ec58 100644 --- 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 index efe9506be9766..a4ae0c05b2e19 100644 --- a/xla/service/conditional_canonicalizer.h +++ b/xla/service/conditional_canonicalizer.h @@ -18,7 +18,7 @@ limitations under the License. #include #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 index cd22c9d4ca0c0..00f22ef9cf870 100644 --- 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 index 6b1907917fe4e..4f36e501ffc98 100644 --- 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 index 19be425100458..01363f94580b6 100644 --- 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 index cbc9cff571a90..4e3676468fd8a 100644 --- 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 index 9c5dbd54e7c23..d574f06ce3b69 100644 --- 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 index 1ab841d3c04ca..dd8186d8aa765 100644 --- 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 index 5579afe2b294f..b531730ab95e0 100644 --- 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 index e41b4e711b8c0..c9ec5f40d3ec7 100644 --- 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 index 86bd5b5330444..b76d47cd1a871 100644 --- 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 index 8d7f4eaefa40f..d70552e1383a7 100644 --- a/xla/service/defuser.h +++ b/xla/service/defuser.h @@ -19,7 +19,7 @@ limitations under the License. #include #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 index 525cb4539db85..9bfe95d3f8715 100644 --- 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 index 4fbe3123f63c5..362f4958f30e7 100644 --- 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 index 53cee5d29b43a..a9e511e0c10f1 100644 --- 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 { diff --git a/xla/service/dot_merger.h b/xla/service/dot_merger.h index 5f7fa58eaf1e8..37081edd76c42 100644 --- a/xla/service/dot_merger.h +++ b/xla/service/dot_merger.h @@ -22,7 +22,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/dynamic_dimension_simplifier.h b/xla/service/dynamic_dimension_simplifier.h index 435dc579dca28..5fd7eedbf5364 100644 --- a/xla/service/dynamic_dimension_simplifier.h +++ b/xla/service/dynamic_dimension_simplifier.h @@ -18,7 +18,7 @@ limitations under the License. #include #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/dynamic_dimension_simplifier_test.cc b/xla/service/dynamic_dimension_simplifier_test.cc index 2131c6c002a3e..3b71fc4bf128a 100644 --- a/xla/service/dynamic_dimension_simplifier_test.cc +++ b/xla/service/dynamic_dimension_simplifier_test.cc @@ -25,12 +25,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/layout_util.h" #include "xla/literal.h" #include "xla/service/hlo_creation_utils.h" #include "xla/service/hlo_parser.h" -#include "xla/service/hlo_pass_fix.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/pattern_matcher.h" #include "xla/service/pattern_matcher_gmock.h" #include "xla/service/shape_inference.h" diff --git a/xla/service/dynamic_index_splitter.h b/xla/service/dynamic_index_splitter.h index 7ca615a9cf4f2..87bf0a95bc3fd 100644 --- a/xla/service/dynamic_index_splitter.h +++ b/xla/service/dynamic_index_splitter.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/dynamic_padder.h b/xla/service/dynamic_padder.h index 71430c31c2ec8..8ec029acee129 100644 --- a/xla/service/dynamic_padder.h +++ b/xla/service/dynamic_padder.h @@ -18,8 +18,8 @@ limitations under the License. #include +#include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/service/dynamic_dimension_inference.h" -#include "xla/service/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/flatten_call_graph.h b/xla/service/flatten_call_graph.h index abde09db57e90..a3fe5aada4f1c 100644 --- a/xla/service/flatten_call_graph.h +++ b/xla/service/flatten_call_graph.h @@ -19,7 +19,7 @@ limitations under the License. #define XLA_SERVICE_FLATTEN_CALL_GRAPH_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/float_normalization.h b/xla/service/float_normalization.h index be0e92ea5ee56..1862edcb14d8a 100644 --- a/xla/service/float_normalization.h +++ b/xla/service/float_normalization.h @@ -19,8 +19,8 @@ limitations under the License. #include #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/fusion_constant_sinking.h b/xla/service/fusion_constant_sinking.h index 636c785696477..96e3fb95e7adc 100644 --- a/xla/service/fusion_constant_sinking.h +++ b/xla/service/fusion_constant_sinking.h @@ -17,7 +17,7 @@ limitations under the License. #define XLA_SERVICE_FUSION_CONSTANT_SINKING_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/hlo_computation_deduplicator.h b/xla/service/hlo_computation_deduplicator.h index 5b25adbcb531d..64eac6de00452 100644 --- a/xla/service/hlo_computation_deduplicator.h +++ b/xla/service/hlo_computation_deduplicator.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_computation.h" -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/hlo_constant_folding.h b/xla/service/hlo_constant_folding.h index 34aa8199787f9..4f56d7fa35562 100644 --- a/xla/service/hlo_constant_folding.h +++ b/xla/service/hlo_constant_folding.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/hlo_cse.h b/xla/service/hlo_cse.h index 1ccab0d5872eb..8dce3940eda94 100644 --- a/xla/service/hlo_cse.h +++ b/xla/service/hlo_cse.h @@ -17,7 +17,7 @@ limitations under the License. #define XLA_SERVICE_HLO_CSE_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/hlo_dce.h b/xla/service/hlo_dce.h index b6d1d9214037a..acd344bb025bc 100644 --- a/xla/service/hlo_dce.h +++ b/xla/service/hlo_dce.h @@ -24,7 +24,7 @@ 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/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/hlo_domain_isolator.h b/xla/service/hlo_domain_isolator.h index db9385847f034..d10dc501babc3 100644 --- a/xla/service/hlo_domain_isolator.h +++ b/xla/service/hlo_domain_isolator.h @@ -18,7 +18,7 @@ limitations under the License. #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/hlo_domain_remover.h b/xla/service/hlo_domain_remover.h index 6510ab684f9c3..970f4f164a5e2 100644 --- a/xla/service/hlo_domain_remover.h +++ b/xla/service/hlo_domain_remover.h @@ -19,7 +19,7 @@ limitations under the License. #include "xla/hlo/ir/hlo_domain_metadata.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" #include "tsl/platform/status.h" namespace xla { diff --git a/xla/service/hlo_domain_verifier.h b/xla/service/hlo_domain_verifier.h index 15ab7e3894bef..32dce237cb03d 100644 --- a/xla/service/hlo_domain_verifier.h +++ b/xla/service/hlo_domain_verifier.h @@ -21,8 +21,8 @@ limitations under the License. #include "xla/hlo/ir/hlo_domain_metadata.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 "tsl/platform/status.h" namespace xla { diff --git a/xla/service/hlo_element_type_converter.h b/xla/service/hlo_element_type_converter.h index 6f1eced4bf3d9..e78edec9d27a2 100644 --- a/xla/service/hlo_element_type_converter.h +++ b/xla/service/hlo_element_type_converter.h @@ -17,7 +17,7 @@ limitations under the License. #define XLA_SERVICE_HLO_ELEMENT_TYPE_CONVERTER_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/hlo_memory_scheduler.h b/xla/service/hlo_memory_scheduler.h index 2fb211ac6531a..fd9ae679110af 100644 --- a/xla/service/hlo_memory_scheduler.h +++ b/xla/service/hlo_memory_scheduler.h @@ -25,8 +25,8 @@ limitations under the License. #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_schedule.h" +#include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/service/hlo_alias_analysis.h" -#include "xla/service/hlo_pass_interface.h" #include "xla/service/logical_buffer.h" #include "xla/service/tuple_points_to_analysis.h" diff --git a/xla/service/hlo_module_dce.h b/xla/service/hlo_module_dce.h index 228e8e5a611ee..453deb26bb847 100644 --- a/xla/service/hlo_module_dce.h +++ b/xla/service/hlo_module_dce.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/hlo_rematerialization.h b/xla/service/hlo_rematerialization.h index 2501ed6f0911c..4eba8f1a2bdc8 100644 --- a/xla/service/hlo_rematerialization.h +++ b/xla/service/hlo_rematerialization.h @@ -26,9 +26,9 @@ #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_schedule.h" +#include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/service/call_graph.h" #include "xla/service/hlo_cost_analysis.h" -#include "xla/service/hlo_pass_interface.h" #include "xla/service/tuple_points_to_analysis.h" #include "xla/shape.h" diff --git a/xla/service/hlo_unstacker.h b/xla/service/hlo_unstacker.h index 222a1e511e6d4..bd312e89a2a66 100644 --- a/xla/service/hlo_unstacker.h +++ b/xla/service/hlo_unstacker.h @@ -24,7 +24,7 @@ limitations under the License. #include "absl/status/statusor.h" #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_instruction.h" -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { // This pass implements unstacking for loop operands. Generally speaking, diff --git a/xla/service/hlo_verifier.h b/xla/service/hlo_verifier.h index 2343ad5fcaad7..83f5e32a6def0 100644 --- a/xla/service/hlo_verifier.h +++ b/xla/service/hlo_verifier.h @@ -23,7 +23,7 @@ limitations under the License. #include "absl/strings/string_view.h" #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h" -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/host_memory_transfer_asyncifier.h b/xla/service/host_memory_transfer_asyncifier.h index 8d4f27cf04da4..0f42d8f1cfa01 100644 --- a/xla/service/host_memory_transfer_asyncifier.h +++ b/xla/service/host_memory_transfer_asyncifier.h @@ -19,7 +19,7 @@ limitations under the License. #include "absl/status/statusor.h" #include "absl/strings/string_view.h" -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/host_offload_legalize.h b/xla/service/host_offload_legalize.h index 5eca4a0f54174..c04a9cd549d26 100644 --- a/xla/service/host_offload_legalize.h +++ b/xla/service/host_offload_legalize.h @@ -20,8 +20,8 @@ #include "absl/container/flat_hash_set.h" #include "absl/strings/string_view.h" +#include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/service/hlo_alias_analysis.h" -#include "xla/service/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/host_offloader.h b/xla/service/host_offloader.h index 39cc8de01b841..bcc761ea2fafa 100644 --- a/xla/service/host_offloader.h +++ b/xla/service/host_offloader.h @@ -23,9 +23,9 @@ #include "absl/status/statusor.h" #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_instruction.h" +#include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/service/hlo_alias_analysis.h" #include "xla/service/hlo_buffer.h" -#include "xla/service/hlo_pass_interface.h" #include "xla/service/host_offload_utils.h" namespace xla { diff --git a/xla/service/host_offloading_prepare.h b/xla/service/host_offloading_prepare.h index bf5f7d5d85238..cb0bfe04078f1 100644 --- a/xla/service/host_offloading_prepare.h +++ b/xla/service/host_offloading_prepare.h @@ -24,7 +24,7 @@ #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/indexed_array_analysis.h b/xla/service/indexed_array_analysis.h index 634c24f206839..1f0f451b1cf07 100644 --- a/xla/service/indexed_array_analysis.h +++ b/xla/service/indexed_array_analysis.h @@ -28,8 +28,8 @@ limitations under the License. #include "xla/hlo/ir/hlo_instruction.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/xla_data.pb.h" #include "tsl/platform/statusor.h" diff --git a/xla/service/instruction_fusion.h b/xla/service/instruction_fusion.h index 70b230df94098..c4952aea15c2a 100644 --- a/xla/service/instruction_fusion.h +++ b/xla/service/instruction_fusion.h @@ -37,8 +37,8 @@ limitations under the License. #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_reachability.h" +#include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/service/fusion_queue.h" -#include "xla/service/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/instruction_hoister.h b/xla/service/instruction_hoister.h index e3db5f37d9e87..a52598e052037 100644 --- a/xla/service/instruction_hoister.h +++ b/xla/service/instruction_hoister.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/latency_hiding_scheduler.h b/xla/service/latency_hiding_scheduler.h index 5c83108d7d984..57cd09362893f 100644 --- a/xla/service/latency_hiding_scheduler.h +++ b/xla/service/latency_hiding_scheduler.h @@ -38,11 +38,11 @@ limitations under the License. #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" #include "xla/hlo/ir/hlo_schedule.h" +#include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/map_util.h" #include "xla/service/hlo_alias_analysis.h" #include "xla/service/hlo_buffer.h" #include "xla/service/hlo_cost_analysis.h" -#include "xla/service/hlo_pass_interface.h" #include "xla/service/hlo_value.h" #include "xla/shape_util.h" #include "xla/xla.pb.h" diff --git a/xla/service/layout_assignment.h b/xla/service/layout_assignment.h index ba59743018c38..6d29b7cd6f53d 100644 --- a/xla/service/layout_assignment.h +++ b/xla/service/layout_assignment.h @@ -37,12 +37,12 @@ limitations under the License. #include "xla/hlo/ir/hlo_instruction.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/layout.h" #include "xla/layout_util.h" #include "xla/map_util.h" #include "xla/service/call_graph.h" #include "xla/service/computation_layout.h" -#include "xla/service/hlo_pass_interface.h" #include "xla/service/logical_buffer.h" #include "xla/service/tuple_points_to_analysis.h" #include "xla/shape.h" diff --git a/xla/service/layout_normalization.h b/xla/service/layout_normalization.h index b847e41e0e94c..da416aa9687f4 100644 --- a/xla/service/layout_normalization.h +++ b/xla/service/layout_normalization.h @@ -26,7 +26,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/logistic_expander.h b/xla/service/logistic_expander.h index 98cbe99126566..8e9aeec2e6795 100644 --- a/xla/service/logistic_expander.h +++ b/xla/service/logistic_expander.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" #include "xla/service/op_expander_pass.h" namespace xla { diff --git a/xla/service/loop_schedule_linearizer.h b/xla/service/loop_schedule_linearizer.h index 0c74c1993e475..3286a2e41a004 100644 --- a/xla/service/loop_schedule_linearizer.h +++ b/xla/service/loop_schedule_linearizer.h @@ -22,9 +22,9 @@ 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/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/map_inliner.h b/xla/service/map_inliner.h index e9c8e414c7205..33821208770ae 100644 --- a/xla/service/map_inliner.h +++ b/xla/service/map_inliner.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/memory_space_propagation.h b/xla/service/memory_space_propagation.h index 4de4b2e47ca1a..8f741f6430cf5 100644 --- a/xla/service/memory_space_propagation.h +++ b/xla/service/memory_space_propagation.h @@ -17,8 +17,8 @@ limitations under the License. #define XLA_SERVICE_MEMORY_SPACE_PROPAGATION_H_ #include "xla/hlo/ir/hlo_module.h" +#include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/service/hlo_dataflow_analysis.h" -#include "xla/service/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/multi_output_fusion.h b/xla/service/multi_output_fusion.h index e57a8e3ca255c..14233ae4b8b9e 100644 --- a/xla/service/multi_output_fusion.h +++ b/xla/service/multi_output_fusion.h @@ -25,7 +25,7 @@ limitations under the License. #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_reachability.h" -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/op_expander_pass.h b/xla/service/op_expander_pass.h index c86c3a44f5563..e8644f5abb30a 100644 --- a/xla/service/op_expander_pass.h +++ b/xla/service/op_expander_pass.h @@ -16,7 +16,7 @@ limitations under the License. #ifndef XLA_SERVICE_OP_EXPANDER_PASS_H_ #define XLA_SERVICE_OP_EXPANDER_PASS_H_ -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/optimize_input_output_buffer_alias.h b/xla/service/optimize_input_output_buffer_alias.h index 2b8dcd621aba5..d03128c8b9138 100644 --- a/xla/service/optimize_input_output_buffer_alias.h +++ b/xla/service/optimize_input_output_buffer_alias.h @@ -25,7 +25,7 @@ limitations under the License. #include "absl/types/span.h" #include "xla/hlo/ir/hlo_input_output_alias_config.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/shape.h" #include "xla/shape_util.h" diff --git a/xla/service/p2p_schedule_preparation.h b/xla/service/p2p_schedule_preparation.h index d28a73f2ab3b4..1545b3b51ddf7 100644 --- a/xla/service/p2p_schedule_preparation.h +++ b/xla/service/p2p_schedule_preparation.h @@ -17,7 +17,7 @@ limitations under the License. #define XLA_SERVICE_P2P_SCHEDULE_PREPARATION_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/reduce_decomposer.h b/xla/service/reduce_decomposer.h index 0907527cc7244..d112002fb5678 100644 --- a/xla/service/reduce_decomposer.h +++ b/xla/service/reduce_decomposer.h @@ -18,7 +18,7 @@ limitations under the License. #include -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/reduce_scatter_combiner.h b/xla/service/reduce_scatter_combiner.h index 79e12799b94e3..26e047bd400f9 100644 --- a/xla/service/reduce_scatter_combiner.h +++ b/xla/service/reduce_scatter_combiner.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/reduce_scatter_decomposer.h b/xla/service/reduce_scatter_decomposer.h index 114ce32566529..ea52de241020b 100644 --- a/xla/service/reduce_scatter_decomposer.h +++ b/xla/service/reduce_scatter_decomposer.h @@ -20,7 +20,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/reduce_scatter_reassociate.h b/xla/service/reduce_scatter_reassociate.h index 9bb8e563439f8..fb7a0eccfce04 100644 --- a/xla/service/reduce_scatter_reassociate.h +++ b/xla/service/reduce_scatter_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/reduce_window_rewriter.h b/xla/service/reduce_window_rewriter.h index 86a96f4a1cc3b..616ec8b701e9a 100644 --- a/xla/service/reduce_window_rewriter.h +++ b/xla/service/reduce_window_rewriter.h @@ -24,7 +24,7 @@ limitations under the License. #include "absl/strings/string_view.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/reshape_decomposer.h b/xla/service/reshape_decomposer.h index 2d9cb5fee68b5..9c10649653cb3 100644 --- a/xla/service/reshape_decomposer.h +++ b/xla/service/reshape_decomposer.h @@ -16,7 +16,7 @@ limitations under the License. #ifndef XLA_SERVICE_RESHAPE_DECOMPOSER_H_ #define XLA_SERVICE_RESHAPE_DECOMPOSER_H_ -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/reshape_mover.h b/xla/service/reshape_mover.h index da0e3a55134bc..14116a0ba0cf1 100644 --- a/xla/service/reshape_mover.h +++ b/xla/service/reshape_mover.h @@ -16,7 +16,7 @@ limitations under the License. #ifndef XLA_SERVICE_RESHAPE_MOVER_H_ #define XLA_SERVICE_RESHAPE_MOVER_H_ -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/reshape_mover_test.cc b/xla/service/reshape_mover_test.cc index 5ad138e1a9430..6aca742e0190e 100644 --- a/xla/service/reshape_mover_test.cc +++ b/xla/service/reshape_mover_test.cc @@ -21,8 +21,8 @@ limitations under the License. #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" +#include "xla/hlo/pass/hlo_pass_fix.h" #include "xla/service/algebraic_simplifier.h" -#include "xla/service/hlo_pass_fix.h" #include "xla/service/hlo_verifier.h" #include "xla/service/pattern_matcher.h" #include "xla/service/pattern_matcher_gmock.h" diff --git a/xla/service/root_instruction_sinker.h b/xla/service/root_instruction_sinker.h index 81672c59539a2..50975bd0a876e 100644 --- a/xla/service/root_instruction_sinker.h +++ b/xla/service/root_instruction_sinker.h @@ -17,7 +17,7 @@ limitations under the License. #define XLA_SERVICE_ROOT_INSTRUCTION_SINKER_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/scan_loop_accumulator_input_unification.h b/xla/service/scan_loop_accumulator_input_unification.h index 34f828167bcc7..e105dddb194cb 100644 --- a/xla/service/scan_loop_accumulator_input_unification.h +++ b/xla/service/scan_loop_accumulator_input_unification.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 { // This pass looks at the nested loops with accumulator patterns and unifies the diff --git a/xla/service/scatter_simplifier_test.cc b/xla/service/scatter_simplifier_test.cc index 2e1d6c8f3dc10..919a68620fee4 100644 --- a/xla/service/scatter_simplifier_test.cc +++ b/xla/service/scatter_simplifier_test.cc @@ -18,9 +18,9 @@ limitations under the License. #include #include "xla/hlo/ir/hlo_casting_utils.h" +#include "xla/hlo/pass/hlo_pass_fix.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/hlo_parser.h" -#include "xla/service/hlo_pass_fix.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/tests/hlo_test_base.h" namespace xla { diff --git a/xla/service/sharding_format_picker.h b/xla/service/sharding_format_picker.h index 1ebd96520c97b..2f8f47a0e48bf 100644 --- a/xla/service/sharding_format_picker.h +++ b/xla/service/sharding_format_picker.h @@ -16,7 +16,7 @@ limitations under the License. #ifndef XLA_SERVICE_SHARDING_FORMAT_PICKER_H_ #define XLA_SERVICE_SHARDING_FORMAT_PICKER_H_ -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/sharding_propagation.h b/xla/service/sharding_propagation.h index 22cb7af042545..eb44d875cca9b 100644 --- a/xla/service/sharding_propagation.h +++ b/xla/service/sharding_propagation.h @@ -25,10 +25,10 @@ limitations under the License. #include "absl/status/statusor.h" #include "absl/types/span.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/custom_call_sharding_helper.h" #include "xla/service/dot_as_convolution_util.h" -#include "xla/service/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/sharding_remover.h b/xla/service/sharding_remover.h index c6a50c9b2512d..3c88ed1e5c0ad 100644 --- a/xla/service/sharding_remover.h +++ b/xla/service/sharding_remover.h @@ -21,7 +21,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/simplify_fp_conversions.h b/xla/service/simplify_fp_conversions.h index 7bf64d7924ac9..2c92319c96476 100644 --- a/xla/service/simplify_fp_conversions.h +++ b/xla/service/simplify_fp_conversions.h @@ -19,7 +19,7 @@ limitations under the License. #include "absl/container/flat_hash_set.h" #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/slice_sinker.h b/xla/service/slice_sinker.h index 61805ca874211..eab42adf5171a 100644 --- a/xla/service/slice_sinker.h +++ b/xla/service/slice_sinker.h @@ -16,7 +16,7 @@ limitations under the License. #ifndef XLA_SERVICE_SLICE_SINKER_H_ #define XLA_SERVICE_SLICE_SINKER_H_ -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/sort_simplifier.h b/xla/service/sort_simplifier.h index dd02f6a585344..bd4d38c855927 100644 --- a/xla/service/sort_simplifier.h +++ b/xla/service/sort_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/space_to_batch_converter.h b/xla/service/space_to_batch_converter.h index 068d6d475e811..47141bb1a6d4c 100644 --- a/xla/service/space_to_batch_converter.h +++ b/xla/service/space_to_batch_converter.h @@ -19,7 +19,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/stochastic_convert_decomposer.h b/xla/service/stochastic_convert_decomposer.h index a51421b438b28..ad53237257c81 100644 --- a/xla/service/stochastic_convert_decomposer.h +++ b/xla/service/stochastic_convert_decomposer.h @@ -17,7 +17,7 @@ limitations under the License. #define XLA_SERVICE_STOCHASTIC_CONVERT_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/sub_byte_normalization.h b/xla/service/sub_byte_normalization.h index 48c3ecd83afa5..69d503944f713 100644 --- a/xla/service/sub_byte_normalization.h +++ b/xla/service/sub_byte_normalization.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/topk_rewriter.h b/xla/service/topk_rewriter.h index 8400c505053a6..cf673fe702b19 100644 --- a/xla/service/topk_rewriter.h +++ b/xla/service/topk_rewriter.h @@ -25,7 +25,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/transpose_folding.h b/xla/service/transpose_folding.h index eacf99ae3ced4..447b96771318b 100644 --- a/xla/service/transpose_folding.h +++ b/xla/service/transpose_folding.h @@ -19,7 +19,7 @@ limitations under the License. #include #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/tree_reduction_rewriter.h b/xla/service/tree_reduction_rewriter.h index 6d0f0da529dd7..9563c769d8ee5 100644 --- a/xla/service/tree_reduction_rewriter.h +++ b/xla/service/tree_reduction_rewriter.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/tuple_simplifier.h b/xla/service/tuple_simplifier.h index 5fa2ad05d229c..81c315d25ec47 100644 --- a/xla/service/tuple_simplifier.h +++ b/xla/service/tuple_simplifier.h @@ -21,7 +21,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/while_loop_all_reduce_code_motion.h b/xla/service/while_loop_all_reduce_code_motion.h index 46a8e945c1ca0..730a06c9e662e 100644 --- a/xla/service/while_loop_all_reduce_code_motion.h +++ b/xla/service/while_loop_all_reduce_code_motion.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/while_loop_concat_code_motion.cc b/xla/service/while_loop_concat_code_motion.cc index 6432c841cd59c..f63c7afd9e96b 100644 --- a/xla/service/while_loop_concat_code_motion.cc +++ b/xla/service/while_loop_concat_code_motion.cc @@ -31,8 +31,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_pipeline.h" #include "xla/service/hlo_dce.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/tuple_simplifier.h" #include "xla/service/while_loop_simplifier.h" #include "xla/shape_util.h" diff --git a/xla/service/while_loop_concat_code_motion.h b/xla/service/while_loop_concat_code_motion.h index bf762c48eae21..99f82c7e27132 100644 --- a/xla/service/while_loop_concat_code_motion.h +++ b/xla/service/while_loop_concat_code_motion.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/while_loop_constant_sinking.h b/xla/service/while_loop_constant_sinking.h index 660760ef3ca37..8d1402ff72d29 100644 --- a/xla/service/while_loop_constant_sinking.h +++ b/xla/service/while_loop_constant_sinking.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/while_loop_expensive_invariant_code_motion.h b/xla/service/while_loop_expensive_invariant_code_motion.h index e285e981b7374..06b4bac1908fa 100644 --- a/xla/service/while_loop_expensive_invariant_code_motion.h +++ b/xla/service/while_loop_expensive_invariant_code_motion.h @@ -24,7 +24,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" #include "xla/shape.h" #include "xla/shape_util.h" #include "xla/util.h" diff --git a/xla/service/while_loop_fusible_sinking.h b/xla/service/while_loop_fusible_sinking.h index 9311c8195eec7..e1b38bd7c4153 100644 --- a/xla/service/while_loop_fusible_sinking.h +++ b/xla/service/while_loop_fusible_sinking.h @@ -23,7 +23,7 @@ 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/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/while_loop_invariant_code_motion.h b/xla/service/while_loop_invariant_code_motion.h index 7224df7853056..695f08d0a706a 100644 --- a/xla/service/while_loop_invariant_code_motion.h +++ b/xla/service/while_loop_invariant_code_motion.h @@ -21,8 +21,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/compile_time_cap.h" -#include "xla/service/hlo_pass_interface.h" #include "xla/shape.h" #include "xla/shape_util.h" diff --git a/xla/service/while_loop_simplifier.h b/xla/service/while_loop_simplifier.h index 851fdb0c39667..7fda6d93f201c 100644 --- a/xla/service/while_loop_simplifier.h +++ b/xla/service/while_loop_simplifier.h @@ -21,7 +21,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/while_loop_trip_count_annotator.h b/xla/service/while_loop_trip_count_annotator.h index f71cf47d2be62..d4185e558f958 100644 --- a/xla/service/while_loop_trip_count_annotator.h +++ b/xla/service/while_loop_trip_count_annotator.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/while_loop_unroller.cc b/xla/service/while_loop_unroller.cc index b8f2bcae972ab..3bb2f8d6c8300 100644 --- a/xla/service/while_loop_unroller.cc +++ b/xla/service/while_loop_unroller.cc @@ -37,6 +37,7 @@ limitations under the License. #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_opcode.h" +#include "xla/hlo/pass/hlo_pass_fix.h" #include "xla/hlo/utils/hlo_query.h" #include "xla/literal.h" #include "xla/literal_util.h" @@ -46,7 +47,6 @@ limitations under the License. #include "xla/service/flatten_call_graph.h" #include "xla/service/hlo_creation_utils.h" #include "xla/service/hlo_cse.h" -#include "xla/service/hlo_pass_fix.h" #include "xla/service/pattern_matcher.h" #include "xla/service/tuple_simplifier.h" #include "xla/service/while_loop_analysis.h" diff --git a/xla/service/while_loop_unroller.h b/xla/service/while_loop_unroller.h index cb513320a742b..def7d9d9a33e1 100644 --- a/xla/service/while_loop_unroller.h +++ b/xla/service/while_loop_unroller.h @@ -27,7 +27,7 @@ limitations under the License. #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_opcode.h" -#include "xla/service/hlo_pass_interface.h" +#include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/xla/service/zero_sized_hlo_elimination.h b/xla/service/zero_sized_hlo_elimination.h index d8d6c467be36c..de7488b959ae8 100644 --- a/xla/service/zero_sized_hlo_elimination.h +++ b/xla/service/zero_sized_hlo_elimination.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" // HLO pass that replaces zero sized Hlos with a zero sized constant literal. namespace xla {