github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix early alloc

Browse Source
This commit is contained in:
George Hotz
2023-01-11 21:07:27 -08:00
parent f1378b3ea1
commit 0724fd61f1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/llops/ops_gpu.py
View File

@@ -198,7 +198,7 @@ class GPUBuffer(ExplicitExecAST):
def __init__(self, shape:Union[ShapeTracker, Tuple[int, ...]], hostbuf:Optional[GPUBuffer]=None, backing:Optional[np.ndarray]=None):
super().__init__(shape, hostbuf)
if hostbuf is not None: hostbuf.cl # does the allocation, pylint: disable=W0104
if hostbuf is not None and hostbuf._backing is None: hostbuf.cl # does the allocation of unbacked buffer, pylint: disable=W0104
self._buf : Optional[CLBuffer] = hostbuf._buf if hostbuf is not None else None
self._base_shape : Tuple[int, ...] = hostbuf._base_shape if hostbuf is not None else self.shape
self._backing : Optional[np.ndarray] = hostbuf._backing if hostbuf is not None else backing