talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Wait until the metal backend finishes its work in the TensorsToImageCalculator.

Browse Source 
PiperOrigin-RevId: 516360846
This commit is contained in:
Jiuqiang Tang
2023-03-13 16:56:46 -07:00
committed by Copybara-Service
parent 89857f33a2
commit 57f106e0a7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/cc/vision/face_stylizer/calculators/tensors_to_image_calculator.cc
+1 -1
View File
@@ -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();
}