jax
f3338be0 - Add a field to plugin_attributes to indicate whether the PjRt plugin supports cross-host device transfers.

Commit
218 days ago
Add a field to plugin_attributes to indicate whether the PjRt plugin supports cross-host device transfers. This is needed for a subsequent change, in which we fall back to an experimental DCN transfer library in IFRT if the PjRt client doesn't implement the transfer APIs. PiperOrigin-RevId: 775915404
Author
Parents
Loading