mirror of https://github.com/commaai/tinygrad.git
Added missing unittest execution code (#4400)
same code as on every other test file, just missing from this one for some reason.
This commit is contained in:
parent
d4062cb6fc
commit
0627e26140
|
@ -11,4 +11,7 @@ class TestArange(unittest.TestCase):
|
|||
f1 = self._get_flops(256)
|
||||
f2 = self._get_flops(2560)
|
||||
print(f"{f1=}, {f2=}")
|
||||
assert f2 / f1 < 15, f"bad complexity, flops {f2/f1:.1f}X while inputs 10X"
|
||||
assert f2 / f1 < 15, f"bad complexity, flops {f2/f1:.1f}X while inputs 10X"
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
Loading…
Reference in New Issue