def test_from_array(self, backend_config):
with cuda.Device(backend_config.cuda_device):
arr = cuda.ndarray((), numpy.float32)
# Test precondition check
assert arr.device.id == backend_config.cuda_device
device = backend.GpuDevice.from_array(arr)
self.check_device(device, backend_config)
assert device == backend.GpuDevice.from_device_id(
backend_config.cuda_device)
def test_from_array(self, backend_config):
with cuda.Device(backend_config.cuda_device):
arr = cuda.ndarray((), numpy.float32)
# Test precondition check
assert arr.device.id == backend_config.cuda_device
device = backend.GpuDevice.from_array(arr)
assert isinstance(device, backend.GpuDevice)
assert device == backend.GpuDevice.from_device_id(
backend_config.cuda_device)
def test_from_array(self, backend_config):
with cuda.Device(backend_config.cuda_device):
arr = cuda.ndarray((), numpy.float32)
# Test precondition check
assert arr.device.id == backend_config.cuda_device
device = backend.GpuDevice.from_array(arr)
assert isinstance(device, backend.GpuDevice)
assert (device
== chainer.get_device((cuda.cupy, backend_config.cuda_device)))
def test_get_device_from_array(self, backend_config):
with cuda.Device(backend_config.cuda_device):
arr = cuda.ndarray((), numpy.float32)
# Test precondition check
assert arr.device.id == backend_config.cuda_device
expected_device = backend_config.device
device = backend.GpuDevice.from_array(arr)
assert device == expected_device
device = backend.get_device_from_array(arr)
assert device == expected_device
def test_get_device_from_array(self, backend_config):
with cuda.Device(backend_config.cuda_device):
arr = cuda.ndarray((), numpy.float32)
# Test precondition check
assert arr.device.id == backend_config.cuda_device
expected_device = backend_config.device
device = backend.GpuDevice.from_array(arr)
assert device == expected_device
device = backend.get_device_from_array(arr)
assert device == expected_device
def test_array_gpu(self):
self._check_array(cuda.ndarray([1, 2]), 'constant array')
def test_array_gpu(self):
self._check_array(cuda.ndarray([1, 2]), 'constant array')