pytorch

9da5646c - Add device logic handling for functions which allow scalar inputs as tensors (#86149)

Add device logic handling for functions which allow scalar inputs as tensors (#86149) Some functions allow scalars as tensor inputs. Add handling for them in device logic. Fix for https://github.com/pytorch/torchdynamo/issues/1445 Pull Request resolved: https://github.com/pytorch/pytorch/pull/86149 Approved by: https://github.com/ezyang, https://github.com/bdhirsh