以下选项为benchmark相关的编译选项,在编译用来跑benchmark的sdk时,必须开启。
选项 | 需要设置的值 | 说明 |
---|---|---|
ENABLE_BENCHMARK | ON | 默认OFF, 是否打开BENCHMARK模式 |
ENABLE_VISION | ON | 默认OFF,是否编译集成视觉模型的部署模块 |
ENABLE_TEXT | ON | 默认OFF,是否编译集成文本NLP模型的部署模块 |
运行FastDeploy C++ Benchmark,需先准备好相应的环境,并在ENABLE_BENCHMARK=ON模式下从源码编译FastDeploy C++ SDK. 以下将按照硬件维度,来说明相应的系统环境要求。不同环境下的详细要求,请参考FastDeploy环境要求
具体flags.h提供选项如下:
选项 | 作用 |
---|---|
--model | 模型路径 |
--image | 图片路径 |
--config_path | config.txt路径,包含具体设备、后端等信息 |
具体config.txt包含信息含义如下:
参数 | 作用 |
---|---|
device | 选择 CPU/GPU/XPU,默认为 CPU |
device_id | GPU/XPU 卡号,默认为 0 |
cpu_thread_nums | CPU 线程数,默认为 1 |
warmup | 跑benchmark的warmup次数,默认为 200 |
repeat | 跑benchmark的循环次数,默认为 1000 |
backend | 指定后端类型,有default, ort, ov, trt, paddle, paddle_trt, lite 等,为default时,会自动选择最优后端,推荐设置为显式设置明确的backend。默认为 default |
profile_mode | 指定需要测试性能的模式,可选值为[runtime, end2end] ,默认为 runtime |
include_h2d_d2h | 是否把H2D+D2H的耗时统计在内,该参数只在profile_mode为runtime时有效,默认为 false |
use_fp16 | 是否开启fp16,当前只对 trt, paddle-trt, lite后端有效,默认为 false |
collect_memory_info | 是否记录 cpu/gpu memory信息,默认 false |
sampling_interval | 记录 cpu/gpu memory信息采样时间间隔,单位ms,默认为 50 |
precision_compare | 是否进行精度比较,默认为 false |
result_path | 记录 Benchmark 数据的 txt 文件路径 |
xpu_l3_cache | 设置XPU L3 Cache大小,默认值为0。设置策略,对于 昆仑2 XPU R200,L3 Cache可用的最大值为 62914560,对于 昆仑1 XPU 则为 16776192 |
Linux上编译需满足:
- gcc/g++ >= 5.4(推荐8.2)
- cmake >= 3.18.0
- CUDA >= 11.2
- cuDNN >= 8.2
- TensorRT >= 8.5
在GPU上编译FastDeploy需要准备好相应的CUDA环境以及TensorRT,详细文档请参考GPU编译文档。
# 源码编译SDK
git clone https://github.com/PaddlePaddle/FastDeploy.git -b develop
cd FastDeploy
mkdir build && cd build
cmake .. -DWITH_GPU=ON \
-DENABLE_ORT_BACKEND=ON \
-DENABLE_PADDLE_BACKEND=ON \
-DENABLE_OPENVINO_BACKEND=ON \
-DENABLE_TRT_BACKEND=ON \
-DENABLE_VISION=ON \
-DENABLE_TEXT=ON \
-DENABLE_BENCHMARK=ON \ # 开启benchmark模式
-DTRT_DIRECTORY=/Paddle/TensorRT-8.5.2.2 \
-DCUDA_DIRECTORY=/usr/local/cuda \
-DCMAKE_INSTALL_PREFIX=${PWD}/compiled_fastdeploy_sdk
make -j12
make install
# 配置SDK路径
cd ..
export FD_GPU_SDK=${PWD}/build/compiled_fastdeploy_sdk
cd benchmark/cpp
mkdir build && cd build
cmake .. -DFASTDEPLOY_INSTALL_DIR=${FD_GPU_SDK}
make -j4
在X86 CPU + NVIDIA GPU下,FastDeploy 目前支持多种推理后端,下面以 PaddleYOLOv8 为例,跑出多后端在 CPU/GPU 对应 benchmark 数据。
- 下载模型文件和测试图片
wget https://bj.bcebos.com/paddlehub/fastdeploy/yolov8_s_500e_coco.tgz
wget https://gitee.com/paddlepaddle/PaddleDetection/raw/release/2.4/demo/000000014439.jpg
tar -zxvf yolov8_s_500e_coco.tgz
- 运行 yolov8 benchmark 示例
# 统计性能,用户根据需求修改config.txt文件,具体含义参考上表
# eg:如果想测paddle gpu backend,将device改为gpu,backend修改为paddle即可
./benchmark_ppyolov8 --model yolov8_s_500e_coco --image 000000014439.jpg --config_path config.txt
注意,为避免对性能统计产生影响,测试性能时,最好不要开启内存显存统计的功能,当把collect_memory_info参数设置为true时,只有内存显存参数是稳定可靠的。更多参数设置,请参考参数设置说明
在准备好相关的环境配置和SDK后,可以使用本目录提供的脚本一键运行后的benchmark数据。
- 获取模型和资源文件
./get_models.sh
- 运行benchmark脚本
# x86 CPU Paddle backend fp32
./benchmark_x86.sh config/config.x86.paddle.fp32.txt
# x86 CPU ONNXRuntime backend fp32
./benchmark_x86.sh config/config.x86.ort.fp32.txt
# x86 CPU OpenVIVO backend fp32
./benchmark_x86.sh config/config.x86.ov.fp32.txt
# NVIDIA GPU Paddle backend fp32
./benchmark_gpu.sh config/config.gpu.paddle.fp32.txt
# NVIDIA GPU ONNXRuntime backend fp32
./benchmark_gpu.sh config/config.gpu.ort.fp32.txt
# NVIDIA GPU Paddle-TRT backend fp32
./benchmark_gpu_trt.sh config/config.gpu.paddle_trt.fp32.txt
# NVIDIA GPU Paddle-TRT backend fp16
./benchmark_gpu_trt.sh config/config.gpu.paddle_trt.fp16.txt
# NVIDIA GPU TRT backend fp32
./benchmark_gpu_trt.sh config/config.gpu.trt.fp32.txt
# NVIDIA GPU TRT backend fp16
./benchmark_gpu_trt.sh config/config.gpu.trt.fp16.txt
# Arm CPU Paddle Lite backend fp32
./benchmark_arm.sh config/config.arm.lite.fp32.txt
# Arm CPU Paddle Lite backend fp16
./benchmark_arm.sh config/config.arm.lite.fp16.txt
# XPU Paddle Lite backend fp32
./benchmark_xpu.sh config/config.xpu.lite.fp32.txt
FastDeploy除了提供包含模型前后处理在内的benchmark_xxx外,也提供常规的benchmark工具,以支持对任意模型进行benchmark。在编译benchmark目录的源码之后,会生成一个benchmark可执行文件,该工具支持选项设置说明中的所有参数,并且提供一些额外参数,便于使用,额外的参数说明如下。注意:该工具仅支持测试纯模型推理时间和推理+H2D+D2H耗时(当config.txt中include_h2d_d2h为true时),不支持测试包含前后处理在内的时间。
参数 | 作用 |
---|---|
shapes | Set input shape for model, default "1,3,224,224" |
names | Set input names for model, default "DEFAULT" |
dtypes | Set input dtypes for model, default "FP32" |
trt_shapes | Set min/opt/max shape for trt/paddle_trt backend. eg:--trt_shape 1,3,224,224:1,3,224,224:1,3,224,224", default "1,3,224,224:1,3,224,224:1,3,224,224" |
batch | TensorRT max batch size, default=1 |
dump | Whether to dump output tensors, default false. |
info | Only check the input infos of model, default false. |
diff | Check the diff between two tensors, default false. |
tensors | The paths to dumped tensors, should look like "tensor_a.txt:tensor_b.txt" |
mem | Whether to force to collect memory info, default false. |
model_file | Optional, set specific model file, eg, model.pdmodel, model.onnx, default "UNKNOWN" |
params_file | Optional, set specific params file, eg, model.pdiparams, default "" |
model_format | Optional, set specific model format, eg, PADDLE/ONNX/RKNN/TORCHSCRIPT/SOPHGO, default "PADDLE" |
disable_mkldnn | Whether to disable mkldnn for paddle backend, default false. |
- 用法说明
./benchmark --helpshort
benchmark: ./benchmark -[info|diff|check|dump|mem] -model xxx -config_path xxx -[shapes|dtypes|names|tensors] -[model_file|params_file|model_format]
...
- 单输入示例:--model,指定模型文件夹,其中包括*.pdmodel/pdiparams文件
./benchmark --model ResNet50_vd_infer --config_path config/config.x86.ov.fp32.txt --shapes 1,3,224,224 --names inputs --dtypes FP32
- 单输入示例:--model_file, --params_file,指定具体的模型文件和参数文件
./benchmark --model_file MobileNetV1_ssld_infer/inference.pdmodel --params_file MobileNetV1_ssld_infer/inference.pdiparams --config_path config/config.x86.ov.fp32.txt --shapes 1,3,224,224 --names inputs --dtypes FP32
- 多输入示例:
./benchmark --model yolov5_s_300e_coco --config_path config/config.arm.lite.fp32.txt --shapes 1,3,640,640:1,2 --names image:scale_factor --dtypes FP32:FP32
- Paddle-TRT示例 :
./benchmark --model ResNet50_vd_infer --config_path config/config.gpu.paddle_trt.fp16.txt --trt_shapes 1,3,224,224:1,3,224,224:1,3,224,224 --names inputs --dtypes FP32
- TensorRT/Paddle-TRT多输入示例:
./benchmark --model rtdetr_r50vd_6x_coco --trt_shapes 1,2:1,2:1,2:1,3,640,640:1,3,640,640:1,3,640,640:1,2:1,2:1,2 --names im_shape:image:scale_factor --shapes 1,2:1,3,640,640:1,2 --config_path config/config.gpu.paddle_trt.fp32.txt --dtypes FP32:FP32:FP32
- 支持FD全部后端和全部模型格式:--model_file, --params_file(optional), --model_format
# ONNX模型示例
./benchmark --model ResNet50_vd_infer --model_file inference.onnx --model_format ONNX --config_path config/config.gpu.trt.fp16.txt --trt_shapes 1,3,224,224:1,3,224,224:1,3,224,224 --names inputs --dtypes FP32
- 统计内显存占用:--mem 或 在config.txt中指定
./benchmark --mem --model ResNet50_vd_infer --config_path config/config.x86.ov.fp32.txt --shapes 1,3,224,224 --names inputs --dtypes FP32
- 推理并dump 输出 tensor用作对比: --dump
./benchmark --dump --model ResNet50_vd_infer --config_path config/config.x86.ov.fp32.txt --shapes 1,3,224,224 --names inputs --dtypes FP32
- 对比两个 dumped 的tensor : --diff
./benchmark --diff --tensors ov_linear_77.tmp_1.txt:lite_linear_77.tmp_1.txt
- 显示模型的输入信息: --info
./benchmark --info --model picodet_l_640_coco_lcnet --config_path config/config.arm.lite.fp32.txt