From 16a048f1ac8102ad4bcce9cf9fc320f791323e45 Mon Sep 17 00:00:00 2001
From: cnugteren <web@cedricnugteren.nl>
Date: Wed, 20 Apr 2016 22:12:51 -0600
Subject: Added support for the iSAMAX/iDAMAX/iCAMAX/iZAMAX routines

---
 src/clblast.cc | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

(limited to 'src/clblast.cc')

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