Figuring out how to get everything done inside a CUDA graph capture has been my go-to PyTorch optimization.
CUDA Graph Capture as Primary PyTorch Optimization Strategy
By
–
By
–
Figuring out how to get everything done inside a CUDA graph capture has been my go-to PyTorch optimization.