cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests 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:53:22 -07:00 committed by Copybara-Service
parent 89857f33a2
commit 57f106e0a7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/tasks/cc/vision/face_stylizer/calculators/tensors_to_image_calculator.cc
View File

@ -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();
} }