From e057a9186a1ed0a169fcf4db7a2598d08f530834 Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Thu, 17 May 2018 09:23:28 +0100 Subject: First version of direct reading from image tensor for convgemm: only for edge cases now --- src/routines/levelx/xconvgemm.cpp | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'src/routines') 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::DoConvgemm(const size_t channels, const size_t height, const kernel.SetArgument(8, result_buffer()); kernel.SetArgument(9, static_cast(result_offset)); kernel.SetArgument(10, static_cast(result_stride)); + kernel.SetArgument(11, static_cast(height)); + kernel.SetArgument(12, static_cast(width)); + kernel.SetArgument(13, static_cast(channels)); + kernel.SetArgument(14, static_cast(kernel_h)); + kernel.SetArgument(15, static_cast(kernel_w)); + kernel.SetArgument(16, static_cast(pad_h)); + kernel.SetArgument(17, static_cast(pad_w)); + kernel.SetArgument(18, static_cast(stride_h)); + kernel.SetArgument(19, static_cast(stride_w)); + kernel.SetArgument(20, static_cast(dilation_h)); + kernel.SetArgument(21, static_cast(dilation_w)); + kernel.SetArgument(22, im_buffer()); + kernel.SetArgument(23, static_cast(im_offset)); + kernel.SetArgument(24, static_cast(output_h)); + kernel.SetArgument(25, static_cast(output_w)); // Computes the global and local thread sizes const auto m_ceiled = Ceil(num_patches, db_["WGD"]); -- cgit v1.2.3