From 57f106e0a72645f871f2875b1eaa51c6ca7f02a2 Mon Sep 17 00:00:00 2001
From: Jiuqiang Tang <jqtang@google.com>
Date: Mon, 13 Mar 2023 16:53:22 -0700
Subject: [PATCH] Wait until the metal backend finishes its work in the
 TensorsToImageCalculator.

PiperOrigin-RevId: 516360846
---
 .../face_stylizer/calculators/tensors_to_image_calculator.cc    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mediapipe/tasks/cc/vision/face_stylizer/calculators/tensors_to_image_calculator.cc b/mediapipe/tasks/cc/vision/face_stylizer/calculators/tensors_to_image_calculator.cc
index 03760c6b3..d9825b15f 100644
--- a/mediapipe/tasks/cc/vision/face_stylizer/calculators/tensors_to_image_calculator.cc
+++ b/mediapipe/tasks/cc/vision/face_stylizer/calculators/tensors_to_image_calculator.cc
@@ -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();
 }