← Back to events
ActiveAIRelease

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

Photo: PyTorch Releases

What happened

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

Summary assembled by rule from the sources below

Why it's spreading

Sources