torch.compiler API 参考

有关 torch.compiler 的简要概述，请参阅 torch.compiler.

compile	有关此函数的参数的详细信息，请参阅 torch.compile().
reset	此函数清除所有编译缓存并将系统恢复到其初始状态。
allow_in_graph	告诉编译器前端 (Dynamo) 跳过函数的符号自省，而是直接将其写入遇到时遇到的图。
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 跟踪。