From 6d8bdd091994eac3ce4c3d710205a9e16a0f38ee Mon Sep 17 00:00:00 2001 From: Sandeep Dasgupta Date: Wed, 18 Sep 2024 01:44:10 -0700 Subject: [PATCH] [HLO Componentization] Create pass sub-component PiperOrigin-RevId: 675892299 --- xla/service/gpu/BUILD | 30 +++++++++---------- xla/service/gpu/amdgpu_compiler.cc | 4 +-- xla/service/gpu/amdgpu_compiler.h | 2 +- xla/service/gpu/fusion_pipeline.cc | 4 +-- xla/service/gpu/fusion_pipeline.h | 2 +- xla/service/gpu/gpu_compiler.cc | 4 +-- xla/service/gpu/gpu_compiler.h | 2 +- xla/service/gpu/gpu_hlo_schedule.cc | 2 +- xla/service/gpu/gpu_p2p_pipeliner.cc | 2 +- xla/service/gpu/gpu_p2p_pipeliner.h | 2 +- xla/service/gpu/gpu_p2p_pipeliner_test.cc | 2 +- xla/service/gpu/gpu_spmd_pipeline.cc | 4 +-- xla/service/gpu/gpu_spmd_pipeline.h | 2 +- xla/service/gpu/gpu_spmd_pipeline_test.cc | 2 +- xla/service/gpu/nvptx_compiler.cc | 4 +-- xla/service/gpu/nvptx_compiler.h | 2 +- .../prepare_hlo_for_ir_emitting_pipeline.cc | 2 +- .../prepare_hlo_for_ir_emitting_pipeline.h | 2 +- 18 files changed, 37 insertions(+), 37 deletions(-) diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD index 3fc11e3bb7296..454647793872e 100644 --- a/xla/service/gpu/BUILD +++ b/xla/service/gpu/BUILD @@ -1305,12 +1305,12 @@ cc_library( hdrs = ["fusion_pipeline.h"], deps = [ "//xla:xla_proto_cc", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/service:cpu_gpu_shape_verifier", "//xla/service:hlo_cost_analysis", "//xla/service:hlo_cse", "//xla/service:hlo_dce", - "//xla/service:hlo_pass", - "//xla/service:hlo_pass_pipeline", "//xla/service:hlo_verifier", "//xla/service:layout_assignment", "//xla/service/gpu/model:gpu_hlo_cost_analysis", @@ -1333,11 +1333,11 @@ cc_library( deps = [ "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/service:copy_insertion", "//xla/service:cpu_gpu_shape_verifier", "//xla/service:hlo_dataflow_analysis", "//xla/service:hlo_dce", - "//xla/service:hlo_pass_pipeline", "//xla/service:hlo_verifier", "//xla/service:layout_assignment", "//xla/service:loop_schedule_linearizer", @@ -1407,6 +1407,8 @@ cc_library( "@llvm-project//mlir:Support", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_module_group", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/pjrt/distributed:key_value_store_interface", "//xla/service/gpu/autotuning:autotuner_util", "//xla/service/gpu/autotuning:custom_kernel_fusion_autotuner", @@ -1515,8 +1517,6 @@ cc_library( "//xla/service:hlo_dce", "//xla/service:hlo_module_config", "//xla/service:hlo_ordering", - "//xla/service:hlo_pass", - "//xla/service:hlo_pass_pipeline", "//xla/service:hlo_proto_cc", "//xla/service:hlo_rematerialization", "//xla/service:hlo_verifier", @@ -1735,6 +1735,8 @@ cc_library( "//xla:xla_data_proto_cc", "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/pjrt/distributed:key_value_store_interface", "//xla/service:algebraic_simplifier", "//xla/service:call_inliner", @@ -1748,8 +1750,6 @@ cc_library( "//xla/service:hlo_dataflow_analysis", "//xla/service:hlo_dce", "//xla/service:hlo_module_config", - "//xla/service:hlo_pass", - "//xla/service:hlo_pass_pipeline", "//xla/service:hlo_verifier", "//xla/service:reshape_mover", "//xla/service:tuple_simplifier", @@ -1981,6 +1981,8 @@ cc_library( "//xla:util", "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/service:algebraic_simplifier", "//xla/service:call_inliner", "//xla/service:convert_mover", @@ -1989,8 +1991,6 @@ cc_library( "//xla/service:float_support", "//xla/service:hlo_constant_folding", "//xla/service:hlo_module_config", - "//xla/service:hlo_pass", - "//xla/service:hlo_pass_pipeline", "//xla/service:hlo_verifier", "//xla/service:reshape_mover", "//xla/service:tuple_simplifier", @@ -2078,11 +2078,11 @@ cc_library( "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/hlo/utils:hlo_query", "//xla/service:buffer_value", "//xla/service:collective_ops_utils", "//xla/service:hlo_memory_scheduler", - "//xla/service:hlo_pass_pipeline", "//xla/service:latency_hiding_scheduler", "//xla/service:p2p_schedule_preparation", "//xla/service:profile_guided_latency_estimator", @@ -2149,10 +2149,10 @@ cc_library( deps = [ "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/service:collective_ops_utils", "//xla/service:collective_pipeliner", "//xla/service:hlo_parser", - "//xla/service:hlo_pass_pipeline", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", @@ -2168,9 +2168,9 @@ xla_cc_test( ":gpu_p2p_pipeliner", "//xla:util", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/service:hlo_module_config", "//xla/service:hlo_parser", - "//xla/service:hlo_pass_pipeline", "//xla/service:hlo_verifier", "//xla/tests:filecheck", "//xla/tests:hlo_test_base", @@ -2188,6 +2188,8 @@ cc_library( deps = [ ":runtime_intrinsics", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/hlo/transforms:hlo_constant_splitter", "//xla/service:algebraic_simplifier", "//xla/service:conditional_simplifier", @@ -2195,8 +2197,6 @@ cc_library( "//xla/service:hlo_constant_folding", "//xla/service:hlo_dce", "//xla/service:hlo_module_config", - "//xla/service:hlo_pass", - "//xla/service:hlo_pass_pipeline", "//xla/service:reshape_mover", "//xla/service:scatter_expander", "//xla/service:sharding_propagation", @@ -2225,10 +2225,10 @@ xla_cc_test( "//xla:util", "//xla/client:executable_build_options", "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass_pipeline", "//xla/service:algebraic_simplifier", "//xla/service:hlo_module_config", "//xla/service:hlo_parser", - "//xla/service:hlo_pass_pipeline", "//xla/stream_executor:device_description", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", diff --git a/xla/service/gpu/amdgpu_compiler.cc b/xla/service/gpu/amdgpu_compiler.cc index 7579d98d4c8e1..86650111c796c 100644 --- a/xla/service/gpu/amdgpu_compiler.cc +++ b/xla/service/gpu/amdgpu_compiler.cc @@ -26,6 +26,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_fix.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/algebraic_simplifier.h" #include "xla/service/call_inliner.h" #include "xla/service/convert_mover.h" @@ -49,8 +51,6 @@ limitations under the License. #include "xla/service/gpu/transforms/triangular_solve_rewriter.h" #include "xla/service/hlo_constant_folding.h" #include "xla/service/hlo_module_config.h" -#include "xla/service/hlo_pass_fix.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/hlo_verifier.h" #include "xla/service/reshape_mover.h" #include "xla/service/tuple_simplifier.h" diff --git a/xla/service/gpu/amdgpu_compiler.h b/xla/service/gpu/amdgpu_compiler.h index 2638ee373c9a9..2ca2b1f567500 100644 --- a/xla/service/gpu/amdgpu_compiler.h +++ b/xla/service/gpu/amdgpu_compiler.h @@ -20,10 +20,10 @@ limitations under the License. #include "absl/status/statusor.h" #include "llvm/IR/Module.h" #include "xla/hlo/ir/hlo_module.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/gpu/autotuning/autotuner_util.h" #include "xla/service/gpu/gpu_compiler.h" #include "xla/service/hlo_module_config.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/stream_executor/device_description.h" #include "xla/stream_executor/device_memory_allocator.h" #include "xla/stream_executor/dnn.h" diff --git a/xla/service/gpu/fusion_pipeline.cc b/xla/service/gpu/fusion_pipeline.cc index 20e58ad4b96f8..7bc4f170980ea 100644 --- a/xla/service/gpu/fusion_pipeline.cc +++ b/xla/service/gpu/fusion_pipeline.cc @@ -18,6 +18,8 @@ limitations under the License. #include #include +#include "xla/hlo/pass/hlo_pass_fix.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/cpu_gpu_shape_verifier.h" #include "xla/service/gpu/model/gpu_hlo_cost_analysis.h" #include "xla/service/gpu/transforms/fusion_merger.h" @@ -30,8 +32,6 @@ limitations under the License. #include "xla/service/hlo_cost_analysis.h" #include "xla/service/hlo_cse.h" #include "xla/service/hlo_dce.h" -#include "xla/service/hlo_pass_fix.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/hlo_verifier.h" #include "xla/service/layout_assignment.h" #include "xla/stream_executor/device_description.h" diff --git a/xla/service/gpu/fusion_pipeline.h b/xla/service/gpu/fusion_pipeline.h index 994dd56802518..fab9924831088 100644 --- a/xla/service/gpu/fusion_pipeline.h +++ b/xla/service/gpu/fusion_pipeline.h @@ -16,8 +16,8 @@ limitations under the License. #ifndef XLA_SERVICE_GPU_FUSION_PIPELINE_H_ #define XLA_SERVICE_GPU_FUSION_PIPELINE_H_ +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/hlo_cost_analysis.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/stream_executor/device_description.h" #include "xla/xla.pb.h" #include "tsl/platform/threadpool.h" diff --git a/xla/service/gpu/gpu_compiler.cc b/xla/service/gpu/gpu_compiler.cc index cdbe98490bf47..00cbec5674d00 100644 --- a/xla/service/gpu/gpu_compiler.cc +++ b/xla/service/gpu/gpu_compiler.cc @@ -68,6 +68,8 @@ limitations under the License. #include "xla/hlo/ir/hlo_module_group.h" #include "xla/hlo/ir/hlo_opcode.h" #include "xla/hlo/ir/hlo_schedule.h" +#include "xla/hlo/pass/hlo_pass_fix.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/maybe_owning.h" #include "xla/service/algebraic_simplifier.h" #include "xla/service/all_gather_broadcast_reorder.h" @@ -193,8 +195,6 @@ limitations under the License. #include "xla/service/hlo_dataflow_analysis.h" #include "xla/service/hlo_dce.h" #include "xla/service/hlo_module_config.h" -#include "xla/service/hlo_pass_fix.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/hlo_rematerialization.h" #include "xla/service/hlo_verifier.h" #include "xla/service/host_memory_transfer_asyncifier.h" diff --git a/xla/service/gpu/gpu_compiler.h b/xla/service/gpu/gpu_compiler.h index 4db57d7059d9f..6b1cf6f0c4558 100644 --- a/xla/service/gpu/gpu_compiler.h +++ b/xla/service/gpu/gpu_compiler.h @@ -28,6 +28,7 @@ limitations under the License. #include "xla/autotune_results.pb.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_module_group.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/pjrt/distributed/key_value_store_interface.h" #include "xla/service/algebraic_simplifier.h" #include "xla/service/compiler.h" @@ -41,7 +42,6 @@ limitations under the License. #include "xla/service/hlo_cost_analysis.h" #include "xla/service/hlo_dataflow_analysis.h" #include "xla/service/hlo_module_config.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/llvm_compiler.h" #include "xla/stream_executor/device_description.h" #include "xla/stream_executor/device_description.pb.h" diff --git a/xla/service/gpu/gpu_hlo_schedule.cc b/xla/service/gpu/gpu_hlo_schedule.cc index be25cd091e6cb..ee2e9874e358c 100644 --- a/xla/service/gpu/gpu_hlo_schedule.cc +++ b/xla/service/gpu/gpu_hlo_schedule.cc @@ -41,6 +41,7 @@ limitations under the License. #include "xla/hlo/ir/hlo_instructions.h" #include "xla/hlo/ir/hlo_opcode.h" #include "xla/hlo/ir/hlo_schedule.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/hlo/utils/hlo_query.h" #include "xla/service/buffer_value.h" #include "xla/service/collective_ops_utils.h" @@ -51,7 +52,6 @@ limitations under the License. #include "xla/service/gpu/transforms/schedule_postprocessing.h" #include "xla/service/gpu/transforms/scheduling_instruction_annotator.h" #include "xla/service/hlo_memory_scheduler.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/latency_hiding_scheduler.h" #include "xla/service/p2p_schedule_preparation.h" #include "xla/service/profile_guided_latency_estimator.h" diff --git a/xla/service/gpu/gpu_p2p_pipeliner.cc b/xla/service/gpu/gpu_p2p_pipeliner.cc index 7ec125d9023ea..d097d1b638251 100644 --- a/xla/service/gpu/gpu_p2p_pipeliner.cc +++ b/xla/service/gpu/gpu_p2p_pipeliner.cc @@ -27,10 +27,10 @@ limitations under the License. #include "absl/strings/str_join.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/collective_ops_utils.h" #include "xla/service/collective_pipeliner.h" #include "xla/service/hlo_parser.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/util.h" namespace xla { diff --git a/xla/service/gpu/gpu_p2p_pipeliner.h b/xla/service/gpu/gpu_p2p_pipeliner.h index 6bd3b588eed8c..30a18b4a48d81 100644 --- a/xla/service/gpu/gpu_p2p_pipeliner.h +++ b/xla/service/gpu/gpu_p2p_pipeliner.h @@ -16,7 +16,7 @@ limitations under the License. #ifndef XLA_SERVICE_GPU_GPU_P2P_PIPELINER_H_ #define XLA_SERVICE_GPU_GPU_P2P_PIPELINER_H_ -#include "xla/service/hlo_pass_pipeline.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" namespace xla { namespace gpu { diff --git a/xla/service/gpu/gpu_p2p_pipeliner_test.cc b/xla/service/gpu/gpu_p2p_pipeliner_test.cc index d3b6edcc0a10a..df46055ce0d7e 100644 --- a/xla/service/gpu/gpu_p2p_pipeliner_test.cc +++ b/xla/service/gpu/gpu_p2p_pipeliner_test.cc @@ -29,9 +29,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/service/hlo_module_config.h" #include "xla/service/hlo_parser.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/hlo_verifier.h" #include "xla/tests/filecheck.h" #include "xla/tests/hlo_test_base.h" diff --git a/xla/service/gpu/gpu_spmd_pipeline.cc b/xla/service/gpu/gpu_spmd_pipeline.cc index 06e1e6fa1594b..b6d99a85194d5 100644 --- a/xla/service/gpu/gpu_spmd_pipeline.cc +++ b/xla/service/gpu/gpu_spmd_pipeline.cc @@ -23,6 +23,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_fix.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/hlo/transforms/hlo_constant_splitter.h" #include "xla/service/algebraic_simplifier.h" #include "xla/service/conditional_simplifier.h" @@ -31,8 +33,6 @@ limitations under the License. #include "xla/service/hlo_constant_folding.h" #include "xla/service/hlo_dce.h" #include "xla/service/hlo_module_config.h" -#include "xla/service/hlo_pass_fix.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/reshape_mover.h" #include "xla/service/scatter_expander.h" #include "xla/service/sharding_propagation.h" diff --git a/xla/service/gpu/gpu_spmd_pipeline.h b/xla/service/gpu/gpu_spmd_pipeline.h index 4749df6c4ab93..e969965f5842b 100644 --- a/xla/service/gpu/gpu_spmd_pipeline.h +++ b/xla/service/gpu/gpu_spmd_pipeline.h @@ -20,8 +20,8 @@ limitations under the License. #include "absl/functional/function_ref.h" #include "xla/hlo/ir/hlo_module.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/algebraic_simplifier.h" -#include "xla/service/hlo_pass_pipeline.h" namespace xla { namespace gpu { diff --git a/xla/service/gpu/gpu_spmd_pipeline_test.cc b/xla/service/gpu/gpu_spmd_pipeline_test.cc index 42a9e7dcad49f..c0a7d68ebc3af 100644 --- a/xla/service/gpu/gpu_spmd_pipeline_test.cc +++ b/xla/service/gpu/gpu_spmd_pipeline_test.cc @@ -25,10 +25,10 @@ limitations under the License. #include "absl/log/log.h" #include "xla/client/executable_build_options.h" #include "xla/hlo/ir/hlo_instruction.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/algebraic_simplifier.h" #include "xla/service/hlo_module_config.h" #include "xla/service/hlo_parser.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/shape_util.h" #include "xla/stream_executor/device_description.h" #include "xla/tests/hlo_test_base.h" diff --git a/xla/service/gpu/nvptx_compiler.cc b/xla/service/gpu/nvptx_compiler.cc index 6a691cea969e2..1efdec351ff22 100644 --- a/xla/service/gpu/nvptx_compiler.cc +++ b/xla/service/gpu/nvptx_compiler.cc @@ -44,6 +44,8 @@ limitations under the License. #include "llvm/Support/raw_ostream.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_opcode.h" +#include "xla/hlo/pass/hlo_pass_fix.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/pjrt/distributed/key_value_store_interface.h" #include "xla/service/algebraic_simplifier.h" #include "xla/service/call_inliner.h" @@ -86,8 +88,6 @@ limitations under the License. #include "xla/service/hlo_dataflow_analysis.h" #include "xla/service/hlo_dce.h" #include "xla/service/hlo_module_config.h" -#include "xla/service/hlo_pass_fix.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/hlo_verifier.h" #include "xla/service/llvm_ir/llvm_util.h" #include "xla/service/reshape_mover.h" diff --git a/xla/service/gpu/nvptx_compiler.h b/xla/service/gpu/nvptx_compiler.h index 060aabccc2fae..4543f01a31717 100644 --- a/xla/service/gpu/nvptx_compiler.h +++ b/xla/service/gpu/nvptx_compiler.h @@ -30,13 +30,13 @@ limitations under the License. #include "llvm/IR/Module.h" #include "xla/autotune_results.pb.h" #include "xla/hlo/ir/hlo_module.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/pjrt/distributed/key_value_store_interface.h" #include "xla/service/gpu/autotuning/autotuner_util.h" #include "xla/service/gpu/gpu_compiler.h" #include "xla/service/gpu/ir_emission_utils.h" #include "xla/service/hlo_dataflow_analysis.h" #include "xla/service/hlo_module_config.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/stream_executor/cuda/ptx_linking_method.h" #include "xla/stream_executor/device_description.h" #include "xla/stream_executor/device_memory_allocator.h" diff --git a/xla/service/gpu/prepare_hlo_for_ir_emitting_pipeline.cc b/xla/service/gpu/prepare_hlo_for_ir_emitting_pipeline.cc index 7ced52cbd17ca..8704737b711e2 100644 --- a/xla/service/gpu/prepare_hlo_for_ir_emitting_pipeline.cc +++ b/xla/service/gpu/prepare_hlo_for_ir_emitting_pipeline.cc @@ -20,6 +20,7 @@ limitations under the License. #include #include "xla/hlo/ir/hlo_module.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/copy_insertion.h" #include "xla/service/cpu_gpu_shape_verifier.h" #include "xla/service/gpu/transforms/alias_passthrough_params.h" @@ -28,7 +29,6 @@ limitations under the License. #include "xla/service/gpu/transforms/sanitize_constant_names.h" #include "xla/service/hlo_dataflow_analysis.h" #include "xla/service/hlo_dce.h" -#include "xla/service/hlo_pass_pipeline.h" #include "xla/service/hlo_verifier.h" #include "xla/service/layout_assignment.h" #include "xla/service/loop_schedule_linearizer.h" diff --git a/xla/service/gpu/prepare_hlo_for_ir_emitting_pipeline.h b/xla/service/gpu/prepare_hlo_for_ir_emitting_pipeline.h index 8a40b10490fe2..095907f39794a 100644 --- a/xla/service/gpu/prepare_hlo_for_ir_emitting_pipeline.h +++ b/xla/service/gpu/prepare_hlo_for_ir_emitting_pipeline.h @@ -17,8 +17,8 @@ limitations under the License. #define XLA_SERVICE_GPU_PREPARE_HLO_FOR_IR_EMITTING_PIPELINE_H_ #include "xla/hlo/ir/hlo_module.h" +#include "xla/hlo/pass/hlo_pass_pipeline.h" #include "xla/service/hlo_dataflow_analysis.h" -#include "xla/service/hlo_pass_pipeline.h" namespace xla { namespace gpu {