You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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.
The text was updated successfully, but these errors were encountered: