diff --git a/src/utils/Decompressor.h b/src/utils/Decompressor.h index 5551b572..5d321e5c 100644 --- a/src/utils/Decompressor.h +++ b/src/utils/Decompressor.h @@ -5,6 +5,7 @@ #include #include #include +#include enum class CompressionType { MIO0, diff --git a/src/utils/TorchUtils.h b/src/utils/TorchUtils.h index 6c8b6c4f..36ce46da 100644 --- a/src/utils/TorchUtils.h +++ b/src/utils/TorchUtils.h @@ -5,6 +5,7 @@ #include #include #include +#include namespace Torch { template< typename T >