25 lines
984 B
Python
25 lines
984 B
Python
from numba.cuda.testing import ensure_supported_ccs_initialized
|
|
from numba.testing import unittest
|
|
from numba.testing import load_testsuite
|
|
from numba import cuda
|
|
from os.path import dirname, join
|
|
|
|
|
|
def load_tests(loader, tests, pattern):
|
|
suite = unittest.TestSuite()
|
|
this_dir = dirname(__file__)
|
|
ensure_supported_ccs_initialized()
|
|
suite.addTests(load_testsuite(loader, join(this_dir, 'nocuda')))
|
|
if cuda.is_available():
|
|
suite.addTests(load_testsuite(loader, join(this_dir, 'cudasim')))
|
|
gpus = cuda.list_devices()
|
|
if gpus and gpus[0].compute_capability >= (2, 0):
|
|
suite.addTests(load_testsuite(loader, join(this_dir, 'cudadrv')))
|
|
suite.addTests(load_testsuite(loader, join(this_dir, 'cudapy')))
|
|
suite.addTests(load_testsuite(loader, join(this_dir, 'doc_examples')))
|
|
else:
|
|
print("skipped CUDA tests because GPU CC < 2.0")
|
|
else:
|
|
print("skipped CUDA tests")
|
|
return suite
|