提交 37bc4b94 编写于 作者: L liuqi

Fix image to buffer write overflow.

上级 75096372
...@@ -22,8 +22,7 @@ void BufferToImageFunctor<DeviceType::OPENCL, T>::operator()( ...@@ -22,8 +22,7 @@ void BufferToImageFunctor<DeviceType::OPENCL, T>::operator()(
image->ResizeImage(buffer->shape(), image_shape); image->ResizeImage(buffer->shape(), image_shape);
} }
} else { } else {
Image *image_buf = dynamic_cast<Image *>(image->UnderlyingBuffer()); CalImage2DShape(image->shape(), type, &image_shape);
image_shape = image_buf->image_shape();
buffer->Resize(image->shape()); buffer->Resize(image->shape());
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册