From a059bbb3c9d145bbcc93351d35b098a0b9d625b9 Mon Sep 17 00:00:00 2001 From: David OK Date: Sun, 17 Dec 2023 16:58:17 +0000 Subject: [PATCH] MAINT: defer the porting of YOLO-v4 to TensorRT in another PR. --- .../TensorRT/test_neuralnetworks_tensorrt_darknet_parser.cpp | 5 +---- 1 file changed, 1 insertion(+), 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 index d75a8c5a0..118c751b2 100644 --- 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_; 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.