diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index f4250173292443d2781b35b866334bd06e8bad4a..9c1939a064a6dd37404eaf7841d4f695ceca4aab 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -1708,8 +1708,10 @@ struct SimpleOpTypeSetTeller : public Teller { return false; } } else { -#if !IS_TRT_VERSION_GE(8000) - VLOG(3) << "The version of TRT must be greater than 8000"; +#if (IS_TRT_VERSION_GE(8000) && IS_TRT_VERSION_LT(8100)) || \ + (IS_TRT_VERSION_LT(7200)) + VLOG(3) << "There are some bugs in v8.0.* and the versions lower than " + "v7.2 are not supported"; return false; #endif } diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_multihead_matmul.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_multihead_matmul.py index 5bc980a227feac7432eab37620015e569c86b11f..9731b7667f793d21e3795dafb4574fa4e3908b60 100644 --- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_multihead_matmul.py +++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_multihead_matmul.py @@ -808,7 +808,7 @@ class TrtConvertVitToMultiHeadMatmulTest(TrtLayerAutoScanTest): for batch in [2, 4]: self.batch = batch - for length in [64, 384]: + for length in [197]: self.length = length ops_config = [ {