From 605836a70f0874eb405cb6e2272bb38b1fa450ba Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sat, 27 Nov 2021 00:29:51 +0800
Subject: [PATCH] fix(mgb): fix trt5.1 test can not create FlattenConcatPlugin

GitOrigin-RevId: 1b9a80b04efd8ac41cc0948b400ae12b9a285009
---
 src/tensorrt/test/tensorrt_runtime.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tensorrt/test/tensorrt_runtime.cpp b/src/tensorrt/test/tensorrt_runtime.cpp
index 69714042..a3362e09 100644
--- a/src/tensorrt/test/tensorrt_runtime.cpp
+++ b/src/tensorrt/test/tensorrt_runtime.cpp
@@ -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);