trunk/717d7a288d82e60e92d7936cf41aaf3ec4b23c2d: torch/_native: route the chunked linear_cross_entropy ops (#195332)
- PyTorch 近 90 天出现 1219 次
- PyTorch 近 90 天第 1205 次发版
- 上一次:同一天稍早 · trunk/1579dafc271f285bc780fd69264e6a2d49cd342c: [CI] Enable ACL in ARM64 CUDA images (#197741)
发生了什么
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…
摘要按规则整理自下方来源原文