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