Inheritance diagram for c10::hip::HIPGuardImplMasqueradingAsCUDA:

Public Member Functions
	HIPGuardImplMasqueradingAsCUDA (DeviceType t)

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

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

Device	getDevice () const override
	Get the current device.

void	setDevice (Device d) 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...

Static Public Attributes
static constexpr DeviceType	static_type = DeviceType::CUDA

Detailed Description

Member Function Documentation

Stream c10::hip::HIPGuardImplMasqueradingAsCUDA::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.

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