PaddleLite使用Bitmain:Sophon BM1682/BM1684 预测部署

Paddle Lite已支持在比特大陆的Sophon BM1682/BM1684处理器上进行预测部署. 目前支持子图接入

支持现状

已支持的芯片

  • Sophon BM1682

  • Sophon BM1684

已支持的设备

  • Sophon SC3 加速卡 (BM1682 X86 PCI-E)

  • Sophon SC5 加速卡 (BM1684 X86 PCI-E)

已支持的Paddle模型

已支持(或部分支持)的Paddle算子

  • norm

  • relu

  • leaky_relu

  • sqrt

  • square

  • sigmoid

  • assign_value

  • batch_norm

  • box_coder

  • cast

  • concat

  • conv2d

  • depthwise_conv2d

  • conv2d_transpose

  • depthwise_conv2d_transpose

  • density_prior_box

  • dropout

  • elementwise_add

  • elementwise_mul

  • elementwise_sub

  • elementwise_div

  • fill_constant

  • nearest_interp

  • bilinear_interp

  • matmul

  • mul

  • multiclass_nms

  • multiclass_nms2

  • pool2d

  • max_pool2d_with_index

  • prior_box

  • reduce_sum

  • reduce_mean

  • reduce_max

  • reshape

  • reshape2

  • flatten

  • flatten2

  • scale

  • shape

  • slice

  • softmax

  • split

  • squeeze

  • squeeze2

  • swish

  • transpose

  • yolo_box

参考示例演示

准备设备环境

  • 请确保您的 Sophon 加速卡已经可以在主机中正常工作.您可以对照下面的步骤快速验证, 如有任何问题, 请联系BITMAIN解决.

    • 可以使用BMNNSDK内附带的bm-smi程序进行测试,如果设备已经正常驱动,您应该能看到一个(或多个)PCIE模式的加速设备.

    • 可以使用ls /dev/bm*, 您应该能看到若干个bm前缀的设备.例如/dev/bmdev-ctl /dev/bm-sophon0

  • 简易安装指南(仅供参考,请以BITMAIN的安装指南为准)

    • 可以使用bmnnsdk_root/scripts/目录的install_libs.shsudo install_driver_pcie.sh完成安装.

    • 对于部分BM1684设备,为了启用动态编译功能,您可能需要为每个芯片启用icache.在bmnnsdk_root/bin/x86目录下,使用./test_update_fw ./bm168x_bmdnn_en_icache.bin ./bm168x_bmdnn_s_en_icache.bin chip_id来启用icache,chip_id为0,1,2等值,代表不同芯片.

准备本地编译环境

  • 目前仅在 Ubuntu 16.04 环境进行过测试,为了避免环境不一致带来的麻烦,建议使用Docker编译环境,请先根据编译环境准备下载好paddlepaddle/paddle-liteDocker镜像.

  • 在执行docker run启动容器时,请确保宿主机内/dev/bm*设备均被正确映射到容器中.可以参考下面的指令启动容器.

sudo docker run -it \
     --name work_bm \
     -v $HOME:/code \
     --device=/dev/bm1682-dev0:/dev/bm1682-dev0 \
     --device=/dev/bmdev-ctl:/dev/bmdev-ctl \
     --net=host \
     paddlepaddle/paddle-lite:latest /bin/bash

编译Paddle-Lite工程

  1. 下载代码

git clone https://github.com/PaddlePaddle/Paddle-Lite.git
  1. 编译

# 进入代码目录
cd Paddle-Lite

# 运行编译脚本
./lite/tools/build_bm.sh --target_name=BM1682
# 或 ./lite/tools/build_bm.sh --target_name=BM1684
  1. 编译结果说明 编译产物将输出至build.lite.bm/inference_lite_lib目录下,该目录结构为

.
|-- cxx 
|   |-- include #cxx头文件目录
|   `-- lib #cxx库目录
|       `-- third_party #第三方cxx依赖库目录
`-- demo
    `-- cxx # cxx示例程序

运行demo

# 测试demo在build.lite.bm/inference_lite_lib/demo/cxx/目录下
cd build.lite.bm/inference_lite_lib/demo/cxx/

wget http://paddle-inference-dist.bj.bcebos.com/mobilenet_v1.tar.gz

tar -xvf mobilenet_v1.tar.gz

./build.sh

./mobilenet_full_api ./mobilenet_v1 224 224

# 如果运行正常,程序最后会输出Done.