Use stream argument when launching bilateral filter kernel

pull/5366/head
Sergey Alexandrov 9 years ago
parent d2add41b6b
commit 4094b2d7a5
  1. 2
      modules/cudaimgproc/src/cuda/bilateral_filter.cu

@ -136,7 +136,7 @@ namespace cv { namespace cuda { namespace device
float sigma_color2_inv_half = -0.5f/(sigma_color * sigma_color);
cudaSafeCall( cudaFuncSetCacheConfig (bilateral_kernel<T, B<T> >, cudaFuncCachePreferL1) );
bilateral_kernel<<<grid, block>>>((PtrStepSz<T>)src, (PtrStepSz<T>)dst, b, kernel_size, sigma_spatial2_inv_half, sigma_color2_inv_half);
bilateral_kernel<<<grid, block, 0, stream>>>((PtrStepSz<T>)src, (PtrStepSz<T>)dst, b, kernel_size, sigma_spatial2_inv_half, sigma_color2_inv_half);
cudaSafeCall ( cudaGetLastError () );
if (stream == 0)

Loading…
Cancel
Save