def prepare(pkg):
if 'cuda' in pkg.__name__:
if not ek.has_cuda():
pytest.skip('CUDA mode is unsupported')
elif 'llvm' in pkg.__name__:
if not ek.has_llvm():
pytest.skip('LLVM mode is unsupported')
def m(request):
gc.collect()
gc.collect()
if 'cuda' in request.param.__name__:
if not ek.has_cuda():
pytest.skip('CUDA mode is unsupported')
else:
if not ek.has_llvm():
pytest.skip('LLVM mode is unsupported')
yield request.param
gc.collect()
gc.collect()
def get_class(name):
"""Resolve a package+class name into the corresponding type"""
if 'cuda' in name:
if not ek.has_cuda():
pytest.skip('CUDA mode is unsupported')
elif 'llvm' in name:
if not ek.has_llvm():
pytest.skip('LLVM mode is unsupported')
name = name.split('.')
value = __import__(".".join(name[:-1]))
for item in name[1:]:
value = getattr(value, item)
return value