diff options
author | cnugteren <web@cedricnugteren.nl> | 2016-04-20 22:12:51 -0600 |
---|---|---|
committer | cnugteren <web@cedricnugteren.nl> | 2016-04-20 22:12:51 -0600 |
commit | 16a048f1ac8102ad4bcce9cf9fc320f791323e45 (patch) | |
tree | bd7df539c92b92cb8841ff27fd83c4e14e5958bc /src/clblast.cc | |
parent | 894983fc3c7c57ffc48c21523641694cde318eca (diff) |
Added support for the iSAMAX/iDAMAX/iCAMAX/iZAMAX routines
Diffstat (limited to 'src/clblast.cc')
-rw-r--r-- | src/clblast.cc | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/src/clblast.cc b/src/clblast.cc index bee63b53..145b6bf6 100644 --- a/src/clblast.cc +++ b/src/clblast.cc @@ -28,6 +28,7 @@ #include "internal/routines/level1/xdotc.h" #include "internal/routines/level1/xnrm2.h" #include "internal/routines/level1/xasum.h" +#include "internal/routines/level1/xamax.h" // BLAS level-2 includes #include "internal/routines/level2/xgemv.h" @@ -430,11 +431,17 @@ template StatusCode PUBLIC_API Asum<double2>(const size_t, // Index of absolute maxium value in a vector: iSAMAX/iDAMAX/iCAMAX/iZAMAX template <typename T> -StatusCode Amax(const size_t, - cl_mem, const size_t, - const cl_mem, const size_t, const size_t, - cl_command_queue*, cl_event*) { - return StatusCode::kNotImplemented; +StatusCode Amax(const size_t n, + cl_mem imax_buffer, const size_t imax_offset, + const cl_mem x_buffer, const size_t x_offset, const size_t x_inc, + cl_command_queue* queue, cl_event* event) { + auto queue_cpp = Queue(*queue); + auto routine = Xamax<T>(queue_cpp, event); + auto status = routine.SetUp(); + if (status != StatusCode::kSuccess) { return status; } + return routine.DoAmax(n, + Buffer<T>(imax_buffer), imax_offset, + Buffer<T>(x_buffer), x_offset, x_inc); } template StatusCode PUBLIC_API Amax<float>(const size_t, cl_mem, const size_t, |