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 状态是否已初始化。
memory_stats	返回给定设备的MTIA内存分配器统计信息字典。
get_device_capability	返回给定设备的能力作为一个元组 (主要版本，次要版本)。
empty_cache	清除 MTIA 设备缓存。
set_device	设置当前设备。
set_stream	设置当前流。这是一个用于设置流的包装 API。
stream	围绕选择给定流的上下文管理器 StreamContext 进行包装。
synchronize	等待 MTIA 设备上所有流中的所有作业完成。
device	上下文管理器，用于更改选定的设备。
set_rng_state	设置随机数生成器的状态。
get_rng_state	返回随机数生成器的状态作为字节张量。
DeferredMtiaCallError

流和事件

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