2 Commits

Author	SHA1	Message	Date
Jane Xu	a41f00ed70	[optim][sgd] group tensors in foreach to maximize perf (#92338) ... Make foreach faster for SGD Pull Request resolved: https://github.com/pytorch/pytorch/pull/92338 Approved by: https://github.com/albanD	2023-01-18 04:02:41 +00:00
Jane Xu	ed7885c254	[utils][foreach] Add group tensor by device and dtype util (#92014) ... Add util that will be commonly used throughout optim Pull Request resolved: https://github.com/pytorch/pytorch/pull/92014 Approved by: https://github.com/albanD	2023-01-11 23:37:20 +00:00