File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed
paddle/fluid/inference/tensorrt
python/paddle/fluid/tests/unittests/ir/inference Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -1708,8 +1708,10 @@ struct SimpleOpTypeSetTeller : public Teller {
1708
1708
return false ;
1709
1709
}
1710
1710
} 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" ;
1713
1715
return false ;
1714
1716
#endif
1715
1717
}
Original file line number Diff line number Diff line change @@ -808,7 +808,7 @@ def generate_weight2():
808
808
809
809
for batch in [2 , 4 ]:
810
810
self .batch = batch
811
- for length in [64 , 384 ]:
811
+ for length in [197 ]:
812
812
self .length = length
813
813
ops_config = [
814
814
{
You can’t perform that action at this time.
0 commit comments