1 files changed, 8 insertions, 7 deletions
diff --git a/src/routines/common.hpp b/src/routines/common.hpp
index bdea0086..47d62027 100644
--- a/src/routines/common.hpp
+++ b/src/routines/common.hpp
@@ -38,17 +38,18 @@ template <typename T>
 void FillMatrix(Queue &queue, const Device &device,
                 const Program &program, const Database &,
                 EventPointer event, const std::vector<Event> &waitForEvents,
-                const size_t n, const size_t ld, const size_t offset,
+                const size_t m, const size_t n, const size_t ld, const size_t offset,
                 const Buffer<T> &dest,
                 const T constant_value) {
   auto kernel = Kernel(program, "FillMatrix");
-  kernel.SetArgument(0, static_cast<int>(n));
-  kernel.SetArgument(1, static_cast<int>(ld));
-  kernel.SetArgument(2, static_cast<int>(offset));
-  kernel.SetArgument(3, dest());
-  kernel.SetArgument(4, GetRealArg(constant_value));
+  kernel.SetArgument(0, static_cast<int>(m));
+  kernel.SetArgument(1, static_cast<int>(n));
+  kernel.SetArgument(2, static_cast<int>(ld));
+  kernel.SetArgument(3, static_cast<int>(offset));
+  kernel.SetArgument(4, dest());
+  kernel.SetArgument(5, GetRealArg(constant_value));
   auto local = std::vector<size_t>{8, 8};
-  auto global = std::vector<size_t>{Ceil(ld, 8), Ceil(n, 8)};
+  auto global = std::vector<size_t>{Ceil(m, 8), Ceil(n, 8)};
   RunKernel(kernel, queue, device, global, local, event, waitForEvents);
 }