From b750d6ed2583dd7ef4b755e246617917ec98848c Mon Sep 17 00:00:00 2001 From: zlsh80826 Date: Wed, 5 Aug 2020 17:05:07 +0800 Subject: [PATCH] enhence error message --- paddle/fluid/inference/tensorrt/convert/scale_op.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/inference/tensorrt/convert/scale_op.cc b/paddle/fluid/inference/tensorrt/convert/scale_op.cc index abf41cd7259..841451a2b31 100644 --- a/paddle/fluid/inference/tensorrt/convert/scale_op.cc +++ b/paddle/fluid/inference/tensorrt/convert/scale_op.cc @@ -90,7 +90,8 @@ class ScaleOpConverter : public OpConverter { power_weights.get(), scale_weights.get(), power_weights.get()); } - PADDLE_ENFORCE_EQ(layer != nullptr, true); + PADDLE_ENFORCE_EQ(layer != nullptr, true, + platform::errors::Fatal("Create scale layer failed.")); if (idim.nbDims == 3) { // TensorRT scale layer is not supporting input dims < 4 when using -- GitLab