fix bugs of avx2 (#360)

27d5295c · BUG1989 · GitHub · 342f780e · 27d5295c · 27d5295c
展开全部隐藏空白更改
内联并排

Showing with 793 addition and 6 deletion

src/dev/cpu/op/conv/conv_hcl_kernel.h src/dev/cpu/op/conv/conv_hcl_kernel.h +1 -1

src/dev/cpu/op/conv/x86/conv_kernel_x86.c src/dev/cpu/op/conv/x86/conv_kernel_x86.c +792 -5

未找到文件。
--- a/src/dev/cpu/op/conv/conv_hcl_kernel.h
+++ b/src/dev/cpu/op/conv/conv_hcl_kernel.h
@@ -40,7 +40,7 @@ int conv_hcl_run(struct ir_tensor* input_tensor, struct ir_tensor* filter_tensor
 int conv_hcl_get_shared_mem_size(struct ir_tensor* input_tensor, struct ir_tensor* output_tensor,
                                 struct conv_param* param) __attribute__((weak));
 int conv_hcl_get_shared_pack4_mem_size(struct ir_tensor* input_tensor, struct ir_tensor* output_tensor,
-                                 struct conv_param* param) __attribute__((weak));                                 
+                                       struct conv_param* param) __attribute__((weak));

 int conv_hcl_set_shared_mem(struct conv_priv_info* priv_info, void* mem, int mem_size) __attribute__((weak));
 int conv_hcl_set_shared_pack4_mem(struct conv_priv_info* priv_info, void* mem, int mem_size) __attribute__((weak));

--- a/src/dev/cpu/op/conv/x86/conv_kernel_x86.c
+++ b/src/dev/cpu/op/conv/x86/conv_kernel_x86.c