diff --git a/paddle/fluid/inference/tensorrt/convert/hard_sigmoid_op.cc b/paddle/fluid/inference/tensorrt/convert/hard_sigmoid_op.cc index 529c845a12a7d95b558ded7ec4d80e3624ee1b79..3b6e464a117d3eedb4848dcaaf5c06fdb453fac1 100644 --- a/paddle/fluid/inference/tensorrt/convert/hard_sigmoid_op.cc +++ b/paddle/fluid/inference/tensorrt/convert/hard_sigmoid_op.cc @@ -25,7 +25,7 @@ class HardSigmoidOpConverter : public OpConverter { public: void operator()(const framework::proto::OpDesc& op, const framework::Scope& scope, bool test_mode) override { -#if IS_TRT_VERSION_GE(5000) +#if IS_TRT_VERSION_GE(5130) VLOG(3) << "convert a fluid HardSigmoid op to tensorrt IActivationLayer " "layer without bias"; framework::OpDesc op_desc(op, nullptr);