def to_gpu( self, device=None, # type: tp.Optional[types.CudaDeviceSpec] ): # type: (...) -> 'DeviceResident' """Copies parameter variables and persistent values to GPU. This method does not handle non-registered attributes. If some of such attributes must be copied to GPU, the link implementation must override :meth:`Link.to_device` to do so. Args: device: Target device specifier. If omitted, the current device is used. Returns: self """ cuda.check_cuda_available() cuda_device = cuda._get_device_or_current(device) device = chainer.backends.cuda.GpuDevice(cuda_device) visitor = _ToDeviceVisitor( device, entry_method_info=('to_gpu', {'device': device.device}), skip_between_cupy_devices=True) self.__to_device(visitor) return self
def to_gpu( self, device=None, # type: tp.Optional[types.CudaDeviceSpec] ): # type: (...) -> 'DeviceResident' """Copies parameter variables and persistent values to GPU. This method does not handle non-registered attributes. If some of such attributes must be copied to GPU, the link implementation must override :meth:`~DeviceResident.device_resident_accept` to do so. .. warning:: This method does not transfer the parameters if they are already on GPU. Use ``to_device`` to perform inter-GPU transfer. Args: device: Target device specifier. If omitted, the current device is used. Returns: self """ cuda.check_cuda_available() cuda_device = cuda._get_device_or_current(device) device = chainer.backends.cuda.GpuDevice(cuda_device) visitor = _ToDeviceVisitor(device, entry_method_info=('to_gpu', { 'device': device.device }), skip_between_cupy_devices=True, starting_device_resident=self) self.__to_device(visitor) return self
def to_gpu(self, device=None): """Copies parameter variables and persistent values to GPU. This method does not handle non-registered attributes. If some of such attributes must be copied to GPU, the link implementation must override :meth:`Link.to_device` to do so. Args: device: Target device specifier. If omitted, the current device is used. Returns: self """ cuda.check_cuda_available() return self._to_device(cuda._get_device_or_current(device), skip_between_cupy_devices=True)
def to_gpu(self, device=None): """Copies parameter variables and persistent values to GPU. This method does not handle non-registered attributes. If some of such attributes must be copied to GPU, the link implementation must override :meth:`Link.to_device` to do so. Args: device: Target device specifier. If omitted, the current device is used. Returns: self """ cuda.check_cuda_available() return self._to_device( cuda._get_device_or_current(device), skip_between_cupy_devices=True)