快捷方式

torch.compiler API 参考

有关 torch.compiler 的快速概述,请参阅 torch.compiler

compile

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

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

指示是否正在导出。

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

查找开发资源并获取问题解答

查看资源