No public description
PiperOrigin-RevId: 571412955
This commit is contained in:
parent
830ee092b9
commit
b503d71be4
|
@ -191,6 +191,11 @@ absl::Status InferenceCalculatorMetalImpl::Process(CalculatorContext* cc) {
|
||||||
[output_encoder endEncoding];
|
[output_encoder endEncoding];
|
||||||
}
|
}
|
||||||
[command_buffer commit];
|
[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));
|
kOutTensors(cc).Send(std::move(output_tensors));
|
||||||
return absl::OkStatus();
|
return absl::OkStatus();
|
||||||
|
|
Loading…
Reference in New Issue
Block a user