未验证 提交 d3259900 编写于 作者: C chalsliu 提交者: GitHub

Fix lite-subgraph xpu inference error (#54607)

上级 d0d7d01f
...@@ -179,7 +179,7 @@ void AnalysisConfig::EnableXpu(int l3_size, ...@@ -179,7 +179,7 @@ void AnalysisConfig::EnableXpu(int l3_size,
const std::string &transformer_encoder_precision, const std::string &transformer_encoder_precision,
bool transformer_encoder_adaptive_seqlen, bool transformer_encoder_adaptive_seqlen,
bool enable_multi_stream) { bool enable_multi_stream) {
#ifdef PADDLE_WITH_XPU #if defined(PADDLE_WITH_XPU) || defined(LITE_SUBGRAPH_WITH_XPU)
use_xpu_ = true; use_xpu_ = true;
xpu_config_.l3_size = l3_size; xpu_config_.l3_size = l3_size;
xpu_config_.conv_autotune_level = conv_autotune; xpu_config_.conv_autotune_level = conv_autotune;
...@@ -198,7 +198,8 @@ void AnalysisConfig::EnableXpu(int l3_size, ...@@ -198,7 +198,8 @@ void AnalysisConfig::EnableXpu(int l3_size,
Update(); Update();
#else #else
PADDLE_THROW(platform::errors::PreconditionNotMet( PADDLE_THROW(platform::errors::PreconditionNotMet(
"To use XPU inference, please compile with option 'WITH_XPU' first.")); "To use XPU inference, please compile with option 'WITH_XPU' or "
"'WITH_LITE & LITE_WITH_XPU' first."));
#endif #endif
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册