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 内存分配器统计信息的字典。
set_device	设置当前设备。
set_stream	设置当前流。这是一个设置流的包装器 API。
stream	围绕选择给定流的上下文管理器 StreamContext 进行包装。
synchronize	等待 MTIA 设备上所有流中的所有作业完成。
device	更改所选设备的上下文管理器。
set_rng_state	设置随机数生成器状态。
get_rng_state	将随机数生成器状态作为 ByteTensor 返回。
DeferredMtiaCallError

流和事件

Event	查询和记录流状态，以识别或控制跨流的依赖关系并测量时间。
Stream	一个按顺序执行任务的异步队列，遵循先进先出 (FIFO) 顺序。