add event

src/operators/kernel/cl/relu_kernel.cpp

@@ -50,9 +50,12 @@ void ReluKernel<GPU_CL, float>::Compute(const ReluParam<GPU_CL>& param) {
 //  clSetKernelArg(kernel_p1, 1, sizeof(cl_mem), &outputImage);
     const size_t work_size[2] = {input->ImageWidth(), input->ImageHeight()};
 
-    clEnqueueNDRangeKernel(this->cl_helper_.CLCommandQueue(), kernel, 2,
+  cl_event out_event = param.Out()->GetClEvent();
+  cl_event wait_event = param.InputX()->GetClEvent();
+
+  clEnqueueNDRangeKernel(this->cl_helper_.CLCommandQueue(), kernel, 2,
     NULL,
-                           work_size, NULL, 0, NULL, NULL);
+                           work_size, NULL, 1, &wait_event, &out_event);
   //  clEnqueueNDRangeKernel(this->cl_helper_.CLCommandQueue(), kernel_p1, 3,
   //  NULL,
   //                         work_size, NULL, 0, NULL, NULL);

src/operators/kernel/cl/reshape_kernel.cpp

@@ -55,8 +55,11 @@ void ReshapeKernel<GPU_CL, float>::Compute(const ReshapeParam<GPU_CL> &param) {
   clSetKernelArg(kernel, 9, sizeof(cl_int), &odims[1]);
   const size_t work_size[2] = {output->ImageWidth(), output->ImageHeight()};
 
+  cl_event out_event = param.Out()->GetClEvent();
+  cl_event wait_event = param.InputX()->GetClEvent();
+
   clEnqueueNDRangeKernel(this->cl_helper_.CLCommandQueue(), kernel, 2, NULL,
-                         work_size, NULL, 0, NULL, NULL);
+                         work_size, NULL, 1, &wait_event, &out_event);
 }
 
 template class ReshapeKernel<GPU_CL, float>;

src/operators/kernel/cl/softmax_kernel.cpp

@@ -55,8 +55,12 @@ void SoftmaxKernel<GPU_CL, float>::Compute(const SoftmaxParam<GPU_CL> &param) {
 //  clSetKernelArg(kernel, 4, sizeof(int), &dims[2]);
 //  clSetKernelArg(kernel, 5, sizeof(int), &dims[3]);
 
+  cl_event out_event = param.Out()->GetClEvent();
+  cl_event wait_event = param.InputX()->GetClEvent();
+
   status = clEnqueueNDRangeKernel(this->cl_helper_.CLCommandQueue(), kernel, default_work_size.size(), NULL,
-                         default_work_size.data(), NULL, 0, NULL, NULL);
+                         default_work_size.data(), NULL, 1, &wait_event, &out_event);
+
   CL_CHECK_ERRORS(status);
 
 }