revert typo in external_multi_gpu.py (#1777)

introduced by fb1cc6bf4b
This commit is contained in:
Pavol Rusnak
2023-09-06 05:46:28 +02:00
committed by GitHub
parent bd06d88c73
commit a50a7ef6f2

View File

@@ -14,7 +14,7 @@ if __name__ == "__main__":
sz = 1024*1024*256 # 1 GB
#sz = 1024*64
with ("CPU creation: ", on_exit=lambda x: f", {(sz*4*2)/x:.2f} GB/sec"):
with Timing("CPU creation: ", on_exit=lambda x: f", {(sz*4*2)/x:.2f} GB/sec"):
c0 = Tensor.ones(sz, device="cpu").realize()
c1 = (Tensor.ones(sz, device="cpu")/2).realize()