提交 605836a7 编写于 作者: M Megvii Engine Team

fix(mgb): fix trt5.1 test can not create FlattenConcatPlugin

GitOrigin-RevId: 1b9a80b04efd8ac41cc0948b400ae12b9a285009
上级 11d75fec
......@@ -243,7 +243,6 @@ TEST(TestOprTensorRT, IOFormatFree) {
func2->execute();
MGB_ASSERT_TENSOR_EQ(y1, y2);
}
#endif
TEST(TestOprTensorRT, FlattenConcatPlugin) {
REQUIRE_GPU(1);
......@@ -272,6 +271,7 @@ TEST(TestOprTensorRT, FlattenConcatPlugin) {
func->execute();
MGB_ASSERT_TENSOR_EQ(host_z1, host_z2);
}
#endif
TEST(TestOprTensorRT, ICudaEngine) {
REQUIRE_GPU(1);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册