mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 15:08:02 -05:00
e7b18cf5c0
filter based on "DEFINE_VAR" instead of "Variable". also added a unit test to make sure dataset includes image and variable kernels
19 lines
785 B
Python
19 lines
785 B
Python
import unittest
|
|
|
|
from extra.optimization.helpers import load_worlds
|
|
|
|
class TestKernelDataset(unittest.TestCase):
|
|
def test_load_worlds_filters(self):
|
|
all_kernels = load_worlds(filter_reduce=False, filter_noimage=False, filter_novariable=False)
|
|
|
|
reduce_kernels = load_worlds(filter_reduce=True, filter_noimage=False, filter_novariable=False)
|
|
self.assertGreater(len(all_kernels), len(reduce_kernels))
|
|
|
|
image_kernels = load_worlds(filter_reduce=False, filter_noimage=True, filter_novariable=False)
|
|
self.assertGreater(len(all_kernels), len(image_kernels))
|
|
|
|
variable_kernels = load_worlds(filter_reduce=False, filter_noimage=False, filter_novariable=True)
|
|
self.assertGreater(len(all_kernels), len(variable_kernels))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |