diff options
Diffstat (limited to 'include/clblast.h')
| -rw-r--r-- | include/clblast.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/include/clblast.h b/include/clblast.h index 020f8e79..aeea4e52 100644 --- a/include/clblast.h +++ b/include/clblast.h @@ -609,6 +609,15 @@ StatusCode Omatcopy(const Layout layout, const Transpose a_transpose, cl_mem b_buffer, const size_t b_offset, const size_t b_ld, cl_command_queue* queue, cl_event* event = nullptr); +// Batched version of AXPY: SAXPYBATCHED/DAXPYBATCHED/CAXPYBATCHED/ZAXPYBATCHED/HAXPYBATCHED +template <typename T> +StatusCode AxpyBatched(const size_t n, + const T *alphas, + const cl_mem *x_buffers, const size_t x_offset, const size_t x_inc, + cl_mem *y_buffers, const size_t y_offset, const size_t y_inc, + const size_t batch_count, + cl_command_queue* queue, cl_event* event = nullptr); + // ================================================================================================= // CLBlast stores binaries of compiled kernels into a cache in case the same kernel is used later on |