fix_equal (#8347)

PaddlePaddle · Feb 4, 2022 · 536ee2a · 536ee2a
1 parent a2c4b60
commit 536ee2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lite/kernels/metal/image_op/compare_image_compute.mm b/lite/kernels/metal/image_op/compare_image_compute.mm
@@ -44,7 +44,7 @@
     bool should_use_mps = false;
     if (@available(iOS 12.1, *)) {
         if (metal_context_->use_mps()) {
-            should_use_mps = true;
+            if (input_buffer_x_->tensor_dim_[0] == 1) should_use_mps = true;
         }
     }