def test_prefer_cached_array():
nn.reset_array_preference()
nn.prefer_cached_array(True)
ac2 = nn.array_classes()
check_cached_array_prefered(ac2)
if hasattr(nn.extensions, 'cuda'):
ac2 = nn.extensions.cuda.array_classes()
check_cached_array_prefered(ac2)
nn.prefer_cached_array(False)
ac2 = nn.array_classes()
check_cached_array_prefered(ac2, False)
if hasattr(nn.extensions, 'cuda'):
ac2 = nn.extensions.cuda.array_classes()
check_cached_array_prefered(ac2, False)
def test_prefer_cached_array():
nn.reset_array_preference()
nn.prefer_cached_array(True)
ac2 = nn.array_classes()
check_cached_array_prefered(ac2)
if hasattr(nn.extensions, 'cuda'):
ac2 = nn.extensions.cuda.array_classes()
check_cached_array_prefered(ac2)
nn.prefer_cached_array(False)
ac2 = nn.array_classes()
check_cached_array_prefered(ac2, False)
if hasattr(nn.extensions, 'cuda'):
ac2 = nn.extensions.cuda.array_classes()
check_cached_array_prefered(ac2, False)
def test_prefer_cached_array():
nn.reset_array_preference()
nn.prefer_cached_array(True)
ac2 = nn.array_classes()
check_cached_array_preferred(ac2)
try:
from nnabla_ext import cuda
except:
cuda = None
if cuda is not None:
ac2 = cuda.array_classes()
check_cached_array_preferred(ac2)
nn.prefer_cached_array(False)
ac2 = nn.array_classes()
check_cached_array_preferred(ac2, False)
if cuda is not None:
ac2 = cuda.array_classes()
check_cached_array_preferred(ac2, False)
def context(**kw):
"""CPU Context."""
return nn.Context('cpu', array_classes()[0], '', 'default')