diff --git a/fastdeploy/runtime/backends/paddle/option.h b/fastdeploy/runtime/backends/paddle/option.h index b05cc6f6f..134fc7ef0 100755 --- a/fastdeploy/runtime/backends/paddle/option.h +++ b/fastdeploy/runtime/backends/paddle/option.h @@ -56,6 +56,8 @@ struct PaddleBackendOption { bool enable_trt = false; /// Whether enable memory optimize, default true bool enable_memory_optimize = true; + /// Whether enable ir debug, default false + bool switch_ir_debug = false; /* * @brief IPU option, this will configure the IPU hardware, if inference model in IPU diff --git a/fastdeploy/runtime/backends/paddle/option_pybind.cc b/fastdeploy/runtime/backends/paddle/option_pybind.cc old mode 100644 new mode 100755 index 46e6a94b7..a08c9ab9c --- a/fastdeploy/runtime/backends/paddle/option_pybind.cc +++ b/fastdeploy/runtime/backends/paddle/option_pybind.cc @@ -41,7 +41,9 @@ void BindPaddleOption(pybind11::module& m) { .def_readwrite("enable_log_info", &PaddleBackendOption::enable_log_info) .def_readwrite("enable_mkldnn", &PaddleBackendOption::enable_mkldnn) .def_readwrite("enable_trt", &PaddleBackendOption::enable_trt) - .def_readwrite("enable_memory_optimize", &PaddleBackendOption::enable_memory_optimize) + .def_readwrite("enable_memory_optimize", + &PaddleBackendOption::enable_memory_optimize) + .def_readwrite("switch_ir_debug", &PaddleBackendOption::switch_ir_debug) .def_readwrite("ipu_option", &PaddleBackendOption::ipu_option) .def_readwrite("collect_trt_shape", &PaddleBackendOption::collect_trt_shape) diff --git a/fastdeploy/runtime/backends/paddle/paddle_backend.cc b/fastdeploy/runtime/backends/paddle/paddle_backend.cc old mode 100644 new mode 100755 index bfee4f618..562b1ce73 --- a/fastdeploy/runtime/backends/paddle/paddle_backend.cc +++ b/fastdeploy/runtime/backends/paddle/paddle_backend.cc @@ -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_);