From 09e457953a86582e7a5d5dcdc71977ae685a2fc8 Mon Sep 17 00:00:00 2001
From: Tianlei Wu <tlwu@microsoft.com>
Date: Fri, 17 Jan 2025 22:59:42 +0000
Subject: [PATCH] fix type cast warning

---
 onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc | 3 +++
 onnxruntime/core/framework/debug_node_inputs_outputs_utils.h  | 1 +
 onnxruntime/core/framework/print_tensor_statistics_utils.h    | 4 ++--
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc b/onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc
index d4ab3a4ac1853..7bd825a9b0bb1 100644
--- a/onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc
+++ b/onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc
@@ -8,7 +8,10 @@
 #include "core/framework/print_tensor_statistics_utils.h"
 #include <iomanip>
 #include <cctype>
+#include <vector>
 #include <string>
+#include <algorithm>
+#include <utility>
 
 #ifdef DEBUG_NODE_INPUTS_OUTPUTS_ENABLE_DUMP_TO_SQLDB
 #include <sqlite3.h>
diff --git a/onnxruntime/core/framework/debug_node_inputs_outputs_utils.h b/onnxruntime/core/framework/debug_node_inputs_outputs_utils.h
index 990df27a2b48c..2ea7d59ad620e 100644
--- a/onnxruntime/core/framework/debug_node_inputs_outputs_utils.h
+++ b/onnxruntime/core/framework/debug_node_inputs_outputs_utils.h
@@ -20,6 +20,7 @@
 #include "core/framework/session_state.h"
 #include "core/graph/graph.h"
 #include <unordered_set>
+#include <unordered_map>
 #include <mutex>
 #include <string>
 
diff --git a/onnxruntime/core/framework/print_tensor_statistics_utils.h b/onnxruntime/core/framework/print_tensor_statistics_utils.h
index 60057150f40c0..c2030424ef19d 100644
--- a/onnxruntime/core/framework/print_tensor_statistics_utils.h
+++ b/onnxruntime/core/framework/print_tensor_statistics_utils.h
@@ -89,8 +89,8 @@ void PrintCommonStats(const T* data, size_t count, TensorStatisticsData& tensor_
   // Statistics for float and double only for now.
   if constexpr (std::is_same<T, float>::value) {
     tensor_statistics.is_float = true;
-    tensor_statistics.float_min = static_cast<double>(min);
-    tensor_statistics.float_max = static_cast<double>(max);
+    tensor_statistics.float_min = static_cast<float>(min);
+    tensor_statistics.float_max = static_cast<float>(max);
   }
 }