chainer.backend.ChainerxDevice

class chainer.backend.ChainerxDevice(device: chainerx.Device)

Device for ChainerX backend

Methods

__enter__()

A dummy definition that simply raises RuntimeError.

chainer.using_device() should be used instead.

__exit__(exc_type, exc_value, traceback)

A dummy definition that should never be called.

create_context()

Returns a context manager in which the device is made current.

See also

chainer.using_device() calls this method internally.

static from_array(array)

static from_fallback_device(device)

Returns a ChainerxDevice corresponding to the fallback device.

See also

fallback_device

is_array_supported(array)

Returns if the specified array is compatible with the device. :param array: An array to be checked :type array: N-dimensional array

Returns

True if the array is compatible with the device. Otherwise False is returned.

send(arrays)

Transfers given arrays to the device.

Parameters

arrays – Array or arrays of NumPy, CuPy, or ChainerX.

Returns

Transferred arrays.

send_array(array)

use()

Makes the device current in the current thread.

__eq__(other)

Return self==value.

__ne__(other)

Return self!=value.

__lt__(value, /)

Return self<value.

__le__(value, /)

Return self<=value.

__gt__(value, /)

Return self>value.

__ge__(value, /)

Return self>=value.

Attributes

fallback_device

Fallback device.

A fallback device is either a CpuDevice or a GpuDevice which shares the same physical device with the original ChainerX device.

For example, the fallback device of native:0 ChainerX device is CpuDevice. The fallback device of cuda:1 ChainerX device is GpuDevice with device ID 1.

name

supported_array_types = (<class 'chainerx.ndarray'>,)