diff --git a/benchmark/portblas/blas2/ger.cpp b/benchmark/portblas/blas2/ger.cpp index 4afdb562f..0b4b853ad 100644 --- a/benchmark/portblas/blas2/ger.cpp +++ b/benchmark/portblas/blas2/ger.cpp @@ -105,7 +105,7 @@ void run(benchmark::State& state, blas::SB_Handle* sb_handle_ptr, index_t m, auto blas_method_def = [&]() -> std::vector { auto event = _ger(sb_handle, m, n, alpha, v_x_gpu, incX, v_y_gpu, incY, m_a_gpu, lda); - sb_handle.wait(); + sb_handle.wait(event); return event; }; diff --git a/benchmark/portblas/blas2/symv.cpp b/benchmark/portblas/blas2/symv.cpp index a9c7baf22..c93241439 100644 --- a/benchmark/portblas/blas2/symv.cpp +++ b/benchmark/portblas/blas2/symv.cpp @@ -105,7 +105,7 @@ void run(benchmark::State& state, blas::SB_Handle* sb_handle_ptr, auto blas_method_def = [&]() -> std::vector { auto event = _symv(sb_handle, *uplo_str, n, alpha, m_a_gpu, lda, v_x_gpu, incX, beta, v_y_gpu, incY); - sb_handle.wait(); + sb_handle.wait(event); return event; }; diff --git a/benchmark/portblas/blas2/syr.cpp b/benchmark/portblas/blas2/syr.cpp index b6c5ebcbc..0f80905aa 100644 --- a/benchmark/portblas/blas2/syr.cpp +++ b/benchmark/portblas/blas2/syr.cpp @@ -96,7 +96,7 @@ void run(benchmark::State& state, blas::SB_Handle* sb_handle_ptr, auto blas_method_def = [&]() -> std::vector { auto event = _syr(sb_handle, *uplo_str, n, alpha, v_x_gpu, incX, m_a_gpu, lda); - sb_handle.wait(); + sb_handle.wait(event); return event; }; diff --git a/benchmark/portblas/blas2/syr2.cpp b/benchmark/portblas/blas2/syr2.cpp index 497d17b66..0ead8dc66 100644 --- a/benchmark/portblas/blas2/syr2.cpp +++ b/benchmark/portblas/blas2/syr2.cpp @@ -106,7 +106,7 @@ void run(benchmark::State& state, blas::SB_Handle* sb_handle_ptr, auto blas_method_def = [&]() -> std::vector { auto event = _syr2(sb_handle, *uplo_str, n, alpha, v_x_gpu, incX, v_y_gpu, incY, m_a_gpu, lda); - sb_handle.wait(); + sb_handle.wait(event); return event; }; diff --git a/benchmark/portblas/blas2/trmv.cpp b/benchmark/portblas/blas2/trmv.cpp index 83e7a8585..ac6d54818 100644 --- a/benchmark/portblas/blas2/trmv.cpp +++ b/benchmark/portblas/blas2/trmv.cpp @@ -99,7 +99,7 @@ void run(benchmark::State& state, blas::SB_Handle* sb_handle_ptr, auto blas_method_def = [&]() -> std::vector { auto event = _trmv(sb_handle, *uplo_str, *t_str, *diag_str, n, m_a_gpu, lda, v_x_gpu, incX); - sb_handle.wait(); + sb_handle.wait(event); return event; };