#49390 [Perf] Raise Blackwell CUDA graph capture default to 1024
原始 PR · 作者 LucasWilkinson · 合并时间 2026-08-08 03:18
Blackwell 默认 CUDA 图捕获上限提至 1024
该 PR 值得精读,因为它展示了如何通过调整默认 CUDA 图捕获上限来获取大幅性能提升,并体现了平台差异化默认策略的设计思路。建议重点关注 `_set_cudagraph_sizes` 中平台判断与约束的结合方式,以及测试对 `current_platform` 的 mock 技巧。后续应跟踪 B200 验证结果,并评估是否需要在其他平台(如消费级 Blackwell)做类似调整。
参与讨论