diff --git a/tinygrad/device.py b/tinygrad/device.py index 7db5310bf8..2e4b1f2520 100644 --- a/tinygrad/device.py +++ b/tinygrad/device.py @@ -5,7 +5,7 @@ from typing import Any, Generic, TypeVar, Iterator, Sequence, cast, Generator import importlib, inspect, functools, pathlib, os, platform, contextlib, sys, re, atexit, pickle, decimal from tinygrad.helpers import CI, OSX, LRU, getenv, diskcache_get, diskcache_put, DEBUG, GlobalCounters, flat_mv, PROFILE, temp, colored, CPU_LLVM from tinygrad.helpers import Context, DISABLE_COMPILER_CACHE, ALLOW_DEVICE_USAGE, MAX_BUFFER_SIZE, cpu_events, ProfileEvent, ProfilePointEvent, dedup -from tinygrad.helpers import unwrap_class_type +from tinygrad.helpers import unwrap_class_type, suppress_finalizing from tinygrad.dtype import DType, ImageDType, PtrDType, dtypes, _to_np_dtype from tinygrad.renderer import Renderer @@ -163,6 +163,7 @@ class Buffer: return self._trace_num @property def nbytes(self): return self.size*self.dtype.itemsize + @suppress_finalizing def __del__(self): (not hasattr(self, '_buf')) or self.deallocate() def __repr__(self): return f"