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: driazati
Differential Revision: D32175958
Pulled By: Gamrix
fbshipit-source-id: 26a9ef41e10a171be6a8779a4e6014e2e7e3f2c1
