From 8c700b6a8d8502778543b6b1d17e403979bf373f Mon Sep 17 00:00:00 2001
From: luxuhui <luxuhui@xiaomi.com>
Date: Tue, 18 Aug 2020 20:06:40 +0800
Subject: [PATCH] fix: set Deconv2d filter's opencl buffer type to
 CONV2D_FILTER

N/A

Signed-off-by: Luxuhui <luxuhui@xiaomi.com>
---
 mace/ops/deconv_2d.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/ops/deconv_2d.cc b/mace/ops/deconv_2d.cc
index 4e0a6047..dc58141a 100644
--- a/mace/ops/deconv_2d.cc
+++ b/mace/ops/deconv_2d.cc
@@ -248,7 +248,7 @@ void RegisterDeconv2D(OpRegistry *op_registry) {
       op_registry,
       OpConditionBuilder("Deconv2D").SetInputMemoryTypeSetter(
           [](OpConditionContext *context) -> void {
-            SetFilterMemoryType(context, OpenCLBufferType::DW_CONV2D_FILTER);
+            SetFilterMemoryType(context, OpenCLBufferType::CONV2D_FILTER);
             if (context->device()->device_type() == DeviceType::GPU) {
               FrameworkType framework_type =
                   static_cast<FrameworkType>(
-- 
GitLab