diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2017-10-15 19:35:21 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2017-10-15 19:35:21 +0200 |
commit | e6da575fff9d55de2b83def06243ca8dc9038f40 (patch) | |
tree | 79366158ad584728673585ead2c96547353b983d /test/routines/level1/xasum.hpp | |
parent | 7663cba23487290d7bf62c268410c840e3ee7972 (diff) |
Modified test interfaces such that they support either OpenCL or CUDA
Diffstat (limited to 'test/routines/level1/xasum.hpp')
-rw-r--r-- | test/routines/level1/xasum.hpp | 18 |
1 files changed, 13 insertions, 5 deletions
diff --git a/test/routines/level1/xasum.hpp b/test/routines/level1/xasum.hpp index 6add9c64..573f1223 100644 --- a/test/routines/level1/xasum.hpp +++ b/test/routines/level1/xasum.hpp @@ -69,13 +69,21 @@ class TestXasum { // Describes how to run the CLBlast routine static StatusCode RunRoutine(const Arguments<T> &args, Buffers<T> &buffers, Queue &queue) { + #ifdef OPENCL_API auto queue_plain = queue(); auto event = cl_event{}; - auto status = Asum<T>(args.n, - buffers.scalar(), args.asum_offset, - buffers.x_vec(), args.x_offset, args.x_inc, - &queue_plain, &event); - if (status == StatusCode::kSuccess) { clWaitForEvents(1, &event); clReleaseEvent(event); } + auto status = Asum<T>(args.n, + buffers.scalar(), args.asum_offset, + buffers.x_vec(), args.x_offset, args.x_inc, + &queue_plain, &event); + if (status == StatusCode::kSuccess) { clWaitForEvents(1, &event); clReleaseEvent(event); } + #elif CUDA_API + auto status = Asum<T>(args.n, + buffers.scalar(), args.asum_offset, + buffers.x_vec(), args.x_offset, args.x_inc, + queue.GetContext()(), queue.GetDevice()()); + cuStreamSynchronize(queue()); + #endif return status; } |