torch.mtia¶

MTIA 后端是在树外实现的，这里只定义了接口。

此软件包提供了一个用于在 Python 中访问 MTIA 后端的接口

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

流和事件¶