openppl-public / ppl.kernel.cpu Goto Github PK

select_algo函数sse分支ret_info默认输入输出的data format都是ndarray，sse_fallback_info中的format也是ndarray，那么gen_algo函数中如何选到n8cx的实现？
https://github.com/openppl-public/ppl.kernel.cpu/blob/master/src/ppl/kernel/x86/fp32/conv2d/conv2d_fp32.cpp#L283
另外gen_algo中sse分支关于input data_format的判断是不是写错了？
https://github.com/openppl-public/ppl.kernel.cpu/blob/master/src/ppl/kernel/x86/fp32/conv2d/conv2d_fp32.cpp#LL382C67-L382C67

具体代码如上，不知道是否是手滑写错了

测试命令如下：
./test_conv2d -cfg cfg.txt -algo ndarray_winograd_b6f3_fp32_sse
cfg.txt如下：
g1_mb1_ic12ih1iw9_oc4oh1ow9_kh3kw3sh1sw1ph1pw1dh0dw0_nwino

此时会出现channel2=0的情况，而winograd_b6f3_transpose_4x4_fp32_sse_func_table[1][0]是nullptr
https://github.com/openppl-public/ppl.kernel.cpu/blob/master/src/ppl/kernel/x86/fp32/conv2d/sse/conv2d_winograd_b6f3_fp32_sse.cpp#L280
https://github.com/openppl-public/ppl.kernel.cpu/blob/master/src/ppl/kernel/x86/fp32/conv2d/sse/conv2d_winograd_b6f3_fp32_sse.cpp#L229