torch.compiler API 参考

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

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