diff options
Diffstat (limited to 'test/performance')
-rw-r--r-- | test/performance/client.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/performance/client.cpp b/test/performance/client.cpp index a2f0f9f4..dc98ffbd 100644 --- a/test/performance/client.cpp +++ b/test/performance/client.cpp @@ -183,7 +183,7 @@ void Client<T,U>::PerformanceTest(Arguments<U> &args, const SetMetric set_sizes) if (args.compare_clblas) { clblasSetup(); } #endif #ifdef CLBLAST_REF_CUBLAS - cudaSetDevice(static_cast<int>(args.device_id)); + if (args.compare_cublas) { cublasSetup(args); } #endif // Iterates over all "num_step" values jumping by "step" each time @@ -272,6 +272,9 @@ void Client<T,U>::PerformanceTest(Arguments<U> &args, const SetMetric set_sizes) #ifdef CLBLAST_REF_CLBLAS if (args.compare_clblas) { clblasTeardown(); } #endif + #ifdef CLBLAST_REF_CUBLAS + if (args.compare_cublas) { cublasTeardown(args); } + #endif } // ================================================================================================= |