jax
ad9f35ae - [PJRT:PLUGIN] Support both string and bytes as the input type of function name for register_custom_call_target in jax-cuda-plugin.

Commit

1 year ago

[PJRT:PLUGIN] Support both string and bytes as the input type of function name for register_custom_call_target in jax-cuda-plugin. PiperOrigin-RevId: 642639867

References

#21814 - [PJRT:PLUGIN] Support both string and bytes as the input type of function name for register_custom_call_target in jax-cuda-plugin.

Author

Jieying Luo

Committer

a-googler

Parents

4400ac45

jax ad9f35ae - [PJRT:PLUGIN] Support both string and bytes as the input type of function name for register_custom_call_target in jax-cuda-plugin.

jax
ad9f35ae - [PJRT:PLUGIN] Support both string and bytes as the input type of function name for register_custom_call_target in jax-cuda-plugin.