torch.compiler API 参考文档 ¶

想快速了解 torch.compiler ，请参阅 torch.compiler。

compile	查看第 0#条以获取此函数参数的详细信息。
reset	此函数清除所有编译缓存并恢复系统到初始状态。
allow_in_graph	告诉编译器前端（Dynamo）在遇到函数时跳过符号性内省，而是直接将其写入图。
substitute_in_graph	注册一个 polyfill 处理器，用于在图内联原始函数时，用其代替原始函数，通常是一个来自 C 扩展的 C 函数。
assume_constant_result	此函数用于标记函数 fn 具有恒定结果。
list_backends	返回可以传递给 torch.compile(..., backend="name")的有效字符串。
disable	该函数提供了一个装饰器，用于禁用函数的编译。
set_stance	设置编译器的当前状态。
cudagraph_mark_step_begin	表示即将开始新的推理或训练迭代。
is_compiling	表示图是否作为 torch.compile()或 torch.export()的一部分执行/跟踪。
is_dynamo_compiling	表示是否通过 TorchDynamo 进行图追踪。
is_exporting	表示是否处于导出状态。