Browse Source

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

GitOrigin-RevId: 1b9a80b04e
release-1.7
Megvii Engine Team 3 years ago
parent
commit
605836a70f
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/tensorrt/test/tensorrt_runtime.cpp

+ 1
- 1
src/tensorrt/test/tensorrt_runtime.cpp View File

@@ -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);


Loading…
Cancel
Save