From dcfcfaccded21137a40bb7ff78c13c2012c89990 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Thu, 21 Nov 2024 12:12:45 +0800 Subject: [PATCH] delete preallocate, it's unused --- tinygrad/device.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tinygrad/device.py b/tinygrad/device.py index 9603cead80..8096bedddf 100644 --- a/tinygrad/device.py +++ b/tinygrad/device.py @@ -55,7 +55,7 @@ class BufferSpec: class Buffer: def __init__(self, device:str, size:int, dtype:DType, opaque:Any=None, options:Optional[BufferSpec]=None, - initial_value:Optional[bytes]=None, lb_refcount=0, base:Optional[Buffer]=None, offset:int=0, preallocate=False): + initial_value:Optional[bytes]=None, lb_refcount=0, base:Optional[Buffer]=None, offset:int=0): if isinstance(dtype, ImageDType): options = BufferSpec(image=dtype) # TODO: image hack shouldn't be here. where should it be? else: assert isinstance(dtype, DType) and not isinstance(dtype, PtrDType) self.device, self.size, self.dtype, self.options, self.offset = device, size, dtype, options, offset @@ -71,7 +71,6 @@ class Buffer: assert base._base is None, "base can't have a base" assert device == base.device, "base must have the same device" self._base = base - if preallocate: self.allocate() @property def base(self) -> Buffer: return self._base if self._base is not None else self @property