First version of direct reading from image tensor for convgemm: only for edge cases now

1 files changed, 15 insertions, 0 deletions

diff --git a/src/routines/levelx/xconvgemm.cpp b/src/routines/levelx/xconvgemm.cpp
index 8cb8093c..23335261 100644
--- a/src/routines/levelx/xconvgemm.cpp
+++ b/src/routines/levelx/xconvgemm.cpp
@@ -117,6 +117,21 @@ void Xconvgemm<T>::DoConvgemm(const size_t channels, const size_t height, const
   kernel.SetArgument(8, result_buffer());
   kernel.SetArgument(9, static_cast<int>(result_offset));
   kernel.SetArgument(10, static_cast<int>(result_stride));
+  kernel.SetArgument(11, static_cast<int>(height));
+  kernel.SetArgument(12, static_cast<int>(width));
+  kernel.SetArgument(13, static_cast<int>(channels));
+  kernel.SetArgument(14, static_cast<int>(kernel_h));
+  kernel.SetArgument(15, static_cast<int>(kernel_w));
+  kernel.SetArgument(16, static_cast<int>(pad_h));
+  kernel.SetArgument(17, static_cast<int>(pad_w));
+  kernel.SetArgument(18, static_cast<int>(stride_h));
+  kernel.SetArgument(19, static_cast<int>(stride_w));
+  kernel.SetArgument(20, static_cast<int>(dilation_h));
+  kernel.SetArgument(21, static_cast<int>(dilation_w));
+  kernel.SetArgument(22, im_buffer());
+  kernel.SetArgument(23, static_cast<int>(im_offset));
+  kernel.SetArgument(24, static_cast<int>(output_h));
+  kernel.SetArgument(25, static_cast<int>(output_w));
 
   // Computes the global and local thread sizes
   const auto m_ceiled = Ceil(num_patches, db_["WGD"]);