10 lines
263 B
Python
10 lines
263 B
Python
|
import pycutlass
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
pycutlass.get_memory_pool(2**26, 2**26)
|
||
|
|
loader = unittest.TestLoader()
|
||
|
|
tests = loader.discover('./', 'gemm_*.py')
|
||
|
|
testRunner = unittest.runner.TextTestRunner()
|
||
|
|
testRunner.run(tests)
|