trunk/717d7a288d82e60e92d7936cf41aaf3ec4b23c2d: torch/_native: route the chunked linear_cross_entropy ops (#195332)
- PyTorch: 1219 events in the last 90 days
- PyTorch: 1205th Release in the last 90 days
- Previous: earlier the same day · trunk/1579dafc271f285bc780fd69264e6a2d49cd342c: [CI] Enable ACL in ARM64 CUDA images (#197741)
What happened
This adds the torch._native override for the two chunked ops together with its eligibility condition, while the implementation still delegates to eager. Keeping the routing in its own commit leaves the kernel commit about the kernel. linear_cross_entropy puts a whole fused loss behind one dispatcher symbol: an (N, F) x (F, C) logits matmul, a row-shifted softmax, the cross-entropy reduction, and three gradient matmu…
Summary assembled by rule from the sources below