From c6fb08769526fdab2efb136966c751bf2259729e Mon Sep 17 00:00:00 2001 From: George Hotz Date: Mon, 29 Nov 2021 01:49:21 -0500 Subject: [PATCH] fix old list behavior --- accel/metal/ops_metal.py | 2 +- tinygrad/tensor.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/accel/metal/ops_metal.py b/accel/metal/ops_metal.py index 95b491038f..2702ef0fb0 100644 --- a/accel/metal/ops_metal.py +++ b/accel/metal/ops_metal.py @@ -84,7 +84,7 @@ if __name__ == "__main__": from tinygrad.tensor import Tensor, Device - r1 = Tensor(np.random.random(16).astype(np.float32)-0.5, device=Device.METAL) + r1 = Tensor([-2,-1,0,1,2], device=Device.METAL) r2 = r1.relu() print(r1.cpu()) print(r2.cpu()) diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index 3490fc9fd2..2beeca15ab 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -139,6 +139,8 @@ class Tensor: @staticmethod def _move_data(data, device): + if isinstance(data, list): + data = np.array(data, dtype=np.float32) if isinstance(data, np.ndarray): data = data.view(Device.buffers[Device.CPU]) if isinstance(data, Device.buffers[device]):