Simon Fan c8205cb354 [autograd] match 0-dim gradients device type regardless of subclassness (#160165) ...

Not sure if there some subclasses where the outer.dim() == 0 but you wouldn't want to move it?

FIXES https://github.com/pytorch/pytorch/issues/160084

Pull Request resolved: https://github.com/pytorch/pytorch/pull/160165
Approved by: https://github.com/ezyang, https://github.com/albanD

2025-08-11 17:57:32 +00:00

99 KiB

Raw Blame History

View Raw