From 181b1f5a305db8e5aee8f980657643f792202464 Mon Sep 17 00:00:00 2001 From: Pei Yang Date: Tue, 26 May 2020 20:14:07 +0800 Subject: [PATCH] adjust minimum trt version for hard_sigmoid converter to 5130. test=develop (#24746) --- paddle/fluid/inference/tensorrt/convert/hard_sigmoid_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/inference/tensorrt/convert/hard_sigmoid_op.cc b/paddle/fluid/inference/tensorrt/convert/hard_sigmoid_op.cc index 529c845a12a..3b6e464a117 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); -- GitLab