commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

oops, remove while(1)

This commit is contained in:
George Hotz 2022-10-29 14:05:13 -07:00
parent 4b6097f81d
commit 544cb0a069
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_speed_v_torch.py
View File

@ -12,7 +12,7 @@ IN_CHANS = [int(x) for x in os.getenv("IN_CHANS", "1,16,64").split(",")]
CNT = 5
class TestSpeed(unittest.TestCase):
def test_gemm(self):
N = 2048
N = 1024
torch.manual_seed(0)
torch_a = torch.rand(N, N)
torch_b = torch.rand(N, N)
@ -23,8 +23,7 @@ class TestSpeed(unittest.TestCase):
for _ in range(CNT):
torch_a += 1
st = time.monotonic()
while 1:
torch_c = torch_a @ torch_b
torch_c = torch_a @ torch_b
et_torch = (time.monotonic() - st) * 1000
ets_torch.append(et_torch)