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.
#62351
This job was skipped
Loading