From 278a2af7d37a033cb3e2dccd4fdf2231c2e08c0f Mon Sep 17 00:00:00 2001 From: dolphin8 Date: Wed, 17 Oct 2018 18:52:02 +0800 Subject: [PATCH] fix relu cl_image --- src/operators/kernel/cl/relu_kernel.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/operators/kernel/cl/relu_kernel.cpp b/src/operators/kernel/cl/relu_kernel.cpp index 91eab240e7..5c8e688639 100644 --- a/src/operators/kernel/cl/relu_kernel.cpp +++ b/src/operators/kernel/cl/relu_kernel.cpp @@ -24,7 +24,7 @@ bool ReluKernel::Init(ReluParam* param) { this->cl_helper_.AddKernel("relu_p0", "relu.cl"); this->cl_helper_.AddKernel("relu_p1", "relu.cl"); const auto dim = const_cast(param->InputX())->ImageDims(); - param->getMidImage().InitEmptyImage(this->cl_helper_.CLContext(), dim); + param->getMidImage().InitEmptyImage(this->cl_helper_.CLContext(), this->cl_helper_.CLCommandQueue(), dim); return true; } -- GitLab