good changes from openpilot_compile2 (#2000)

* good changed from openpilot_compile2

* float32 image type was wrong

* cleaner way to write that + a test
This commit is contained in:
George Hotz
2023-10-06 13:33:24 -07:00
committed by GitHub
parent 05be57f57f
commit ffa33d743a
10 changed files with 78 additions and 43 deletions

View File

@@ -322,5 +322,10 @@ class TestSchedule(unittest.TestCase):
out = x.permute(0,2,3,1).contiguous()
check_schedule(out, 2, filter_loadops=False)
def test_double_from(self):
x = Tensor([1,2,3,4])
out = x.to('cpu')
check_schedule(out, 0, filter_loadops=False)
if __name__ == '__main__':
unittest.main(verbosity=2)