Viernes 16 de octubre de 2026 ·

pytorch/pytorch: [MPS] Usar los kernels atómicos histc e index_add_ para bincount (#201045)

Relevancia 38 de 100 Estados Unidos, California

Este cambio en PyTorch corrige la lentitud de bincount en MPS cuando casi todos los valores caen en un mismo bin. El bincount sin pesos pasa a usar el kernel histc, como en CUDA, y el ponderado usa index_add. Las pruebas muestran aceleraciones de hasta 12.718 veces en tensores de 4 millones de elementos, por ejemplo de 6.776 ms a 0,533 ms.

Qué significa

Quien ejecute torch.bincount en GPU de Apple con MPS verá mucho menos tiempo de cálculo en estos casos.

Fuentes originales

Resumen de Hoy en IA. Pulsa una fuente para leer la noticia completa en su idioma original.

También ese día

Todas las noticias de IA del viernes 16 de octubre de 2026 ->