pytorch
93e70c59 - [FakeTensor] Use the device of the meta tensor for fallback kernel (#80193)

Commit
2 years ago
[FakeTensor] Use the device of the meta tensor for fallback kernel (#80193) Otherwise, I would run into issues such as fp16 not supported for Conv Cpu, or cudnn_rnn not implemented for cpu. Pull Request resolved: https://github.com/pytorch/pytorch/pull/80193 Approved by: https://github.com/Chillee
Author
Committer
Parents
Loading