From 47ccfccf7af172dfb0c666dce0e795705d360cde Mon Sep 17 00:00:00 2001
From: Ettore Di Giacinto <mudler@users.noreply.github.com>
Date: Fri, 25 Jul 2025 22:45:09 +0200
Subject: [PATCH] fix(ci): add nvidia-l4t capability to l4t images (#5914)

Signed-off-by: Ettore Di Giacinto <mudler@users.noreply.github.com>
---
 Dockerfile | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 4e8e29cb2..fa885dd34 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -72,6 +72,12 @@ RUN <<EOT bash
     fi
 EOT
 
+RUN <<EOT bash
+    if [ "${BUILD_TYPE}" = "cublas" ] && [ "${TARGETARCH}" = "arm64" ]; then
+        echo "nvidia-l4t" > /run/localai/capability
+    fi
+EOT
+
 # If we are building with clblas support, we need the libraries for the builds
 RUN if [ "${BUILD_TYPE}" = "clblas" ] && [ "${SKIP_DRIVERS}" = "false" ]; then \
         apt-get update && \