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];
|
threadsPerThreadgroup:threads_per_group];
|
||||||
[compute_encoder endEncoding];
|
[compute_encoder endEncoding];
|
||||||
[command_buffer commit];
|
[command_buffer commit];
|
||||||
|
[command_buffer waitUntilCompleted];
|
||||||
kOutputImage(cc).Send(Image(output));
|
kOutputImage(cc).Send(Image(output));
|
||||||
return absl::OkStatus();
|
return absl::OkStatus();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user