快捷方式

torch.compiler API 参考

要快速了解 torch.compiler,请参阅 torch.compiler

编译

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

重置

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

允许在图中

告诉编译器前端 (Dynamo) 跳过函数的符号内省,而是在遇到时将其直接写入图中。

替换在图中

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

假设结果为常量

此函数用于将函数 fn 标记为具有常量结果。

列出后端

返回可以传递给 torch.compile(..., backend="name") 的有效字符串。

禁用

此函数提供装饰器和上下文管理器来禁用函数的编译,它还提供递归禁用被调用函数的选项。

cudagraph_mark_step_begin

指示推理或训练的新迭代即将开始。

正在编译

指示图是否作为 torch.compile() 或 torch.export() 的一部分执行/跟踪。

is_dynamo_compiling

指示图是否通过 TorchDynamo 跟踪。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源