[XLA:GPU] Clang-tidy cleanup for xla/service/gpu/cudnn_workspace_rewr…

…iter.h PiperOrigin-RevId: 636966632
Intel-tensorflow · May 24, 2024 · 3068334 · 3068334
1 parent f3b19c7
commit 3068334
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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"