(self)
| 41 | self.assertEqual(mem, 512*1024*2) # 1 read + 1 write |
| 42 | |
| 43 | def test_expanded(self): |
| 44 | a = Tensor.empty(1024, 1, dtype=dtypes.uint8).expand(1024, 1024) |
| 45 | b = Tensor.empty(1024, 1024, dtype=dtypes.uint8) |
| 46 | _, mem = get_stats(a+b) |
| 47 | self.assertEqual(mem, 1024*1024*2 + 1024) # 1 full read + 1 lil read + 1 write |
| 48 | |
| 49 | @unittest.skip("no longer supported") |
| 50 | def test_both_expanded(self): |