enhence error message

b750d6ed · zlsh80826 · 1cd61122 · b750d6ed
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/inference/tensorrt/convert/scale_op.cc paddle/fluid/inference/tensorrt/convert/scale_op.cc +2 -1

未找到文件。
--- 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