(self)
| 30 | self.assertEqual(mem, 1024*1024*3) # 2 reads + 1 write |
| 31 | |
| 32 | def test_add_const(self): |
| 33 | a = Tensor.empty(1024, 1024, dtype=dtypes.uint8) |
| 34 | _, mem = get_stats(a+3) |
| 35 | self.assertEqual(mem, 1024*1024*2) # 1 read + 1 write |
| 36 | |
| 37 | @unittest.skip("depends on subbuffer working") |
| 38 | def test_add_slice(self): |