Module: Chainer::Device

Defined in:
lib/chainer/device.rb

Class Method Summary collapse

Class Method Details

.change_default(device_spec) ⇒ Object

Changes default device

Parameters:

  • device_spec (Object)

See Also:



23
24
25
26
 
# File 'lib/chainer/device.rb', line 23

def change_default(device_spec)
  @default = create(device_spec)
  @default.use
end

.create(device_spec) ⇒ Chainer::AbstractDevice

Creates device

Parameters:

  • device_spec (Integer or Chainer::AbstractDevice) —

    Device specifier. Negative integer indicates CPU. 0 or positive integer indicates GPU. If a device object is given, itself is returned.

Returns:



9
10
11
12
13
14
15
16
 
# File 'lib/chainer/device.rb', line 9

def create(device_spec)
  return device_spec if device_spec.kind_of?(AbstractDevice)
  if device_spec.kind_of?(Integer)
    return CpuDevice.new if device_spec < 0
    return GpuDevice.new(device_spec)
  end
  raise "Invalid device_spec: #{device_spec}"
end

.default ⇒ Chainer::AbstractDevice

Gets default device

Returns:



32
33
34
 
# File 'lib/chainer/device.rb', line 32

def default
  @default ||= CpuDevice.new
end