trunk/63e6a716eb5225b0bab5661876de2d8111472c4d: Refactor CUDAGraph device support (#196625) (#196625)
- PyTorch 近 90 天出现 776 次
- PyTorch 近 90 天第 766 次发版
- 上一次:同一天稍早 · ciflow/trunk/196481
发生了什么
Summary: Very minor refactor. Custom CUDAGraphPolicy implementations can override _CUDAGRAPH_SUPPORTED_DEVICE_TYPES to enable graph capture for accelerator backends other than CUDA, while preserving CUDA as the default. Test Plan: Verified in the subsequent diffs. Differential Revision: D119430665 Pull Request resolved: #196625 Approved by: https://github.com/egienvalue
摘要按规则整理自下方来源原文