benchmark
e759c474 - Use cuBLAS default workspace size in Lt (#153556)

Commit
301 days ago
Use cuBLAS default workspace size in Lt (#153556) Summary: Also enables unified workspaces by default for non-FBCODE use cases. Default Lt workspace size is also updated to match cuBLAS logic for default, including for Blackwell (SM 10.0) and GeForce Blackwell (SM 12.0). Recommended defaults are documented here: https://docs.nvidia.com/cuda/cublas/#cublassetworkspace X-link: https://github.com/pytorch/pytorch/pull/153556 Approved by: https://github.com/Skylion007, https://github.com/ngimel Reviewed By: izaitsevfb Differential Revision: D75387537 fbshipit-source-id: 7ec68ba7362c49a3cfc16fa7e46c0ba490a601c1
Author
generatedunixname499836121
Parents
Loading