Wait until the metal backend finishes its work in the TensorsToImageCalculator.
PiperOrigin-RevId: 516360846
This commit is contained in:
parent
89857f33a2
commit
57f106e0a7
|
@ -294,7 +294,7 @@ absl::Status TensorsToImageCalculator::MetalProcess(CalculatorContext* cc) {
|
|||
threadsPerThreadgroup:threads_per_group];
|
||||
[compute_encoder endEncoding];
|
||||
[command_buffer commit];
|
||||
|
||||
[command_buffer waitUntilCompleted];
|
||||
kOutputImage(cc).Send(Image(output));
|
||||
return absl::OkStatus();
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue
Block a user