Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Adding compile_function as execute option. #536

Merged
merged 9 commits into from
Aug 7, 2024
Merged

Adding compile_function as execute option. #536

merged 9 commits into from
Aug 7, 2024

Commits on Aug 3, 2024

  1. Adding compile_function as execute option. 

    This allows users to use jit or aot compilation during the dag finalization process within executors. It should work straightforwardly on jax/numba style jit compilation. It's possible, but maybe ugly, to perform jax-aot-style compilation.
    @alxmrs
    alxmrs committed Aug 3, 2024
    Configuration menu
    Copy the full SHA
    e38a2df View commit details
    Browse the repository at this point in the history

  2. Add numba for compilation tests.

    @alxmrs
    alxmrs committed Aug 3, 2024
    Configuration menu
    Copy the full SHA
    b8577ea View commit details
    Browse the repository at this point in the history

  3. Singlequotes not needed.

    @alxmrs
    alxmrs committed Aug 3, 2024
    Configuration menu
    Copy the full SHA
    eec946a View commit details
    Browse the repository at this point in the history

  4. Update function doc.

    @alxmrs
    alxmrs committed Aug 3, 2024
    Configuration menu
    Copy the full SHA
    737111a View commit details
    Browse the repository at this point in the history

  5. Added another compile test for failure case.

    @alxmrs
    alxmrs committed Aug 3, 2024
    Configuration menu
    Copy the full SHA
    058ceb8 View commit details
    Browse the repository at this point in the history

  6. Added another test to ensure config was applied.

    @alxmrs
    alxmrs committed Aug 3, 2024
    Configuration menu
    Copy the full SHA
    2ec4fda View commit details
    Browse the repository at this point in the history

Commits on Aug 5, 2024

  1. Improve tests 

    - remove todo for new test
- use pytest conventions
    @alxmrs
    alxmrs authored Aug 5, 2024
    Configuration menu
    Copy the full SHA
    bf52e4d View commit details
    Browse the repository at this point in the history

  2. I don’t think numba jit works well on jax arrays.

    @alxmrs
    alxmrs authored Aug 5, 2024
    Configuration menu
    Copy the full SHA
    3543123 View commit details
    Browse the repository at this point in the history

Commits on Aug 6, 2024

  1. Update plan.py 

    Simplifying type to make mypy happy.
    @alxmrs
    alxmrs authored Aug 6, 2024
    Configuration menu
    Copy the full SHA
    9f0906a View commit details
    Browse the repository at this point in the history