mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-13 08:28:55 +08:00
d483e4153a
* buffer view is like buffer * fix * swap_reshape_shrink * contiguous on gguf, fix overlap * revert that * _device_supports_view * this * fix that test * 0 buffers * that test was wrong * this * check correct size * contig BUFFER_VIEW * this * fix tests * buffer view tests * om * fix torch * no MOCKGPU * skip
15 lines
458 B
Python
15 lines
458 B
Python
import unittest
|
|
from tinygrad.helpers import GlobalCounters
|
|
from tinygrad.nn.datasets import mnist
|
|
|
|
class TestDataset(unittest.TestCase):
|
|
def test_dataset_is_realized(self):
|
|
X_train, _, _, _ = mnist()
|
|
X_train[0].contiguous().realize()
|
|
GlobalCounters.reset()
|
|
X_train[0].contiguous().realize()
|
|
self.assertLessEqual(GlobalCounters.kernel_count, 1) # 0 if BUFFER_VIEW (zero-copy), 1 otherwise
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|