pytorch

2b7271c7 - Support cond and out_dtype for predispatch (#107941)

Support cond and out_dtype for predispatch (#107941) Summary: Title Test Plan: CI Differential Revision: D48675742 Pull Request resolved: https://github.com/pytorch/pytorch/pull/107941 Approved by: https://github.com/jerryzh168