Inheritance diagram for at::detail::CPUGuardImpl:

Public Member Functions
DeviceType	type () const override
	Return the type of device managed by this guard implementation.

Device	exchangeDevice (Device) const override
	Set the current device to Device, and return the previous Device.

Device	getDevice () const override
	Get the current device.

void	setDevice (Device) const override
	Set the current device to Device.

void	uncheckedSetDevice (Device d) const noexceptoverride
	Set the current device to Device, without checking for errors (so, e.g., this can be called from a destructor).

Stream	getStream (Device d) const noexceptoverride
	Get the current stream for a given device.

Stream	exchangeStream (Stream s) const noexceptoverride
	Set a stream to be the thread local current stream for its device. More...

DeviceIndex	deviceCount () const override
	Get the number of devices.

Public Member Functions inherited from c10::impl::DeviceGuardImplInterface
virtual	~DeviceGuardImplInterface ()=default
	Intended use of this class is to leak the DeviceGuardImpl at program end. More...

Detailed Description

Definition at line 9 of file CPUGuardImpl.h.

Member Function Documentation

Stream at::detail::CPUGuardImpl::exchangeStream ( Stream ) const

inlineoverridevirtualnoexcept

Set a stream to be the thread local current stream for its device.

Return the previous stream for that device. You are NOT required to set the current device to match the device of this stream.

Definition at line 33 of file CPUGuardImpl.h.

The documentation for this struct was generated from the following file: