Miércoles 14 de octubre de 2026
PyTorch mejora la caché de embedding bags N-bit con caché thread-local en CPU
Un commit en trunk de pytorch/pytorch hace que la implementación de embedding bags cuantizados para CPU (caffe2) use las cachés thread-local de los kernels N-bit de FBGEMM, eliminando la contención del lock de caché global en la ruta de lookup de producción. Mantiene compatibilidad con versiones antiguas de FBGEMM. Las pruebas de CPU aplicables pasan; los casos exclusivos de GPU no aplican.
Fuentes
- GitHub pytorch/pytorchpytorch/pytorch: trunk/296101d35a7ab06f3cc8a4c3ec70f4bb020d5a9e: [caffe2] Use thread-local cache for N-bit embedding bags (#200898)