diff options
| author | Kuy Mainwaring <kuym@google.com> | 2024-05-20 17:27:23 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2024-05-20 18:45:58 -0700 |
| commit | 32e5b6ff793d5cbff41a99b85887b6058cb434fb (patch) | |
| tree | dfe976176541ebdbae0c37447001585e7a463b7d | |
| parent | bdfcc5062b12d7592614d22ddb0a5cd85e11b176 (diff) | |
| download | tensorflow-upstream-master.tar.gz | |
[XLA:GPU] Clang-tidy cleanup for xla/service/gpu/cudnn_workspace_rewriter.ccupstream-master
PiperOrigin-RevId: 635617583
| -rw-r--r-- | third_party/xla/xla/service/gpu/BUILD | 4 | ||||
| -rw-r--r-- | third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.cc | 17 |
2 files changed, 10 insertions, 11 deletions
diff --git a/third_party/xla/xla/service/gpu/BUILD b/third_party/xla/xla/service/gpu/BUILD index 60169c2a9a9..bf2d24bc389 100644 --- a/third_party/xla/xla/service/gpu/BUILD +++ b/third_party/xla/xla/service/gpu/BUILD @@ -2995,6 +2995,8 @@ cc_library( ":stream_executor_util", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -3008,7 +3010,7 @@ cc_library( "//xla/stream_executor/cuda:cudnn_plugin", "@local_tsl//tsl/platform:errors", "@local_tsl//tsl/platform:statusor", - "@com_google_absl//absl/container:flat_hash_set", + "//xla:status_macros", ]), ) diff --git a/third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.cc b/third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.cc index 8e37ea1be34..55c2c4f913c 100644 --- a/third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.cc +++ b/third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.cc @@ -16,15 +16,16 @@ limitations under the License. #include "xla/service/gpu/cudnn_workspace_rewriter.h" #include <optional> -#include <string> +#include <utility> #include <vector> +#include "absl/container/flat_hash_set.h" +#include "absl/container/inlined_vector.h" #include "absl/log/check.h" #include "absl/log/log.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/string_view.h" -#include "third_party/gpus/cudnn/cudnn_version.h" #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_clone_context.h" @@ -32,17 +33,16 @@ 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/ir/hlo_opcode.h" -#include "xla/primitive_util.h" #include "xla/service/gpu/backend_configs.pb.h" #include "xla/service/gpu/cublas_cudnn.h" #include "xla/service/gpu/gpu_fused_mha_runner.h" -#include "xla/service/gpu/ir_emission_utils.h" #include "xla/service/gpu/stream_executor_util.h" +#include "xla/shape.h" +#include "xla/shape_util.h" +#include "xla/status_macros.h" #include "xla/stream_executor/cuda/cuda_dnn.h" -#include "xla/stream_executor/cuda/cudnn_frontend_helpers.h" +#include "xla/stream_executor/dnn.h" #include "xla/util.h" -#include "tsl/platform/errors.h" #include "tsl/platform/statusor.h" namespace xla { @@ -50,9 +50,6 @@ namespace gpu { namespace { -namespace fe = cudnn_frontend; -namespace graph = fe::graph; - // create cuDNN graphs from HloCustomCall absl::StatusOr<se::gpu::CudnnGraph> HloCustomCallToCuDnnGraph( se::dnn::DnnSupport& dnn_support, |