mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-26 17:33:44 +08:00
github-actions[bot]
eefc084302
version: sunnypilot v2025.003.000 (dev) date: 2025-12-18T05:48:43 master commit: 16c052af0835f049a67b80251d8cc1114fc08bb4
17 lines
524 B
Python
17 lines
524 B
Python
if __name__ == "__main__":
|
|
import os
|
|
if "DEBUG" not in os.environ: os.environ["DEBUG"] = "2"
|
|
|
|
from tinygrad import Tensor, GlobalCounters
|
|
from tinygrad.helpers import getenv
|
|
|
|
if (seed := getenv("SEED", 0)) != 0:
|
|
Tensor.manual_seed(seed)
|
|
print(f"using seed {Tensor._seed}")
|
|
|
|
for N in [10_000_000, 100_000_000, 1_000_000_000]:
|
|
GlobalCounters.reset()
|
|
t = Tensor.rand(N)
|
|
t.realize()
|
|
print(f"N {N:>20_}, global_ops {GlobalCounters.global_ops:>20_}, global_mem {GlobalCounters.global_mem:>20_}")
|