Skip to content
Snippets Groups Projects
Unverified Commit 639aff69 authored by Clemens Giuliani's avatar Clemens Giuliani Committed by GitHub
Browse files

jit a few internal functions to make them work with global device arrays (#1507) 

i.e. arrays which are `isinstance(x, jax.Array) and not x.is_fully_addressable`.
parent dfb79488
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment