|
|
|
@ -392,7 +392,7 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor<
|
|
|
|
|
AnalysisConfig, PaddleEngineKind::kAnalysis>(const AnalysisConfig &config) {
|
|
|
|
|
VLOG(3) << "create AnalysisConfig";
|
|
|
|
|
if (config.use_gpu()) {
|
|
|
|
|
// 1. GPU memeroy
|
|
|
|
|
// 1. GPU memory
|
|
|
|
|
PADDLE_ENFORCE_GT(config.memory_pool_init_size_mb(), 0.f);
|
|
|
|
|
PADDLE_ENFORCE_GE(config.gpu_device_id(), 0, "Invalid device id %d",
|
|
|
|
|
config.gpu_device_id());
|
|
|
|
|