torch.mtia¶

The MTIA backend is implemented out of the tree, only interfaces are be defined here.

This package enables an interface for accessing MTIA backend in python

`StreamContext`	Context-manager that selects a given stream.
`current_device`	Return the index of a currently selected device.
`current_stream`	Return the currently selected `Stream` for a given device.
`default_stream`	Return the default `Stream` for a given device.
`device_count`	Return the number of MTIA devices available.
`init`
`is_available`	Return true if MTIA device is available
`is_initialized`	Return whether PyTorch's MTIA state has been initialized.
`set_stream`	Set the current stream.This is a wrapper API to set the stream.
`stream`	Wrap around the Context-manager StreamContext that selects a given stream.
`synchronize`	Waits for all jobs in all streams on a MTIA device to complete.
`device`	Context-manager that changes the selected device.
`DeferredMtiaCallError`

Streams and events¶