mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 14:43:57 -05:00
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:
@@ -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()
|
||||
Reference in New Issue
Block a user