From e587261e4d921295f92e9b1090336d99f59691c8 Mon Sep 17 00:00:00 2001 From: augustin64 Date: Mon, 15 May 2023 11:34:58 +0200 Subject: [PATCH] test/cnn_convolution: update to newer function --- test/cnn_convolution.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/cnn_convolution.cu b/test/cnn_convolution.cu index d7765de..d18c5c2 100644 --- a/test/cnn_convolution.cu +++ b/test/cnn_convolution.cu @@ -140,7 +140,7 @@ void run_convolution_test(int input_width, int output_width, int rows, int colum double cpu_time_used, gpu_time_used; start_time = omp_get_wtime(); - make_convolution_device(kernel, input, output_gpu, output_width, 1); + make_convolution_device(kernel, input, output_gpu, output_width, 1, 0); end_time = omp_get_wtime(); @@ -149,7 +149,7 @@ void run_convolution_test(int input_width, int output_width, int rows, int colum start_time = omp_get_wtime(); - make_convolution_cpu(kernel, input, output_cpu, output_width, 1); + make_convolution_cpu(kernel, input, output_cpu, output_width, 1, 0); end_time = omp_get_wtime(); cpu_time_used = end_time - start_time;