未验证 提交 d7855fe8 编写于 作者: F feng_shuai 提交者: GitHub

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
上级 6e154fc6
...@@ -1708,8 +1708,10 @@ struct SimpleOpTypeSetTeller : public Teller { ...@@ -1708,8 +1708,10 @@ struct SimpleOpTypeSetTeller : public Teller {
return false; return false;
} }
} else { } else {
#if !IS_TRT_VERSION_GE(8000) #if (IS_TRT_VERSION_GE(8000) && IS_TRT_VERSION_LT(8100)) || \
VLOG(3) << "The version of TRT must be greater than 8000"; (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; return false;
#endif #endif
} }
......
...@@ -808,7 +808,7 @@ class TrtConvertVitToMultiHeadMatmulTest(TrtLayerAutoScanTest): ...@@ -808,7 +808,7 @@ class TrtConvertVitToMultiHeadMatmulTest(TrtLayerAutoScanTest):
for batch in [2, 4]: for batch in [2, 4]:
self.batch = batch self.batch = batch
for length in [64, 384]: for length in [197]:
self.length = length self.length = length
ops_config = [ ops_config = [
{ {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册