factor out allocation limit

vyperlang · Oct 4, 2023 · b5d9263 · b5d9263
1 parent 9c71339
commit b5d9263
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/vyper/codegen/memory_allocator.py b/vyper/codegen/memory_allocator.py
@@ -46,6 +46,8 @@ class MemoryAllocator:
 
     next_mem: int
 
+    _ALLOCATION_LIMIT: int = 2**64
+
     def __init__(self, start_position: int = MemoryPositions.RESERVED_MEMORY):
         """
         Initializer.
@@ -111,10 +113,11 @@ def _expand_memory(self, size: int) -> int:
         self.next_mem += size
         self.size_of_mem = max(self.size_of_mem, self.next_mem)
 
-        if self.size_of_mem >= 2**64:
+        if self.size_of_mem >= self._ALLOCATION_LIMIT:
             # this should not be caught
             raise MemoryAllocationException(
-                "Tried to allocate {self.size_of_mem} bytes! (limit is 2**32 bytes)"
+                f"Tried to allocate {self.size_of_mem} bytes! "
+                f"(limit is 2**64 bytes)"
             )
 
         return before_value