print test durations and add speed (#2107)

* print test durations

* decrease sizes to increase speed

* faster

* GPU/CLANG onnx in seperate runner

* test split, move ONNX CPU CI

* simpler tests

* simpler uops test

* faster

* less cuda apt

* running ninja install

* apt install

* split fancy indexing
This commit is contained in:
George Hotz
2023-10-18 13:46:42 -07:00
committed by GitHub
parent e2a1c2aaa6
commit 15da96f393
8 changed files with 123 additions and 112 deletions

View File

@@ -68,7 +68,7 @@ class TestEfficientNet(unittest.TestCase):
self.assertEqual(label, "hen")
def test_chicken_bigbatch(self):
label = _infer(self.model, chicken_img, 4)
label = _infer(self.model, chicken_img, 2)
self.assertEqual(label, "hen")
def test_car(self):