打印
[技术讨论]

在全志D1开发板上玩ncnn

[复制链接]
382|0
手机看帖
扫描二维码
随时随地手机跟帖
跳转到指定楼层
楼主
神棍地海棠|  楼主 | 2024-2-23 10:10 | 只看该作者 回帖奖励 |倒序浏览 |阅读模式
0x0 工具链变得更好了
距上次在全志d1开发板上玩ncnn,已经过去了5个月
在此期间,ncnn收到perfxlab和腾讯犀牛鸟开源人才的学生有关riscv vector的优化
但更重要的是,平头哥收到了社区的反馈,提供了新版工具链
  • 支持了 risc-v vector intrinsic v1.0
  • 修复了 release 模式编译 ncnn 时的非法指令问题
occ.t-head.cn/community/download?id=3987221940543754240
旧版本工具链的 gcc 比较笨,经常做些负优化,于是试试全新的工具链
0x1 配置新的 cmake toolchain旧-march=rv64gcvxtheadc -mabi=lp64d -mtune=c906 -DRVV_SPEC_0_7 -D__riscv_zfh=1 -static新-march=rv64gcv0p7_zfh_xtheadc -mabi=lp64d -mtune=c906 -static
  • arch 参数要用 v0p7,不能用默认的 v,否则会生成非法指令
  • 删除 -DRVV_SPEC_0_7,开启 ncnn 的 rvv-1.0 intrinsic 代码
  • 删除 -D__riscv_zfh=1,arch 参数的 zfh 中已经指代
放在 ncnn/toolchains/c906-v222.toolchain.cmake
0x2 工具链修复
因为 rvv-0.7 缺少某些指令支持,遇到一些 rvv-1.0的代码会生成 unknown op
fnegfrec7frsqrt7
因此要修改下工具链头文件
玄铁-900-gcc-linux-5.10.4-glibc-x86_64-V2.2.2/lib/gcc/riscv64-unknown-linux-gnu
/10.2.0/include/riscv_vector.h
找到以下三行_RVV_FLOAT_ITERATOR_ARG (_RVV_FLOAT_UNARY_OP, rec7)_RVV_FLOAT_ITERATOR_ARG (_RVV_FLOAT_UNARY_OP, rsqrt7)_RVV_FLOAT_ITERATOR_ARG (_RVV_FLOAT_UNARY_OP, neg)注释掉// _RVV_FLOAT_ITERATOR_ARG (_RVV_FLOAT_UNARY_OP, rec7)// _RVV_FLOAT_ITERATOR_ARG (_RVV_FLOAT_UNARY_OP, rsqrt7)// _RVV_FLOAT_ITERATOR_ARG (_RVV_FLOAT_UNARY_OP, neg)找到文件末尾的三个 #endif,添加以下兼容代码,保存#endif#define vfneg_v_f32m1(x, vl) vfsgnjn_vv_f32m1(x, x, vl)#define vfneg_v_f32m2(x, vl) vfsgnjn_vv_f32m2(x, x, vl)#define vfneg_v_f32m4(x, vl) vfsgnjn_vv_f32m4(x, x, vl)#define vfneg_v_f32m8(x, vl) vfsgnjn_vv_f32m8(x, x, vl)#define vfneg_v_f16m1(x, vl) vfsgnjn_vv_f16m1(x, x, vl)#define vfneg_v_f16m2(x, vl) vfsgnjn_vv_f16m2(x, x, vl)#define vfneg_v_f16m4(x, vl) vfsgnjn_vv_f16m4(x, x, vl)#define vfneg_v_f16m8(x, vl) vfsgnjn_vv_f16m8(x, x, vl)#define vfrec7_v_f32m1(x, vl) vfrdiv_vf_f32m1(x, 1.f, vl)#define vfrec7_v_f32m2(x, vl) vfrdiv_vf_f32m2(x, 1.f, vl)#define vfrec7_v_f32m4(x, vl) vfrdiv_vf_f32m4(x, 1.f, vl)#define vfrec7_v_f32m8(x, vl) vfrdiv_vf_f32m8(x, 1.f, vl)#define vfrec7_v_f16m1(x, vl) vfrdiv_vf_f16m1(x, 1.f, vl)#define vfrec7_v_f16m2(x, vl) vfrdiv_vf_f16m2(x, 1.f, vl)#define vfrec7_v_f16m4(x, vl) vfrdiv_vf_f16m4(x, 1.f, vl)#define vfrec7_v_f16m8(x, vl) vfrdiv_vf_f16m8(x, 1.f, vl)#define vfrsqrt7_v_f32m1(x, vl) vfrdiv_vf_f32m1(vfsqrt_v_f32m1(x, vl), 1.f, vl)#define vfrsqrt7_v_f32m2(x, vl) vfrdiv_vf_f32m2(vfsqrt_v_f32m2(x, vl), 1.f, vl)#define vfrsqrt7_v_f32m4(x, vl) vfrdiv_vf_f32m4(vfsqrt_v_f32m4(x, vl), 1.f, vl)#define vfrsqrt7_v_f32m8(x, vl) vfrdiv_vf_f32m8(vfsqrt_v_f32m8(x, vl), 1.f, vl)#define vfrsqrt7_v_f16m1(x, vl) vfrdiv_vf_f16m1(vfsqrt_v_f16m1(x, vl), 1.f, vl)#define vfrsqrt7_v_f16m2(x, vl) vfrdiv_vf_f16m2(vfsqrt_v_f16m2(x, vl), 1.f, vl)#define vfrsqrt7_v_f16m4(x, vl) vfrdiv_vf_f16m4(vfsqrt_v_f16m4(x, vl), 1.f, vl)#define vfrsqrt7_v_f16m8(x, vl) vfrdiv_vf_f16m8(vfsqrt_v_f16m8(x, vl), 1.f, vl)#endif#endif0x3 下载和编译ncnn
这次可以用 release 编译啦!
git clone https://github.com/Tencent/ncnn.gitcd ncnnmkdir build-c906cd build-c906cmake -DCMAKE_TOOLCHAIN_FILE=../toolchains/c906-v222.toolchain.cmake -DCMAKE_BUILD_TYPE=release -DNCNN_OPENMP=OFF -DNCNN_THREADS=OFF -DNCNN_RUNTIME_CPU=OFF -DNCNN_RVV=ON -DNCNN_SIMPLEOCV=ON -DNCNN_BUILD_EXAMPLES=ON ..make -j320x4 新旧工具链的性能测试对比(耗时)

使用特权

评论回复

相关帖子

发新帖 我要提问
您需要登录后才可以回帖 登录 | 注册

本版积分规则

282

主题

290

帖子

1

粉丝