From df16cc94c905930d8c4e6c4a01d05d3c259af0cd Mon Sep 17 00:00:00 2001
From: xla authors <google-ml-automation@google.com>
Date: Thu, 24 Oct 2024 19:32:37 -0700
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 689602164
---
 .../hlo_legalize_to_arithmetic/hlo_legalize_to_arithmetic.cc    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/xla/mlir_hlo/mhlo/transforms/hlo_legalize_to_arithmetic/hlo_legalize_to_arithmetic.cc b/xla/mlir_hlo/mhlo/transforms/hlo_legalize_to_arithmetic/hlo_legalize_to_arithmetic.cc
index 9c03b38717473..f30742b801532 100644
--- a/xla/mlir_hlo/mhlo/transforms/hlo_legalize_to_arithmetic/hlo_legalize_to_arithmetic.cc
+++ b/xla/mlir_hlo/mhlo/transforms/hlo_legalize_to_arithmetic/hlo_legalize_to_arithmetic.cc
@@ -15,6 +15,8 @@ limitations under the License.
 
 // This file implements logic for lowering HLO dialect to LHLO dialect.
 
+#include <cassert>
+#include <cstdint>
 #include <memory>
 #include <optional>
 #include <utility>