Allow individual layouts in the device_layouts
argument of PjRtClient::CreateBuffersForAsyncHostToDevice
to be std::nullopt
, meaning that those buffers should use the default device layout.
#33305
This job was skipped
Loading