Set fixed double print precision in logging

NVIDIA-Merlin · May 20, 2021 · 8eef3f4 · 8eef3f4
1 parent f5a6e99
commit 8eef3f4
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/HugeCTR/include/common.hpp b/HugeCTR/include/common.hpp
@@ -368,11 +368,20 @@ typedef struct DataSetHeader_ {
   } while (0)
 
 template <typename T>
-inline void print_func(T& t) {
+inline void print_func(T const& t) {
   std::cout << t << ", ";
   return;
 }
 
+// Set precision for double type
+template <>
+inline void print_func<double>(double const& t) {
+  std::stringstream ss;
+  ss << std::fixed << std::setprecision(2) << t << ", ";
+  std::cout << ss.str();
+  return;
+}
+
 template <typename... Args>
 inline void LOG(const Args&... args) {
   std::cout << "[";