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
Job | Run time |
---|---|
0s | |
0s |