PyGraph: Robust Compiler Support for CUDA Graphs in PyTorch
针对 PyTorch 中 CUDA Graph 部署困难的编译器框架。提出三项优化：CGCT（自动代码变换使 ML 程序兼容 CUDA Graph）、PI（参数间接寻址将数据拷贝转为指针拷贝，最高减少 99% 拷贝量）、SCG（基于成本效益分析的有选择部署）。在 25 个 ML 工作负载上，PyGraph 比 PyTorch2-CG 平均提速 29%，最高 3.36×，且在分布式 Tensor Parallelism 下表现更强。