Opt-in unified cuBLAS + cuBLASLt workspaces (#151163)
Summary:
opt-in version of https://github.com/pytorch/pytorch/pull/145130 as there was a lack of repro for the 70% forward issue
`TORCH_CUBLASLT_UNIFIED_WORKSPACE=1`
izaitsevfb could you comment if it was repeatable per every forward pass, on startup, or something else?
X-link: https://github.com/pytorch/pytorch/pull/151163
Approved by: https://github.com/ngimel
Reviewed By: ZainRizvi
Differential Revision: D73519132
fbshipit-source-id: cf3f786893a3b64e742411a1c027b375ed9103c9
Author
generatedunixname499836121
