forked from tinygrad/tinygrad
remove no-op contiguous from rand (#12329)
This commit is contained in:
+1
-1
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user