jax
958ea15a - Enable command buffer support for buffer callbacks.

Commit

1 year ago

Enable command buffer support for buffer callbacks. This enables `command_buffer_compatible=True` for buffer callbacks on GPU. PiperOrigin-RevId: 755391950

References

#28510 - Enable command buffer support for buffer callbacks.

Author

dfm

Committer

Google-ML-Automation

Parents

9c65f62d

jax 958ea15a - Enable command buffer support for buffer callbacks.

jax
958ea15a - Enable command buffer support for buffer callbacks.