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)