← 返回事件
持续讨论AI发版

trunk/63e6a716eb5225b0bab5661876de2d8111472c4d: Refactor CUDAGraph device support (#196625) (#196625)

图:PyTorch Releases

发生了什么

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

摘要按规则整理自下方来源原文

为什么在扩散

来源