diff --git a/.gitignore b/.gitignore index 9a0ed566..808dd71d 100644 --- a/.gitignore +++ b/.gitignore @@ -26,3 +26,4 @@ datasets/imagenet/ datasets/squad/ datasets/img_align_celeba* datasets/open-images-v6-mlperf +datasets/kits/ diff --git a/tinygrad/lazy.py b/tinygrad/lazy.py index 1e21725f..b3399961 100644 --- a/tinygrad/lazy.py +++ b/tinygrad/lazy.py @@ -71,7 +71,7 @@ def create_lazybuffer(device:str, shape:Union[ShapeTracker, Tuple[int, ...]], op st = shape if isinstance(shape, ShapeTracker) else ShapeTracker(tuple(shape)) # fromcpu aren't cached - if optype == LoadOps and op.op == LoadOps.FROMCPU: return LazyBuffer(device, st, optype, op, dtype) + if optype == LoadOps and op.op in [LoadOps.FROMCPU, LoadOps.EMPTY]: return LazyBuffer(device, st, optype, op, dtype) #print("create_lazybuffer", device, shape, optype, op, dtype)