| 12345678910 |
- import os
- if "DEBUG" not in os.environ: os.environ["DEBUG"] = "2"
- if "THREEFRY" not in os.environ: os.environ["THREEFRY"] = "1"
- from tinygrad import Tensor, GlobalCounters
- for N in [10_000_000, 100_000_000, 1_000_000_000]:
- GlobalCounters.reset()
- Tensor.rand(N).realize()
- print(f"N {N:>20_}, global_ops {GlobalCounters.global_ops:>20_}, global_mem {GlobalCounters.global_mem:>20_}")
|
