torch.compiler API 参考

有关 torch.compiler 的快速概览，请参阅 torch.compiler。

编译	有关此函数的参数详情，请参阅 torch.compile()。
重置	此函数清除所有编译缓存并恢复系统到其初始状态。
allow_in_graph	自定义编译时将包含在生成图中的函数。
假设常量结果	此函数用于将函数 fn 标记为具有常量结果。
list_backends	返回可以传递给 torch.compile(..., backend="name") 的有效字符串。
disable	此函数提供了一个装饰器和一个上下文管理器，用于在函数上禁用编译。它还提供了递归禁用被调用函数的选项。
cudagraph_mark_step_begin	表示一个新的推理或训练迭代即将开始。
is_compiling	指示图是否作为 torch.compile() 或 torch.export() 的一部分执行/追踪。
is_dynamo_compiling	指示图是否通过 TorchDynamo 进行追踪。