pytorch
ac9b305a - Back out "cherry-picking autodiff support for gather/index_select (#93333)" (#95565)

Commit
1 year ago
Back out "cherry-picking autodiff support for gather/index_select (#93333)" (#95565) Summary: A bisect blamed #93333 for GPU memory leakage. This diff backs it out. Test Plan: Monitor max GPU memory usage to see if there's a leak. Reviewed By: hyuen, yinbinm Differential Revision: D43511893 Pull Request resolved: https://github.com/pytorch/pytorch/pull/95565 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading