torch.mps

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

device_count	返回可用 MPS 设备的数量。
synchronize	等待 MPS 设备上所有流中的所有内核完成。
get_rng_state	将随机数生成器状态作为 ByteTensor 返回。
set_rng_state	设置随机数生成器状态。
manual_seed	设置生成随机数的种子。
seed	将生成随机数的种子设置为随机数。
empty_cache	释放缓存分配器当前持有的所有未使用的缓存内存，以便这些内存可以在其他 GPU 应用程序中使用。
set_per_process_memory_fraction	设置 MPS 设备上限制进程内存分配的内存比例。
current_allocated_memory	返回以字节为单位的张量当前占用的 GPU 内存。
driver_allocated_memory	以字节为单位返回 Metal 驱动程序为进程分配的总 GPU 内存。

MPS Profiler

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

MPS Event

event.Event

MPS 事件的包装器。