torch.compiler API 参考

关于 torch.compiler 的快速概述,请参见 torch.compiler

compile

关于此函数参数的详细信息,请参见 torch.compile()

reset

此函数会清除所有编译缓存,并将系统恢复到初始状态。

allow_in_graph

告诉编译前端(Dynamo),在遇到该函数时,跳过对其的符号分析,直接将函数写入图中。

substitute_in_graph

为一个函数注册一个垫片处理程序,通常这个函数是C扩展中的C函数。当在图中内联原始函数时,使用该处理程序来替代原始函数。

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进行跟踪。

本页目录