torch.mtia#
创建日期:2023年7月11日 | 最后更新日期:2025年6月8日
MTIA 后端在树外实现,此处仅定义接口。
此包提供了一个用于在 python 中访问 MTIA 后端的接口
选择给定流的上下文管理器。 |
|
返回当前选定设备的索引。 |
|
返回给定设备当前选择的 |
|
返回给定设备的默认 |
|
返回可用 MTIA 设备的数量。 |
|
如果 MTIA 设备可用则返回 true |
|
返回 PyTorch 的 MTIA 状态是否已初始化。 |
|
返回给定设备的 MTIA 内存分配器统计信息的字典。 |
|
以 (主版本, 次版本) 元组形式返回给定设备的功能。 |
|
清空 MTIA 设备缓存。 |
|
在 MTIA 分配器上启用/禁用内存分析器 |
|
返回 MTIA 内存分配器历史的字典 |
|
向 MTIA 内存分配器附加一个内存不足观察器 |
|
设置当前设备。 |
|
设置当前流。这是一个用于设置流的包装 API。 |
|
包装选择给定流的上下文管理器 StreamContext。 |
|
等待 MTIA 设备上所有流中的所有作业完成。 |
|
更改选定设备的上下文管理器。 |
|
设置随机数生成器状态。 |
|
以 ByteTensor 形式返回随机数生成器状态。 |
|