From b35e3d1e5326cdc257daa170eb243800616cfc26 Mon Sep 17 00:00:00 2001
From: Cedric Nugteren <web@cedricnugteren.nl>
Date: Sun, 14 Jan 2018 19:50:27 +0100
Subject: Small improvements to benchmarking for cuBLAS

---
 test/performance/client.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'test/performance')

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));
-- 
cgit v1.2.3