adjust conditions to enable jit decomposition pass only for GPU device (#73637)
Summary:
Signed-off-by: Feng Yuan <feng1.yuan@intel.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/73637
Reviewed By: cpuhrsch
Differential Revision: D34931641
Pulled By: davidberard98
fbshipit-source-id: 849ace475eb9038912e902462f0eb2cdebc14ada
(cherry picked from commit ae61c4fba20345c0122f24e02c661b0da29f0ee4)