Handle hiding the @jit_kernel

iree-org / iree-jax

Apache License 2.0

48 stars 19 forks source link

Handle hiding the @jit_kernel #22

Open rsuderman opened 2 years ago

rsuderman commented 2 years ago

Address comment linked below. jit_kernel is more an implementation detail and should not be exposed to the user. See about excluding it from import / reworking to be non-visible.

https://github.com/iree-org/iree-jax/blob/main/iree/jax/builtins.py#L46