torch.mps

此包提供了一个接口，用于在 Python 中访问 MPS（Metal Performance Shaders）后端。Metal 是 Apple 用于对金属 GPU（图形处理单元）进行编程的 API。使用 MPS 意味着可以通过在金属 GPU 上运行工作来实现更高的性能。有关更多详细信息，请参见 https://developer.apple.com/documentation/metalperformanceshaders。

同步	等待 MPS 设备上所有流中的所有内核完成。
获取随机数生成器状态	将随机数生成器状态作为 ByteTensor 返回。
设置随机数生成器状态	设置随机数生成器状态。
手动设置种子	设置生成随机数的种子。
种子	将生成随机数的种子设置为随机数。
清空缓存	释放缓存分配器当前持有的所有未占用的缓存内存，以便这些内存可以在其他 GPU 应用程序中使用。
设置每个进程的内存比例	设置内存比例，以限制 MPS 设备上进程的内存分配。
当前已分配内存	返回张量在字节中占用的当前 GPU 内存。
驱动程序已分配内存	返回 Metal 驱动程序为进程在字节中分配的总 GPU 内存。

MPS 分析器

分析器.开始	从 MPS 后端开始 OS Signpost 跟踪。
分析器.停止	停止从 MPS 后端生成 OS Signpost 跟踪。
分析器.分析	上下文管理器，用于启用从 MPS 后端生成 OS Signpost 跟踪。

MPS 事件

事件.事件

MPS 事件的包装器。