A function that is annotated with jax.jit() can run asynchronous and only when the result is requested it will block until the computation has finished.
Implementing this behaviour in JaCe is very difficult, since the source that is generated by DaCe will blcok until the execution has ended.
This means we have to rework how the entire SDFG<->Python interface works and patch the code generator.
A function that is annotated with
jax.jit()can run asynchronous and only when the result is requested it will block until the computation has finished.Implementing this behaviour in JaCe is very difficult, since the source that is generated by DaCe will blcok until the execution has ended.
This means we have to rework how the entire SDFG<->Python interface works and patch the code generator.