fix relu cl_image

278a2af7 · dolphin8 · 8d2e49d9 · 278a2af7
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

src/operators/kernel/cl/relu_kernel.cpp src/operators/kernel/cl/relu_kernel.cpp +1 -1

未找到文件。
--- a/src/operators/kernel/cl/relu_kernel.cpp
+++ b/src/operators/kernel/cl/relu_kernel.cpp
@@ -24,7 +24,7 @@ bool ReluKernel<GPU_CL, float>::Init(ReluParam<GPU_CL>* param) {
  this->cl_helper_.AddKernel("relu_p0", "relu.cl");
  this->cl_helper_.AddKernel("relu_p1", "relu.cl");
  const auto dim = const_cast<framework::CLImage*>(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;
 }