Miércoles 14 de octubre de 2026

PyTorch corrige el almacenamiento sobredimensionado en la pérdida Smooth L1 reducida

El PR #200894 de pytorch/pytorch corrige que F.smooth_l1_loss reservara el buffer elementwise completo al usar reducciones mean o sum. Siguiendo el enfoque ya aplicado a mse_loss en #199823, la salida pasa a ser un escalar. En la prueba con entradas FP32 de forma (4, 5, 6), los resultados reducidos pasan de 480 a 4 bytes. Los tests son solo CPU; CUDA, XPU y MPS no se ejecutaron.

Fuentes

• GitHub pytorch/pytorchpytorch/pytorch: trunk/23047d0839b1b5ef8254539f46f191c44659af97: Fix oversized storage for reduced Smooth L1 loss (#200894)