diff options
author | Kuy Mainwaring <kuym@google.com> | 2024-05-24 10:42:30 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2024-05-24 11:00:10 -0700 |
commit | 306833457d32dd27a198ec8f6839b4966f941d4d (patch) | |
tree | c9b58c870641bc8d520d06dee21051cc32116975 | |
parent | f3b19c7dfea2da355277a2c026ef331b1ef06e00 (diff) | |
download | tensorflow-upstream-master.tar.gz |
[XLA:GPU] Clang-tidy cleanup for xla/service/gpu/cudnn_workspace_rewriter.hupstream-master
PiperOrigin-RevId: 636966632
-rw-r--r-- | third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.h b/third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.h index 530646b0b43..a4283e488c1 100644 --- a/third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.h +++ b/third_party/xla/xla/service/gpu/cudnn_workspace_rewriter.h @@ -16,9 +16,12 @@ limitations under the License. #ifndef XLA_SERVICE_GPU_CUDNN_WORKSPACE_REWRITER_H_ #define XLA_SERVICE_GPU_CUDNN_WORKSPACE_REWRITER_H_ +#include <string> + +#include "absl/container/flat_hash_map.h" +#include "absl/container/flat_hash_set.h" #include "absl/status/statusor.h" #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/stream_executor/dnn.h" |