torch.mtia¶

MTIA后端是独立实现的，这里只定义接口。

此软件包提供了访问 MTIA 后端的接口，支持 Python。

`StreamContext`	上下文管理器，用于选择指定的流。
`current_device`	返回当前选定设备的索引。
`current_stream`	返回给定设备当前选择的 `Stream`。
`default_stream`	返回给定设备的默认 `Stream`。
`device_count`	返回可用的MTIA设备数量。
`init`
`is_available`	如果可用，则返回 MTIA 设备为真
`is_initialized`	返回 PyTorch 的 MTIA 状态是否已初始化。
`set_stream`	设置当前流。这是一个用于设置流的包装 API。
`stream`	围绕选择给定流的上下文管理器 StreamContext 进行包装。
`synchronize`	等待 MTIA 设备上所有流中的所有作业完成。
`device`	上下文管理器，用于更改选定的设备。
`DeferredMtiaCallError`

流和事件¶