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 狀態是否已初始化。
memory_stats	返回給定裝置的 MTIA 記憶體分配器統計資訊字典。
get_device_capability	以 (主要版本, 次要版本) 的 tuple 形式返回給定裝置的功能。
empty_cache	清空 MTIA 裝置快取。
set_device	設定目前的裝置。
set_stream	設定目前的 stream。這是一個用來設定 stream 的 wrapper API。
stream	包裝 Context-manager StreamContext，用於選擇給定的 stream。
synchronize	等待 MTIA 裝置上所有 stream 中的所有工作完成。
device	Context-manager，用於更改選定的裝置。
set_rng_state	設定亂數產生器的狀態。
get_rng_state	以 ByteTensor 形式返回亂數產生器的狀態。
DeferredMtiaCallError

Streams 和 events

Event	查詢和記錄 Stream 狀態，以識別或控制跨 Stream 的相依性並測量時序。
Stream	一個依序佇列，以先進先出 (FIFO) 的順序非同步執行相應的任務。