B200 实测:torch.compile 关一个开关,同一计算提速 4.53 倍
cfregly · x · 2026-10-26
开发者 cfregly 公开了一组 B200 上的 PyTorch 2.14.1+cu130 编译器微基准,指出一个容易被忽视的配置差异:同一个 64 个 ATen 算子的计算图,开关不同会导致编译器生成完全不同的 region 结构。
关键数据:
- False 档:生成 16 个 region,每个 4 个算子,每次调用 16 次 replay launch + 16 个 copy kernel
- True 档:只生成 2 个 region,每个 32 个算子,8 个重复块融合进一个 kernel,每次调用仅 2 次 replay launch + 2 个 copy kernel
- 稳态加速 4.53×;但前 10 个编译调用区间反而慢 33.2%(编译开销)
作者贴出了 benchmark 代码、分析器链接、24 次运行数据和可视化图表,并说明两档保留了相同的显式内层 graph break,region 框图只是实际编译输出的摘要。
所属事件:PyTorch 新开关在 B200 上实测同一计算提速 4.53 倍(2 条相关)→
「编程与Agent」频道最新
- Agent 自动支付落地了吗?Reddit 用户实测疑虑:卡单重试会重复扣款吗 — SpreadUsual4084 · 2026-10-26
- 用 2x2 分镜网格驯服 MiniMax 视频生成,复杂场景不再翻车 — Cheap_Credit_3957 · 2026-10-26
- Copilot Studio 接入 SAP S/4HANA:端到端用户身份传播方案详解 — ColdImportance2484 · 2026-10-26
- Base44 微调 Qwen 2B 做提示词自动补全,半天上线服务百万用户 — tristanbob · 2026-10-26
- Grok 盘点 Blender MCP 实战:一条提示词生成完整可玩足球游戏 — chongdashu · 2026-10-26
- 前前沿实验室外包者发布免费 AI 工程课程站:208 本手册+130 个交互构建 — iamsausi · 2026-10-26