trunk/c262fc64bd3a40834366246d22e9e7575f219a16: [CUDA] Capture Python launch stacks for CUDA graphs (#198419)
- PyTorch 近 90 天出现 1457 次
- PyTorch 近 90 天第 1438 次发版
- 上一次:同一天稍早 · ciflow/torchtitan/198536: [DTensor] Migrate unbind sharding propagation to single-dim strategy
发生了什么
Add record_py_stacks to CUDA graph annotation_config and expose live stack lookup and gzip JSON export. CUPTI node-creation callbacks record bounded user launch stacks, sharing strings within a capture. The graph-owned registry from the preceding commit handles rekeying and cleanup without scanning other live graphs; only source-less aliases need additional stack handling here. By default, omit framework and generat…
摘要按规则整理自下方来源原文