summaryrefslogtreecommitdiff
path: root/src/utils
diff options
context:
space:
mode:
authorKiritoDv <kiritodev01@gmail.com>2024-01-31 12:45:31 -0600
committerKiritoDv <kiritodev01@gmail.com>2024-01-31 12:46:05 -0600
commitde35c441fcc6301a43d1572aceb7b4225ea45e0b (patch)
tree6108463b2e80f01b74ab596ec41fbc1f74c7dda3 /src/utils
parent16a825c0d02c3f28f273216b86582758b5710d94 (diff)
Fixed linux issues
Diffstat (limited to 'src/utils')
-rw-r--r--src/utils/Decompressor.h1
-rw-r--r--src/utils/TorchUtils.h1
2 files changed, 2 insertions, 0 deletions
diff --git a/src/utils/Decompressor.h b/src/utils/Decompressor.h
index 4e3c298..5551b57 100644
--- a/src/utils/Decompressor.h
+++ b/src/utils/Decompressor.h
@@ -4,6 +4,7 @@
#include <cstdint>
#include <unordered_map>
#include <yaml-cpp/yaml.h>
+#include <optional>
enum class CompressionType {
MIO0,
diff --git a/src/utils/TorchUtils.h b/src/utils/TorchUtils.h
index 330c7fa..e3d89e6 100644
--- a/src/utils/TorchUtils.h
+++ b/src/utils/TorchUtils.h
@@ -3,6 +3,7 @@
#include <string>
#include <fstream>
#include <sstream>
+#include <iomanip>
namespace Torch {
template< typename T >