forked from tinygrad/tinygrad
chenyuandGitHub
bdede4924e
factor might not be a integer if input global_size has an odd number in it
11 lines
464 B
Python
11 lines
464 B
Python
import unittest
|
|
from tinygrad.engine.search import get_test_global_size
|
|
|
|
class TestSearchUtil(unittest.TestCase):
|
|
def test_get_test_global_size(self):
|
|
self.assertEqual(get_test_global_size([256, 256, 256], 65536, {}), ([256, 16, 16], 256.0))
|
|
self.assertEqual(get_test_global_size([65536, 1, 1], 256, {}), ([256, 1, 1], 256.0))
|
|
self.assertEqual(get_test_global_size([77, 1, 1], 16, {}), ([9, 1, 1], 77/9))
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() |