diff options
author | Koichi Akabe <vbkaisetsu@gmail.com> | 2018-12-17 21:57:35 +0900 |
---|---|---|
committer | Koichi Akabe <vbkaisetsu@gmail.com> | 2018-12-17 21:57:35 +0900 |
commit | d9db543d75605fb02873e48197572450447481e1 (patch) | |
tree | fd865e5f9cf516143df05483f25cc0ab302f9306 /test/routines/levelx | |
parent | 0c9411c84465d14d2de33046536403648909eb9f (diff) |
Fix half-float+kernel_mode test cases of im2col, col2im, and convgemm
Diffstat (limited to 'test/routines/levelx')
-rw-r--r-- | test/routines/levelx/xcol2im.hpp | 1 | ||||
-rw-r--r-- | test/routines/levelx/xconvgemm.hpp | 4 | ||||
-rw-r--r-- | test/routines/levelx/xim2col.hpp | 1 |
3 files changed, 5 insertions, 1 deletions
diff --git a/test/routines/levelx/xcol2im.hpp b/test/routines/levelx/xcol2im.hpp index c28727e7..c740e4c7 100644 --- a/test/routines/levelx/xcol2im.hpp +++ b/test/routines/levelx/xcol2im.hpp @@ -204,6 +204,7 @@ StatusCode RunReference<half>(const Arguments<half> &args, BuffersHost<half> &bu auto buffers2 = BuffersHost<float>{dummy, dummy, a_buffer2, b_buffer2, dummy, dummy, dummy}; auto args2 = Arguments<float>(); args2.a_size = args.a_size; args2.b_size = args.b_size; + args2.kernel_mode = args.kernel_mode; args2.channels = args.channels; args2.height = args.height; args2.width = args.width; args2.kernel_h = args.kernel_h; args2.kernel_w = args.kernel_w; args2.pad_h = args.pad_h; args2.pad_w = args.pad_w; diff --git a/test/routines/levelx/xconvgemm.hpp b/test/routines/levelx/xconvgemm.hpp index e67b8174..786bb733 100644 --- a/test/routines/levelx/xconvgemm.hpp +++ b/test/routines/levelx/xconvgemm.hpp @@ -31,7 +31,8 @@ public: // The list of arguments relevant for this routine static std::vector<std::string> GetOptions() { - return {kArgChannels, kArgHeight, kArgWidth, kArgKernelH, kArgKernelW, kArgPadH, kArgPadW, + return {kArgKernelMode, + kArgChannels, kArgHeight, kArgWidth, kArgKernelH, kArgKernelW, kArgPadH, kArgPadW, kArgStrideH, kArgStrideW, kArgDilationH, kArgDilationW, kArgNumKernels, kArgBatchCount, kArgAOffset, kArgBOffset, kArgCOffset}; } @@ -232,6 +233,7 @@ StatusCode RunReference<half>(const Arguments<half> &args, BuffersHost<half> &bu auto buffers2 = BuffersHost<float>{dummy, dummy, a_buffer2, b_buffer2, c_buffer2, dummy, dummy}; auto args2 = Arguments<float>(); args2.a_size = args.a_size; args2.b_size = args.b_size; args2.c_size = args.c_size; + args2.kernel_mode = args.kernel_mode; args2.channels = args.channels; args2.height = args.height; args2.width = args.width; args2.kernel_h = args.kernel_h; args2.kernel_w = args.kernel_w; args2.pad_h = args.pad_h; args2.pad_w = args.pad_w; diff --git a/test/routines/levelx/xim2col.hpp b/test/routines/levelx/xim2col.hpp index 2a3577c3..2a5ebf8e 100644 --- a/test/routines/levelx/xim2col.hpp +++ b/test/routines/levelx/xim2col.hpp @@ -203,6 +203,7 @@ StatusCode RunReference<half>(const Arguments<half> &args, BuffersHost<half> &bu auto buffers2 = BuffersHost<float>{dummy, dummy, a_buffer2, b_buffer2, dummy, dummy, dummy}; auto args2 = Arguments<float>(); args2.a_size = args.a_size; args2.b_size = args.b_size; + args2.kernel_mode = args.kernel_mode; args2.channels = args.channels; args2.height = args.height; args2.width = args.width; args2.kernel_h = args.kernel_h; args2.kernel_w = args.kernel_w; args2.pad_h = args.pad_h; args2.pad_w = args.pad_w; |