Skip to content

Commit 8252a74

Browse files
committed
Fix stream in KernelContext in convolution and laplacian tests.
Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
1 parent dfde565 commit 8252a74

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

dali/kernels/imgproc/convolution/convolution_gpu_test.cu

+1
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@ struct ConvolutionGpuKernelTest : public ::testing::Test {
158158

159159
void RunTest() {
160160
KernelContext ctx_cpu, ctx_gpu;
161+
ctx_gpu.gpu.stream = 0;
161162
KernelCpu kernel_cpu;
162163
KernelGpu kernel_gpu;
163164

dali/kernels/imgproc/convolution/laplacian_gpu_test.cu

+1
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,7 @@ struct LaplacianGpuTest : public ::testing::Test {
161161

162162
void RunTest() {
163163
KernelContext ctx_cpu = {}, ctx_gpu = {};
164+
ctx_gpu.gpu.stream = 0;
164165
KernelCpu kernel_cpu;
165166
KernelGpu kernel_gpu;
166167
int nsamples = in_.shape.size();

0 commit comments

Comments
 (0)