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`

串流和事件¶