Fix?

Author: NicolasHug

src/torchcodec/_core/AVIOBytesContext.cpp

@@ -68,7 +68,11 @@ int64_t AVIOBytesContext::seek(void* opaque, int64_t offset, int whence) {
 }
 
 AVIOToTensorContext::AVIOToTensorContext()
-    : dataContext_{torch::empty({INITIAL_TENSOR_SIZE}, {torch::kUInt8}), 0} {
+    : dataContext_{
+          torch::empty(
+              {AVIOToTensorContext::INITIAL_TENSOR_SIZE},
+              {torch::kUInt8}),
+          0} {
   createAVIOContext(nullptr, &write, &seek, &dataContext_);
 }
 
@@ -78,9 +82,10 @@ int AVIOToTensorContext::write(void* opaque, const uint8_t* buf, int buf_size) {
 
   if (dataContext->current + buf_size > dataContext->outputTensor.numel()) {
     TORCH_CHECK(
-        dataContext->outputTensor.numel() * 2 <= MAX_TENSOR_SIZE,
+        dataContext->outputTensor.numel() * 2 <=
+            AVIOToTensorContext::MAX_TENSOR_SIZE,
         "We tried to allocate an output encoded tensor larger than ",
-        MAX_TENSOR_SIZE,
+        AVIOToTensorContext::MAX_TENSOR_SIZE,
         " bytes. If you think this should be supported, please report.");
 
     // We double the size of the outpout tensor. Calling cat() may not be the

src/torchcodec/_core/AVIOBytesContext.h

@@ -42,8 +42,8 @@ class AVIOToTensorContext : public AVIOContextHolder {
     int64_t current;
   };
 
-  static const int INITIAL_TENSOR_SIZE = 10'000'000; // 10MB
-  static const int MAX_TENSOR_SIZE = 320'000'000; // 320 MB
+  static constexpr int INITIAL_TENSOR_SIZE = 10'000'000; // 10MB
+  static constexpr int MAX_TENSOR_SIZE = 320'000'000; // 320 MB
   static int write(void* opaque, const uint8_t* buf, int buf_size);
   // We need to expose seek() for some formats like mp3.
   static int64_t seek(void* opaque, int64_t offset, int whence);