pytorch

f2c85575 - Revert "Make `kl_div` a composite function. (#80334)"

Revert "Make `kl_div` a composite function. (#80334)" This reverts commit 828c787ea98da39eb786925eedcb8527aae07153. Reverted https://github.com/pytorch/pytorch/pull/80334 on behalf of https://github.com/ezyang due to doesn't work with xla