PyTorch (rocm-mi300): un cero se mantiene exacto al pasar por un redondeo de baja precisión
Redondear un cero constante a float16 o bfloat16 sigue dando cero exacto, así que el análisis de máscaras de cero de PyTorch puede conservarlo. Un prologue cuya única operación de baja precisión es un redondeo mantiene sus ceros enmascarados y no añade un tl.where extra, igual que el código generado en la versión 2.14. Los demás valores siguen considerándose desconocidos, y el test de extremo a extremo ahora verifica el código generado para la máscara.
Qué significa
Solo afecta a quien usa PyTorch con GPUs AMD MI300 y compila kernels con prologues de baja precisión; el código generado vuelve a coincidir con el de la versión 2.14.
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