torch.compiler API 参考¶
有关 torch.compiler
的快速概述,请参阅 torch.compiler。
有关此函数的参数详细信息,请参阅 |
|
此函数清除所有编译缓存并将系统恢复到初始状态。 |
|
告诉编译器前端 (Dynamo) 跳过函数的符号内省,而是在遇到函数时直接将其写入图。 |
|
为函数注册一个 polyfill 处理程序,通常是来自 C 扩展的 C 函数,以便在图中内联原始函数时,用它来代替原始函数。 |
|
此函数用于将函数 fn 标记为具有常量结果。 |
|
返回可以传递给 torch.compile(..., backend="name") 的有效字符串。 |
|
此函数提供了一个装饰器来禁用函数上的编译。它还提供了递归禁用被调用函数的选项 |
|
设置编译器的当前立场。 |
|
指示新的推理或训练迭代即将开始。 |
|
指示图是否作为 torch.compile() 或 torch.export() 的一部分执行/跟踪。 |
|
指示是否通过 TorchDynamo 跟踪图。 |