[CustomDevice] Not reset pass_builder (#56755)

220f13bd · ronnywang · GitHub · 0ce66c1c · 220f13bd
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/inference/api/analysis_predictor.cc paddle/fluid/inference/api/analysis_predictor.cc +2 -0

未找到文件。
--- a/paddle/fluid/inference/api/analysis_predictor.cc
+++ b/paddle/fluid/inference/api/analysis_predictor.cc
@@ -1571,6 +1571,8 @@ void AnalysisPredictor::PrepareArgument() {
        }
      } else if (config_.use_xpu()) {
        // All passes support fp16. Not reset pass_builder.
+      } else if (config_.use_custom_device()) {
+        // All passes support fp16. Not reset pass_builder.
      } else {
        pass_builder->ClearPasses();
      }