From 25beea576956b26b802e17379bc3a0cabc6003c9 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 21 Oct 2025 09:04:36 +0800 Subject: [PATCH] hotfix: suppress_finalizing on device __del__ --- tinygrad/device.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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"