Adding Custom Rules to Device Propagation (#66973)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66973
Test Plan: Imported from OSS
Reviewed By: albanD
Differential Revision: D32497549
Pulled By: Gamrix
fbshipit-source-id: 5732682c0b39709f76cf218490e5f5136c0d83f8
