torch.mps.profiler.profile¶ torch.mps.profiler.profile(mode='interval', wait_until_completed=False)[源代码]¶ 上下文管理器，用于从MPS后端启用生成OS Signpost跟踪。 Parameters 模式 (字符串) – OS Signpost 跟踪模式可以是“间隔”、“事件”， 或者两者“间隔,事件”。 间隔模式跟踪操作的执行持续时间， 而事件模式标记执行的完成。 有关更多信息，请参阅文档 记录性能数据。 wait_until_completed (bool) – 等待直到MPS流完成执行每个编码的GPU操作。这有助于在跟踪的时间线上生成单个调度。 请注意，启用此选项会对性能产生负面影响。
