Modified test interfaces such that they support either OpenCL or CUDA

1 files changed, 17 insertions, 8 deletions

diff --git a/test/routines/levelx/xinvert.hpp b/test/routines/levelx/xinvert.hpp
index cc02a88b..3df1e2b0 100644
--- a/test/routines/levelx/xinvert.hpp
+++ b/test/routines/levelx/xinvert.hpp
@@ -164,14 +164,23 @@ class TestXinvert {
   // Describes how to run the CLBlast routine
   static StatusCode RunRoutine(const Arguments<T> &args, Buffers<T> &buffers, Queue &queue) {
     try {
-      auto event = cl_event{};
-      auto inverter = Xinvert<T>(queue, &event);
-      inverter.InvertMatrixDiagonalBlocks(args.layout, args.triangle, args.diagonal,
-                                          args.n, args.m,
-                                          buffers.a_mat, args.a_offset, args.a_ld,
-                                          buffers.b_mat);
-      clWaitForEvents(1, &event);
-      clReleaseEvent(event);
+      #ifdef OPENCL_API
+        auto event = cl_event{};
+        auto inverter = Xinvert<T>(queue, &event);
+        inverter.InvertMatrixDiagonalBlocks(args.layout, args.triangle, args.diagonal,
+                                            args.n, args.m,
+                                            buffers.a_mat, args.a_offset, args.a_ld,
+                                            buffers.b_mat);
+        clWaitForEvents(1, &event);
+        clReleaseEvent(event);
+      #elif CUDA_API
+        auto inverter = Xinvert<T>(queue, nullptr);
+        inverter.InvertMatrixDiagonalBlocks(args.layout, args.triangle, args.diagonal,
+                                            args.n, args.m,
+                                            buffers.a_mat, args.a_offset, args.a_ld,
+                                            buffers.b_mat);
+        cuStreamSynchronize(queue());
+      #endif
     } catch (...) { return DispatchException(); }
     return StatusCode::kSuccess;
   }