Small improvements to benchmarking for cuBLAS

1 files changed, 4 insertions, 1 deletions

diff --git a/test/performance/client.cpp b/test/performance/client.cpp
index 83088223..9480d11a 100644
--- a/test/performance/client.cpp
+++ b/test/performance/client.cpp
@@ -271,7 +271,10 @@ void Client<T,U>::PerformanceTest(Arguments<U> &args, const SetMetric set_sizes)
       auto buffers_cuda = BuffersCUDA<T>();
       DeviceToHost(args, buffers, buffers_host, queue, buffers_in_);
       HostToCUDA(args, buffers_cuda, buffers_host, buffers_in_);
-      auto ms_cublas = TimedExecution(args.num_runs, args, buffers_cuda, queue, run_reference3_, "cuBLAS");
+      auto ms_cublas = 0.0;
+      try {
+        ms_cublas = TimedExecution(args.num_runs, args, buffers_cuda, queue, run_reference3_, "cuBLAS");
+      } catch (std::runtime_error e) { }
       CUDAToHost(args, buffers_cuda, buffers_host, buffers_out_);
       HostToDevice(args, buffers, buffers_host, queue, buffers_out_);
       timings.push_back(std::pair<std::string, double>("cuBLAS", ms_cublas));