torch.compiler API 参考
关于 torch.compiler
的快速概述,请参见 torch.compiler。
关于此函数参数的详细信息,请参见 |
|
此函数会清除所有编译缓存,并将系统恢复到初始状态。 |
|
告诉编译前端(Dynamo),在遇到该函数时,跳过对其的符号分析,直接将函数写入图中。 |
|
为一个函数注册一个垫片处理程序,通常这个函数是C扩展中的C函数。当在图中内联原始函数时,使用该处理程序来替代原始函数。 |
|
此函数用于标记函数 fn 具有恒定的结果。 |
|
返回可用于传递给 torch.compile(..., backend="name") 的有效字符串。 |
|
该函数提供了一个装饰器和一个上下文管理器来禁用函数的编译,并且还可以选择递归地禁用被调用的函数。 |
|
表示新一轮的推理或训练即将开始。 |
|
表示图是否作为 torch.compile() 或 torch.export() 的一部分进行执行或追踪。 |
|
表示图是否通过TorchDynamo进行跟踪。 |