gpu_opt.cc
imports xla/stream_executor/platform.himports xla/stream_executor/platform/initialize.h
SHA-256a9c50b9732d90c8998e28c0bc0e00c680aec6a37c6fe09320716427f7460f3dd
MaleculeMd(In₄)
Evidence
1/* Copyright 2023 The OpenXLA Authors.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is …
11:42… IND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14==============================================================================*/
15
16#include <cstdint>
17#include <memory>
18#include <optional>
19#include <set>
20#include <string>
21#include <utility>
22
23#include "absl/log/log.h"
24#include "absl/status/statusor.h"
25#include "absl/strings/string_view.h"
26#include "llvm/IR/LLVMContext.h"
27#include "xla/backends/gpu/transforms/collectives/all_gather_optimizer.h"
28#include "xla/backends/gpu/transforms/cudnn_custom_call_converter.h"
29#include "xla/backends/gpu/transforms/dot_algorithm_rewriter.h"
30#include "xla/backends/gpu/transforms/dot_dimension_sorter.h"
31#include "xla/backends/gpu/transforms/dot_normalizer.h"
32#include "xla/backends/gpu/transforms/dot_operand_converter.h"
33#include "xla/backends/gpu/transforms/gemm_broadcast_folding_rewriter.h"
34#include "xla/backends/gpu/transforms/gemm_fusion.h"
35#include "xla/backends/gpu/transforms/gemv_rewriter.h"
36#include "xla/backends/gpu/transforms/reduce_scatter_creator.h"
37#include "xla/backends/gpu/transforms/reduction_degenerate_dim_remover.h"
38#include "xla/backends/gpu/transforms/reduction_dimension_grouper.h"
39#include "xla/backends/gpu/transforms/reduction_layout_normalizer.h"
40#include "xla/backends/gpu/transforms/sanitize_constant_names.h"
41#include "xla/backends/gpu/transforms/topk_specializer.h"
42#include "xla/backends/gpu/transforms/topk_splitter.h"
43#include "xla/backends/gpu/transforms/windowed_einsum_handler.h"
44#include "xla/hlo/ir/hlo_module.h"
45#include "xla/hlo/tools/hlo_opt/opt_lib.h"
46#include "xla/hlo/transforms/host_offloader.h"
47#include "xla/hlo/transforms/simplifiers/hlo_memory_scheduler.h"
48#include "xla/layout.h"
49#include "xla/service/buffer_value.h"
50#include "xla/service/compiler.h"
51#include "xla/service/copy_insertion.h"
52#include "xla/service/dump.h"
53#include "xla/service/executable.h"
54#include "xla/service/gpu/alias_info.h"
55#include "xla/service/gpu/compile_module_to_llvm_ir.h"
56#include "xla/service/gpu/gpu_compiler.h"
57#include "xla/service/gpu/gpu_executable.h"
58#include "xla/service/gpu/gpu_hlo_schedule.h"