mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-06-27 08:52:05 +08:00
Adeeb Shihadeh
100f89a161
* openpilot v0.9.9 release date: 2025-06-05T19:54:08 master commit: 8aadf02b2fd91f4e1285e18c2c7feb32d93b66f5 * AGNOS 12.4 (#35558) agnos12.4 --------- Co-authored-by: Vehicle Researcher <user@comma.ai> Co-authored-by: Maxime Desroches <desroches.maxime@gmail.com>
28 lines
901 B
Python
28 lines
901 B
Python
import unittest
|
|
from tinygrad import Tensor, Device
|
|
import time
|
|
|
|
def time_tensor_numpy(out:Tensor):
|
|
times = []
|
|
for _ in range(5):
|
|
st = time.perf_counter()
|
|
out.lazydata.base.realized.as_buffer(allow_zero_copy=True)
|
|
et = time.perf_counter() - st
|
|
times.append(et)
|
|
return min(times)
|
|
|
|
N = 4096
|
|
class TestZeroCopy(unittest.TestCase):
|
|
@unittest.skipIf(Device.DEFAULT not in {"CPU", "LLVM", "METAL"}, "device isn't zero copy")
|
|
def test_zero_copy_from_default_to_cpu(self):
|
|
demo = Tensor.rand(1).realize()
|
|
t1 = time_tensor_numpy(demo)
|
|
out = Tensor.rand(N, N).realize()
|
|
t2 = time_tensor_numpy(out)
|
|
gbps = out.nbytes()*1e-9/max(t2-t1, 1e-10)
|
|
print(f"time(base): {t1*1e3:.2f} ms, time(copy): {t2*1e3:.2f} ms : copy speed {gbps:.2f} GB/s")
|
|
self.assertGreater(gbps, 600) # more than 600 GB/s = no copy
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=2)
|