From fce289cad9777b769e22a5d5df926995eb2d4079 Mon Sep 17 00:00:00 2001
From: Roman Meshko <roman.meshko@gmail.com>
Date: Sun, 7 Jun 2026 14:47:14 +0300
Subject: [PATCH] Fixed pipeline

---
 .woodpecker/e2e-convert-jetson.yml   | 8 ++++++--
 e2e/docker-compose.test.yml          | 2 +-
 e2e/scripts/publish_jetson_engine.sh | 9 +++++++++
 3 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/.woodpecker/e2e-convert-jetson.yml b/.woodpecker/e2e-convert-jetson.yml
index 2fa667c..625f52c 100644
--- a/.woodpecker/e2e-convert-jetson.yml
+++ b/.woodpecker/e2e-convert-jetson.yml
@@ -34,7 +34,11 @@ steps:
           echo "ERROR: fixtures/models/azaion.onnx is missing; cannot convert Jetson engine"
           exit 1
         fi
-        ls -lh fixtures/image_small.jpg fixtures/models/azaion.onnx
+        if [ ! -f fixtures/models/azaion.int8_calib.cache ]; then
+          echo "ERROR: fixtures/models/azaion.int8_calib.cache is missing; cannot verify INT8 Jetson conversion"
+          exit 1
+        fi
+        ls -lh fixtures/image_small.jpg fixtures/models/azaion.onnx fixtures/models/azaion.int8_calib.cache
 
         E2E_PROFILE=jetson \
         E2E_WAIT_FOR_ENGINE_ENABLED=1 \
@@ -48,6 +52,6 @@ steps:
           exit 1
         fi
 
-        bash scripts/publish_jetson_engine.sh
+        REQUIRE_INT8_ENGINE=1 bash scripts/publish_jetson_engine.sh
     volumes:
       - /var/run/docker.sock:/var/run/docker.sock
diff --git a/e2e/docker-compose.test.yml b/e2e/docker-compose.test.yml
index 83da3fb..7d0b18c 100644
--- a/e2e/docker-compose.test.yml
+++ b/e2e/docker-compose.test.yml
@@ -65,7 +65,7 @@ services:
       - jetson
     build:
       context: ..
-      dockerfile: Dockerfile.jetson
+      dockerfile: ${JETSON_DOCKERFILE:-Dockerfile.jetson}
     runtime: nvidia
     depends_on:
       - mock-loader
diff --git a/e2e/scripts/publish_jetson_engine.sh b/e2e/scripts/publish_jetson_engine.sh
index 1fc64a5..2fcad0a 100644
--- a/e2e/scripts/publish_jetson_engine.sh
+++ b/e2e/scripts/publish_jetson_engine.sh
@@ -28,6 +28,15 @@ if [[ "$engine_count" == "0" ]]; then
   exit 1
 fi
 
+if [[ "${REQUIRE_INT8_ENGINE:-0}" == "1" ]]; then
+  int8_engine_count="$(find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.int8.engine' | wc -l | tr -d ' ')"
+  if [[ "$int8_engine_count" == "0" ]]; then
+    echo "ERROR: INT8 engine is required, but no azaion*.int8.engine file was produced"
+    find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;
+    exit 1
+  fi
+fi
+
 echo "--- Converted TensorRT engine files:"
 find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;