Viernes 23 de octubre de 2026 ·
PyTorch sustituye en MPS el max_pool2d de MPSGraph por kernels Metal
Útil para
Un commit de PyTorch (#201756) reemplaza en el backend MPS la pasada hacia delante de max_pool2d de MPSGraph por kernels Metal propios, con especializaciones para 2D con tensores contiguos y channels_last. Los kernels de pooling hacia delante (1D, 2D, 3D y adaptativo) ahora propagan NaN como en CPU, y se soporta indexado de 64 bits. La pasada hacia atrás de 2D sigue usando MPSGraph y queda para una mejora posterior.
Qué significa
Quien ejecuta modelos de PyTorch en Macs con chips de Apple obtiene el pooling hacia delante en Metal y el mismo tratamiento de NaN que en CPU, aunque el backward sigue en MPSGraph.
Fuentes originales
Resumen de Hoy en IA. Pulsa una fuente para leer la noticia completa en su idioma original.
- GitHub pytorch/pytorchLeer la noticia completa en GitHub pytorch/pytorch
También ese día
Todas las noticias de IA del viernes 23 de octubre de 2026 ->