diff options
Diffstat (limited to 'src/routines/common.hpp')
| -rw-r--r-- | src/routines/common.hpp | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/routines/common.hpp b/src/routines/common.hpp index 47f98de0..8046c0be 100644 --- a/src/routines/common.hpp +++ b/src/routines/common.hpp @@ -52,6 +52,25 @@ void FillMatrix(Queue &queue, const Device &device, RunKernel(kernel, queue, device, global, local, event, waitForEvents); } +// Sets all elements of a vector to a constant value +template <typename T> +void FillVector(Queue &queue, const Device &device, + const Program &program, const Database &, + EventPointer event, const std::vector<Event> &waitForEvents, + const size_t n, const size_t inc, const size_t offset, + const Buffer<T> &dest, + const T constant_value) { + auto kernel = Kernel(program, "FillVector"); + kernel.SetArgument(0, static_cast<int>(n)); + kernel.SetArgument(1, static_cast<int>(inc)); + kernel.SetArgument(2, static_cast<int>(offset)); + kernel.SetArgument(3, dest()); + kernel.SetArgument(4, GetRealArg(constant_value)); + auto local = std::vector<size_t>{64}; + auto global = std::vector<size_t>{Ceil(n, 64)}; + RunKernel(kernel, queue, device, global, local, event, waitForEvents); +} + // ================================================================================================= // Copies or transposes a matrix and optionally pads/unpads it with zeros. This method is also able |