SuperAPI AI 工具集

pytorch-knowledge-patch

通用与其它

查看项目

更新于 10-16 01:35

技能介绍

该技能用于在修改 PyTorch 代码前查阅与当前工作匹配的参考文件。基线为 PyTorch 2.3,覆盖 2.4.0–2.11.0 及若干主题批次,资料采集时间标注为 2026 年 7 月 12 日。升级既有项目时建议先看迁移与安全,再按需加载编译器、导出、分布式或平台参考;被标为 API-unstable 或实验性的接口视为可能变化。

参考索引包括:migrations-and-security.md(安全加载检查点、仓库信任、TorchScript、export/ONNX/PT2E 移除、MAGMA 与 varlen_attn 迁移);compiler-and-attention.md(torch.compile、Dynamo、Inductor、FlexAttention、变长注意力、缓存、确定性、编译器调试);export-aot-and-custom-ops.md(torch.export、动态形状、控制流、Core ATen、ONNX、AOTInductor 打包、Triton 与 out 变体);distributed-and-checkpointing.md(FSDP2、DTensor、DCP、XCCL、上下文并行、函数式集合通信、Symmetric Memory、float8);platforms-extensions-and-runtime.md(wheel 与 ABI、CUDA/ROCm/XPU/MPS、C++/SYCL 扩展、量化、流、autograd、profiler)。

迁移要点:自 2.6 起 torch.load() 默认 weights_only=True,可信对象建议显式白名单,仅可信旧 pickle 才用 weights_only=False;TorchScript 自 2.10 弃用,新导出用 torch.export,export_for_training() 在 2.11 移除;pt2e 量化包在 2.11 移除,改用 TorchAO;varlen_attn 可选参数改为仅关键字,用 window_size 替代 is_causal。分布式方面需先初始化进程组再构造 DeviceMesh,并移除 enable_symm_mem_for_group 调用。二进制选择上 2.11 在 Linux x86_64/aarch64 的 PyPI 默认为 CUDA 13.0,旧架构需显式选择 CUDA 12.6 等构建。其余编译器、FlexAttention、FSDP2 与导出细节按参考文件展开。官方未提供的具体版本兼容矩阵、性能数据与收费信息暂未提供。