mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 09:07:10 +08:00
[Backend]Add switch_ir_debug for Paddle Backend (#1700)
* avoid mem copy for cpp benchmark * set CMAKE_BUILD_TYPE to Release * Add SegmentationDiff * change pointer to reference * fixed bug * cast uint8 to int32 * Add diff compare for OCR * Add diff compare for OCR * rm ppocr pipeline * Add yolov5 diff compare * Add yolov5 diff compare * deal with comments * deal with comments * fixed bug * fixed bug * fixed thread nums * Add Failed log * optimizer x86 pipeline * Add switch_ir_debug for paddle backend * fixed for ci
This commit is contained in:
4
fastdeploy/runtime/backends/paddle/paddle_backend.cc
Normal file → Executable file
4
fastdeploy/runtime/backends/paddle/paddle_backend.cc
Normal file → Executable file
@@ -24,6 +24,10 @@ void PaddleBackend::BuildOption(const PaddleBackendOption& option) {
|
||||
option_ = option;
|
||||
if (option.device == Device::GPU) {
|
||||
config_.EnableUseGpu(option.gpu_mem_init_size, option.device_id);
|
||||
if (option_.switch_ir_debug) {
|
||||
FDINFO << "Will Enable ir_debug for Paddle Backend." << std::endl;
|
||||
config_.SwitchIrDebug();
|
||||
}
|
||||
if (option_.external_stream_) {
|
||||
FDINFO << "Will use external stream for Paddle Backend." << std::endl;
|
||||
config_.SetExecStream(option_.external_stream_);
|
||||
|
Reference in New Issue
Block a user