未验证 提交 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 {
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
}
......
......@@ -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 = [
{
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册