From 9bcb4137cd0fc0ba58288ff00b66e3bea2df4780 Mon Sep 17 00:00:00 2001
From: Jim Wang <yutsung.jim.wang@intel.com>
Date: Wed, 14 Feb 2024 16:02:17 -0700
Subject: [PATCH] feat: add capi-yolov8 ensemble into smoke tests

Signed-off-by: Jim Wang <yutsung.jim.wang@intel.com>
---
 run_smoke_test.sh | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/run_smoke_test.sh b/run_smoke_test.sh
index 85f60b22..2dc9ff0d 100755
--- a/run_smoke_test.sh
+++ b/run_smoke_test.sh
@@ -280,6 +280,18 @@ waitForLogFile
 verifyNonEmptyPipelineLog capi_yolov5_ensemble $input_src
 teardown
 
+#11. gst capi capi_yolov8_ensemble profile:
+make build-capi_yolov8_ensemble
+echo "Running capi_yolov8_ensemble profile..."
+input_src="rtsp://127.0.0.1:8554/camera_0"
+PIPELINE_PROFILE="capi_yolov8_ensemble" RENDER_MODE=0 sudo -E ./run.sh --platform core --inputsrc "$input_src"
+status_code=$?
+verifyStatusCode capi_yolov8_ensemble $status_code $input_src
+# allowing some time to process
+waitForLogFile
+verifyNonEmptyPipelineLog capi_yolov8_ensemble $input_src
+teardown
+
 #-----------------------------------------------------------------------------------------------------------------
 # tests for running on device GPU
 #
@@ -317,6 +329,17 @@ waitForLogFile
 verifyNonEmptyPipelineLog capi_yolov5_ensemble_gpu $input_src
 teardown
 
+# capi_yolov8 ensemble
+echo "Running capi_yolov8_ensemble profile on GPU.0..."
+input_src="rtsp://127.0.0.1:8554/camera_0"
+PIPELINE_PROFILE="capi_yolov8_ensemble" DEVICE="GPU" RENDER_MODE=0 sudo -E ./run.sh --platform dgpu.0 --inputsrc "$input_src"
+status_code=$?
+verifyStatusCode capi_yolov8_ensemble_gpu $status_code $input_src
+# allowing some time to process
+waitForLogFile
+verifyNonEmptyPipelineLog capi_yolov8_ensemble_gpu $input_src
+teardown
+
 source benchmark-scripts/get-gpu-info.sh
 if [ "$HAS_ARC" != 1 ]
 then
@@ -357,3 +380,14 @@ verifyStatusCode capi_yolov5_ensemble_ARC_gpu $status_code $input_src
 waitForLogFile
 verifyNonEmptyPipelineLog capi_yolov5_ensemble_ARC_gpu $input_src
 teardown
+
+# capi_yolov8 ensemble
+echo "Running capi_yolov8_ensemble profile on GPU.1..."
+input_src="rtsp://127.0.0.1:8554/camera_0"
+PIPELINE_PROFILE="capi_yolov8_ensemble" DEVICE="GPU" RENDER_MODE=0 sudo -E ./run.sh --platform dgpu.1 --inputsrc "$input_src"
+status_code=$?
+verifyStatusCode capi_yolov8_ensemble_ARC_gpu $status_code $input_src
+# allowing some time to process
+waitForLogFile
+verifyNonEmptyPipelineLog capi_yolov8_ensemble_ARC_gpu $input_src
+teardown