From 5270b335bde9a90135174ba0cb44f64fa426d2cb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=85?= <liyin@xiaomi.com>
Date: Wed, 2 May 2018 15:14:16 +0800
Subject: [PATCH] Fix lambda access memory bug

---
 mace/kernels/conv_2d.h | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/mace/kernels/conv_2d.h b/mace/kernels/conv_2d.h
index be134f7b..790d6f3a 100644
--- a/mace/kernels/conv_2d.h
+++ b/mace/kernels/conv_2d.h
@@ -424,7 +424,9 @@ struct Conv2dFunctor<DeviceType::CPU, float> : Conv2dFunctorBase {
         transformed_filter_ptr = transformed_filter_.data<float>();
       }
 
-      conv_func = [&](const float *pad_input, float *pad_output) {
+      float *transformed_input_data = transformed_input.mutable_data<float>();
+      float *transformed_output_data = transformed_output.mutable_data<float>();
+      conv_func = [=](const float *pad_input, float *pad_output) {
         WinoGradConv3x3s1(pad_input,
                           transformed_filter_ptr,
                           batch,
@@ -433,8 +435,8 @@ struct Conv2dFunctor<DeviceType::CPU, float> : Conv2dFunctorBase {
                           input_channels,
                           channels,
                           winograd_out_tile_size,
-                          transformed_input.mutable_data<float>(),
-                          transformed_output.mutable_data<float>(),
+                          transformed_input_data,
+                          transformed_output_data,
                           pad_output);
       };
     } else if (use_neon_3x3_s1) {
-- 
GitLab