def check_device_spec_intel64(self, device_spec):
device = backend.get_device(device_spec)
assert isinstance(device, backend.Intel64Device)
assert device.xp is numpy
with backend.using_device(device_spec):
# TODO(niboshi): Test the Chainer default device
pass
def check_device_spec_intel64(self, device_spec):
device = backend.get_device(device_spec)
assert isinstance(device, backend.Intel64Device)
assert device.xp is numpy
with backend.using_device(device_spec):
# TODO(niboshi): Test the Chainer default device
pass
def check_device_spec_cupy(self, device_spec, expected_device_id):
device = backend.get_device(device_spec)
assert isinstance(device, backend.GpuDevice)
assert isinstance(device.device, cuda.Device)
assert device.xp is cuda.cupy
assert device.device.id == expected_device_id
with backend.using_device(device_spec):
# TODO(niboshi): Test the Chainer default device
assert cuda.Device() == cuda.Device(expected_device_id)
def check_device_spec_cupy(self, device_spec, expected_device_id):
device = backend.get_device(device_spec)
assert isinstance(device, backend.GpuDevice)
assert isinstance(device.device, cuda.Device)
assert device.xp is cuda.cupy
assert device.device.id == expected_device_id
with backend.using_device(device_spec):
# TODO(niboshi): Test the Chainer default device
assert cuda.Device() == cuda.Device(expected_device_id)
def check_device_spec_chainerx(self, device_spec, expected_device_name):
device = backend.get_device(device_spec)
assert isinstance(device, backend.ChainerxDevice)
assert device.xp is chainerx
assert isinstance(device.device, chainerx.Device)
assert device.device.name == expected_device_name
with backend.using_device(device_spec):
# TODO(niboshi): Test the Chainer default device
assert (chainerx.get_default_device() == chainerx.get_device(
expected_device_name))
def check_device_spec_chainerx(self, device_spec, expected_device_name):
device = backend.get_device(device_spec)
assert isinstance(device, backend.ChainerxDevice)
assert device.xp is chainerx
assert isinstance(device.device, chainerx.Device)
assert device.device.name == expected_device_name
with backend.using_device(device_spec):
# TODO(niboshi): Test the Chainer default device
assert (
chainerx.get_default_device()
== chainerx.get_device(expected_device_name))
def check_invalid(self, device_spec):
with pytest.raises(Exception):
backend.get_device(device_spec)
with pytest.raises(Exception):
backend.using_device(device_spec)
def check_invalid(self, device_spec):
with pytest.raises(Exception):
backend.get_device(device_spec)
with pytest.raises(Exception):
backend.using_device(device_spec)
