From 8498ea4f10a44363f70a63c9293d92cd80fd801b Mon Sep 17 00:00:00 2001 From: feng_shuai Date: Mon, 12 Dec 2022 11:06:32 +0800 Subject: [PATCH] fix: there are some bugs with trt 8.0 (#48921) * fix: there are some bugs with trt 8.0 * fix:windows CI trt is too old --- paddle/fluid/inference/tensorrt/op_teller.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 4826b5ff27c..98d865247de 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -1899,8 +1899,9 @@ struct SimpleOpTypeSetTeller : public Teller { return false; } } else { -#if !IS_TRT_VERSION_GE(8100) - VLOG(3) << "The version of TRT must be greater than 8100"; +#if (IS_TRT_VERSION_GE(8000) && IS_TRT_VERSION_LT(8100)) || \ + (IS_TRT_VERSION_LT(7200)) + VLOG(3) << "There are some bugs with trt 8.0"; return false; #endif } -- GitLab