基于Linux C语言的多核并行卷积神经网络库源码

CNN C C, C, Shell 共32个文件 c: 10 h: 6 sh: 4 json: 3 out: 2 gitignore: 1 md: 1 png: 1 so: 1 makefile: 1 基于Linux C 的, 多核并行卷积运算, 通过配置CPU核, 以提升CPU最大并行计算性能 API调用生命周期 接口函数详情 /* * 并行高速算法处理接口, ---由江苏捷世智通提供 * * 2023-03-10: member * hanguangxue <hanguangxue_x@163.com> - C for PCIE driver and CPU do CNN * qianmen <qianmen@xx.com> - FPGA for PCIE driver and FPGA do CNN * * [中文] 该程序用于并行高速算法处理类接口, 提供需要对相应算法做加速的模块 * * [English] This program is used for parallel high speed algorithm processing class interface, * provides the need to do the corresponding algorithm to accelerate the module * * version 1.0, 2023-03-10 * support CNN Accelerate * */ #ifndef __JSZT_CNN__H #define __JSZT_CNN__H #define uint8 unsigned char #define uint16 unsigned short typedef enum { JS_OK, JS_FAILED, /** 参数错误 */ JS_ERROR_BORDER, //卷积数据和卷积核边界错误, 如果卷积数据相对卷积核来说过小则报此错 JS_ERROR_CONFIG, //配置出错, eg: 选择CPU模式的时候, CPU核数给的0 /** 系统错误 */ JS_ERROR_CREATE__TH, //创建线程资源失败,针对某些裁剪后的linux,对线程资源有限制,如果超出系统最大数会返回错误 JS_ERROR_RELEASE_TH, //线程销毁时返回该错误, 销毁线程需要调用 jszt_setEnd 接口 JS_ERROR_MALLOC, //堆内存申请出错 JS_ERROR_NULL, //空指针异常 JS_ERROR_CNNING, //执行卷积运算时出错 } JSZT_CODE; typedef enum { CALCULAT_FPGA, CALCULAT_CPU, } MODE_CALCULAT; typedef struct { uint8 * stream; //流输入, 输入为处理后的流, 1024*1024, 一位占 8bit, 也就是一字节 uint16 width; //流矩阵宽度, width * height 则为 stream 的实际长度 uint16 height; //流矩阵长度 uint8 * cnnker; //卷积核, 同样的一位占 8bit, 长度根据 cnnker_matrix_len 输入解析, 是数组 cnnker_matrix_len (x1+x2+..+x4)² uint8 ckrlen[4]; //卷积核矩阵长度, 参数 @CNNKER_XX, 按次序输入, 如果是只需要一次, 则 [x,0,0,0], 需要计算2次, 则[x1, x2, 0, 0] /* 返回参数 */ uint8 * dstream; //处理后的数据, 申请内存大小 >= (width-ckerlen+1)*(height-ckerlen+1) uint16 dwidth; //处理后矩阵 宽 uint16 dheight; //处理后矩阵 高 } CNN_PARA; /** * @details 初始化相关资源 * @param mode 初始化计算类型 * @return 0: 返回OK, 否则返回失败, 查看对应错误信息 */ extern JSZT_CODE jszt_init(MODE_CALCULAT mode, int CPUS); /** * @details 开始计算前预备工作 * @return 0: 返回OK, 否则返回失败, 查看对应错误信息 */ extern JSZT_CODE jszt_setReady(); /** * @param cnnPara 调用时需要的参数 * @return 处理后的数据长度, 处理后的数据放在 dstream 中, 直接从 stream 中读取指定长度数据即可 */ extern JSZT_CODE jszt_CNN(const CNN_PARA *cnnPara); /** * @details 计算完成后需要释放的资源 * @return 0: 返回OK, 否则返回失败, 查看对应错误信息 */ extern JSZT_CODE jszt_setEnd(); /** * @details 程序结束后需要调用释放的资源 * @return 0: 返回OK, 否则返回失败, 查看对应错误信息 */ extern JSZT_CODE jszt_destroy(); #endif // !__JSZT_CNN__H https://gitee.com/han_gx/cnn