From 37bc4b94baad5eb1bfb919e31b88da0adea1f8d6 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Fri, 23 Mar 2018 15:37:07 +0800
Subject: [PATCH] Fix image to buffer write overflow.

---
 mace/kernels/opencl/buffer_to_image.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/mace/kernels/opencl/buffer_to_image.cc b/mace/kernels/opencl/buffer_to_image.cc
index 126fda77..6d8f3ef1 100644
--- a/mace/kernels/opencl/buffer_to_image.cc
+++ b/mace/kernels/opencl/buffer_to_image.cc
@@ -22,8 +22,7 @@ void BufferToImageFunctor<DeviceType::OPENCL, T>::operator()(
       image->ResizeImage(buffer->shape(), image_shape);
     }
   } else {
-    Image *image_buf = dynamic_cast<Image *>(image->UnderlyingBuffer());
-    image_shape = image_buf->image_shape();
+    CalImage2DShape(image->shape(), type, &image_shape);
     buffer->Resize(image->shape());
   }
 
-- 
GitLab