Skip to content

Commit d7855fe

Browse files
author
feng_shuai
authored
cherry-pick:Some version of TensorRT don't support qkv_plugin (#49425)
* cherry-pick:Some version of TensorRT don't support qkv_plugin * cherry-pick:support coverage CI
1 parent 6e154fc commit d7855fe

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

paddle/fluid/inference/tensorrt/op_teller.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1708,8 +1708,10 @@ struct SimpleOpTypeSetTeller : public Teller {
17081708
return false;
17091709
}
17101710
} else {
1711-
#if !IS_TRT_VERSION_GE(8000)
1712-
VLOG(3) << "The version of TRT must be greater than 8000";
1711+
#if (IS_TRT_VERSION_GE(8000) && IS_TRT_VERSION_LT(8100)) || \
1712+
(IS_TRT_VERSION_LT(7200))
1713+
VLOG(3) << "There are some bugs in v8.0.* and the versions lower than "
1714+
"v7.2 are not supported";
17131715
return false;
17141716
#endif
17151717
}

python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_multihead_matmul.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -808,7 +808,7 @@ def generate_weight2():
808808

809809
for batch in [2, 4]:
810810
self.batch = batch
811-
for length in [64, 384]:
811+
for length in [197]:
812812
self.length = length
813813
ops_config = [
814814
{

0 commit comments

Comments
 (0)