diff --git a/paddle/fluid/inference/tensorrt/convert/scale_op.cc b/paddle/fluid/inference/tensorrt/convert/scale_op.cc index abf41cd72591b971f2475cfce6671845bcff0e83..841451a2b3122f16bb7804ad492a7829735ac50d 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