可以看一下github mluops
Hello , the word "scalability" may be wrong in
http://www.cambricon.com/docs/bangc/developer_guide_html/4ProgrammingModel/index.html line 5
1 | <p style= "line-height: 1.5em;" ><span style= "font-family: 微软雅黑, "Microsoft YaHei";" >#include "macro.h" <br>#include "mlu.h" <br>__mlu_entry__ void ConvKernel(half* out_data, half* in_data, half* filter_data,<br> int in_channel, int in_height, int in_width,<br> int stride_height, int stride_width,<br> int out_channel) {<br> __nram__ half nram_out_data[OUT_DATA_NUM];<br> __nram__ half nram_in_data[OUT_DATA_NUM];<br> __wram__ half wram_filter[FILTER_DATA_NUM];<br> __memcpy(nram_in_data, in_data, IN_DATA_NUM * sizeof (half),GDRAM2NRAM); <br> __memcpy(wram_filter, filter_data, FILTER_DATA_NUM * sizeof (half),GDRAM2WRAM); <br> __bang_conv(nram_out_data, nram_in_data, wram_filter, in_channel,<br> IN_HEIGHT, IN_WIDTH, filter_height, filter_width, stride_height,<br> stride_width, out_channel);<br> __memcpy(out_data, nram_out_data, OUT_DATA_NUM * sizeof (half),NRAM2GDRAM);<br>}<br></span></p> |
这个改一下吧,格式不对
请登录后评论