torch.mps

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

同步	等待MPS设备上所有流中的所有内核完成。
获取随机数生成器状态	返回随机数生成器状态为 ByteTensor。
设置随机数生成器状态	设置随机数生成器的状态。
manual_seed	设置生成随机数的种子。
种子	设置生成随机数的种子为一个随机数。
empty_cache	释放缓存分配器当前持有的所有未占用的缓存内存，以便这些内存可以用于其他GPU应用程序。
set_per_process_memory_fraction	设置内存分数以限制MPS设备上的进程内存分配。
当前分配的内存	返回当前由张量占用的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

MPS事件的包装器。