Hello, I’d like to dynamically call a function based on an integer (fct_selector here). I have several Python modules, each has a different compute function. FCT_MODULES = [ a_first_module, a_second_module, … ] @njit def dispatch(fct_selector: int) → Any: return FCT_MODULES[selector].compu…

You can store first-class jitted functions into a typed list and then access the list with an integer index.

Thank you for your response. I am trying this : COMPUTE_DOMAINS_FUNCTIONS = List.empty_list(numba.typeof(compute_domains_affine_eq)) COMPUTE_DOMAINS_FUNCTIONS.append(compute_domains_affine_eq) COMPUTE_DOMAINS_FUNCTIONS.append(compute_domains_affine_geq) COMPUTE_DOMAINS_FUNCTIONS.append(compute_…

I could not make it work with a global constant for the list. But I could make it work with: def init_compute_domains_functions() → List[Callable]: functions = List.empty_list(types.int64(int32[::1, :], int32[:]).as_type()) functions.append(compute_domains_affine_eq) functions.append(compute_do…

If the work being done in the function is small, the typed list overhead could be significant. There are other techniques, like storing pointers to the functions in a numpy array or simply using an if-else construct, which may well be fastest. Note that first-class functions return zero-initialize…

Hello, Thank you for your answer. I was using the if-else already, I’ll give the numpy array of function pointers a try. Cheers Yan

[image] nelson2005: There are other techniques, like storing pointers to the functions in a numpy arr Hello, How can I, from a jitted function, call another jitted function with a pointer ? Thank you, Yan

Thank you very much, this is what I was looking for. Of course my case was slightly more complex because I have an array of functions that I want to call based on a dynamic index but I was able to make it work ! Cheers, Yan

@DannyWeitekamp Unfortunately Any numba equivalent for casting a raw pointer to a StructRef, Dict, List etc? - #24 by DannyWeitekamp does not seem to work anymore with Numba 0.61.0rc2, any idea why ?

@Yangeorget Perhaps you can provide the error you get when trying this example to help locate the source of the issue?

How do I dynamically call a function?

Numba Support: How do I do ...?

DannyWeitekamp September 12, 2024, 7:10pm 8

See this post

TLDR. I usually use a custom intrinsic, not sure if the devs have added a standard way since that post.

Topic		Replies	Views
Any numba equivalent for casting a raw pointer to a StructRef, Dict, List etc? Support: How do I do ...?	29	4233	January 25, 2023
How to specify the type signature of funtion pointers to jitclass member function Support: How do I do ...?	10	970	October 9, 2023
Typed list of jitted functions in jitclass Support: How do I do ...?	17	3940	November 30, 2023
How do I make a jitclass with a jitted function member? Support: How do I do ...?	1	869	January 13, 2021
Invoke pybind from cfunc Support: How do I do ...?	7	141	March 6, 2025

How do I dynamically call a function?

Related topics