diff --git a/xla/mlir/framework/transforms/outline_with_xla_framework.cc b/xla/mlir/framework/transforms/outline_with_xla_framework.cc
index 7cafdfa3bcb23e..b960958a7d6344 100644
--- a/xla/mlir/framework/transforms/outline_with_xla_framework.cc
+++ b/xla/mlir/framework/transforms/outline_with_xla_framework.cc
@@ -14,7 +14,6 @@ limitations under the License.
 ==============================================================================*/
 
 #include <memory>
-#include <stdexcept>
 #include <utility>
 
 #include "llvm/ADT/STLExtras.h"
diff --git a/xla/mlir/framework/transforms/xla_framework_to_llvm_pass.cc b/xla/mlir/framework/transforms/xla_framework_to_llvm_pass.cc
index 703b6c9af785d5..c40a7ad1b9aa46 100644
--- a/xla/mlir/framework/transforms/xla_framework_to_llvm_pass.cc
+++ b/xla/mlir/framework/transforms/xla_framework_to_llvm_pass.cc
@@ -14,8 +14,9 @@ limitations under the License.
 ==============================================================================*/
 
 #include <array>
+#include <cassert>
+#include <cstdint>
 #include <memory>
-#include <stdexcept>
 #include <utility>
 
 #include "llvm/ADT/ArrayRef.h"