MAINT: defer the porting of YOLO-v4 to TensorRT in another PR.

oddkiva · Dec 17, 2023 · a059bbb · a059bbb
1 parent dac5c52
commit a059bbb
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/cpp/test/Shakti/Cuda/TensorRT/test_neuralnetworks_tensorrt_darknet_parser.cpp b/cpp/test/Shakti/Cuda/TensorRT/test_neuralnetworks_tensorrt_darknet_parser.cpp
@@ -40,7 +40,6 @@ using PinnedTensor = sara::Tensor_<T, N, shakti::PinnedMemoryAllocator>;
 
 BOOST_AUTO_TEST_SUITE(TestTensorRT)
 
-#if defined(TEST_YOLOV4_TINY)
 BOOST_AUTO_TEST_CASE(test_yolo_v4_tiny_conversion)
 {
   // Instantiate a network and automatically manage its memory.
@@ -221,9 +220,9 @@ BOOST_AUTO_TEST_CASE(test_yolo_v4_tiny_conversion)
     std::cout << "out 1 =\n" << u_out_tensor[1][0].matrix() << std::endl;
   }
 }
-#endif
 
 
+#if defined(YOLO_V4_TRT_WORKS)
 auto get_yolov4_model() -> d::Network
 {
   // Load the network on the host device (CPU).
@@ -405,8 +404,6 @@ BOOST_AUTO_TEST_CASE(test_yolo_v4_check_each_unary_layer_individually)
   }
 }
 
-
-#if defined(SEE_LATER)
 BOOST_AUTO_TEST_CASE(test_yolo_v4_conversion_incrementally_and_exhaustively)
 {
   // Get my CPU inference implementation of YOLO v4.