Portability: g++-11 has different std policy.

It requires stdexcept and limits for std::invalid_arguments and std::numeric_limits.

Fixes #1857

Signed-off-by: MyungJoo Ham <myungjoo.ham@samsung.com>

diff --git a/nntrainer/tensor/memory_pool.cpp b/nntrainer/tensor/memory_pool.cpp
index 9b620c4..e9aed59 100644 (file)
--- a/nntrainer/tensor/memory_pool.cpp
+++ b/nntrainer/tensor/memory_pool.cpp
@@ -12,6 +12,7 @@
 
 #include <numeric>
 #include <vector>
+#include <limits>
 
 #include <memory_pool.h>
 #include <nntrainer_error.h>

diff --git a/nntrainer/tensor/tensor.h b/nntrainer/tensor/tensor.h
index dbb8776..8279e61 100644 (file)
--- a/nntrainer/tensor/tensor.h
+++ b/nntrainer/tensor/tensor.h
@@ -30,6 +30,7 @@
 #include <functional>
 #include <memory>
 #include <vector>
+#include <stdexcept>
 
 #include <tensor_dim.h>