diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index a1b0c6ac76..fe15acb2ec 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -535,7 +535,7 @@ class Tensor(MathTrait): device=device, dtype=dtypes.uint32, requires_grad=False) Tensor._device_rng_counters[device] = Tensor([num], device=device, dtype=dtypes.uint32, requires_grad=False) # increment rng counter for devices - else: Tensor._device_rng_counters[device].assign(Tensor._device_rng_counters[device] + num).contiguous() + else: Tensor._device_rng_counters[device].assign(Tensor._device_rng_counters[device] + num) # threefry random bits bits_count = Tensor._device_rng_counters[device] - num