Skip to content

Improve Numba compile time #179

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.

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

Open
aseyboldt opened this issue Jan 6, 2023 · 0 comments
Open

Improve Numba compile time #179

aseyboldt opened this issue Jan 6, 2023 · 0 comments

Comments

@aseyboldt
Copy link
Member

Compile times with numba are often relatively long right now. The mean reason is that numba compiles every function we use in the graph separately. (see here for a detailed description).

There are things we can do to improve this in the meantime however:

  • Reuse output functions of funcify_* as much as possible. This is good anyway, because it also saves llvm some work.
  • Make sure we pass appropriate flags to numba functions, like no_cpython_wrapper=True.
  • Provide signatures to as many njit functions as we can. We usually have the type information available anyway, so why not use it, make things more robust and save numba some trouble to infer it? A problem is though that we don't have alignment info as part of our types, not sure how to handle this...
@michaelosthege michaelosthege changed the title Improve compile time Improve Numba compile time Jan 7, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants