Implementar ISA usando bucles y condicionales simples y numba

[astrojuanlu]

• Asignar a @AunSiro.

[astrojuanlu]

@AunSiro veo que hay algunos problemas con tu implementación usando funciones ventana así que me asigno yo esta parte, a ver si me da tiempo a intentarlo mañana.

[astrojuanlu]

Tengo ya una versión preliminar (ver dd4f29a) pero que no hace nada con los offsets de temperatura porque aún no hemos acordado cómo validar esos resultados (ver #8). Todos los tests pasan y el rendimiento es decente, pero me gustaría probar cuánto acelera numba el código.

[astrojuanlu]

Voy a dar por cerrada esta tarea porque, aunque escribí la implementación con numba en mente, la anotación está desactivada en versiones recientes (ver numba/numba#105) y no fui capaz de reducir el tiempo de ejecución. El código se queda en la rama juanlu-python para referencias futuras.

[astrojuanlu]

Reabriendo esto porque he hecho experimentos con numba y ya sé cómo exprimirlo al máximo. Volveré a probar a ver si consigo mejorar el rendimiento y repetir los benchmarks.

[astrojuanlu]

Parece que tenemos nuevo ganador 😉

Y en la segunda prueba:

$ python test/performance.py 
newlawrence running...
siro running...
juanlu running...
jnumba running...
alex running...
   size          juanlu            alex          jnumba            siro     newlawrence
      1        0.022 ms        0.273 ms        0.018 ms        1.976 ms        0.035 ms
     10        0.040 ms        0.346 ms        0.015 ms        4.659 ms        0.032 ms
    100        0.240 ms        0.384 ms        0.028 ms        5.033 ms        0.036 ms
   1000        2.288 ms        0.773 ms        0.213 ms        9.224 ms        0.080 ms
  10000       22.190 ms        4.798 ms        1.298 ms       43.707 ms        0.566 ms
 100000      230.619 ms       50.851 ms       13.048 ms      645.848 ms        5.034 ms

Código en la rama juanlu-numba.

[astrojuanlu]

La prueba de concepto ya está hecha, cerrando

[astrojuanlu]

Como bien apunta @newlawrence, la versión de C++ está compilada con OpenMP y he corrido el test en un Intel i7. Desactivando esto, queda la versión de numba el doble de rápida que la de C++. ¡Impresionante!

[astrojuanlu]

Reabro esta incidencia para recordarme que este fin de semana voy a reemplazar la implementación de la atmósfera estándar.

[astrojuanlu]

"Este fin de semana" 😅

[AlexS12]

@newlawrence hizo un gráfico de esto. Alguno lo tiene localizable? Puede pegarlo aquí por ejemplo?

He buscado en el twitter de aeropython y pybonacci, pero no aparece

[astrojuanlu]

¡Aquí está!

https://github.com/newlawrence/Pyaisa#the-roots

[AlexS12]

Gracias! qué veloz!