No public description

PiperOrigin-RevId: 571412955
This commit is contained in:
Chris McClanahan 2023-10-06 12:57:26 -07:00 committed by Copybara-Service
parent 830ee092b9
commit b503d71be4

View File

@ -191,6 +191,11 @@ absl::Status InferenceCalculatorMetalImpl::Process(CalculatorContext* cc) {
[output_encoder endEncoding];
}
[command_buffer commit];
// The below call is found (manual testing) to resolve flickering issues for
// some use cases where multiple Metal calculators are involved.
// TODO: investigate and ensure proper synchronization
// (e.g. fences/barriers/events).
[command_buffer waitUntilScheduled];
kOutTensors(cc).Send(std::move(output_tensors));
return absl::OkStatus();