pytorch
de9c82f4 - [Meta] Register aten.pixel_shuffle.default for meta (#91605)

Commit
1 year ago
[Meta] Register aten.pixel_shuffle.default for meta (#91605) **Summary** Fixes #91551 `aten.pixel_shuffle.default` is not registered for meta and it always generates contiguous (channels-first) layout of outputs. It can be reproduced by `torch.compile` (as described in the issue #91551) and running in FakeTensorMode. **Test plan** python test/inductor/test_torchinductor.py -k test_pixel_shuffle_channels_last python test/test_proxy_tensor.py Pull Request resolved: https://github.com/pytorch/pytorch/pull/91605 Approved by: https://github.com/jgong5, https://github.com/mingfeima, https://github.com/anijain2305
Author
Committer
Parents
Loading