From 3dc1bb6a35485949699b429acb675705dcd6b486 Mon Sep 17 00:00:00 2001
From: Daniel Hiltgen <daniel@ollama.com>
Date: Sun, 10 Mar 2024 14:45:38 -0700
Subject: [PATCH] Harden for deps file being empty (or short)

---
 llm/generate/gen_linux.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/llm/generate/gen_linux.sh b/llm/generate/gen_linux.sh
index 0b8cb344..c182b2a5 100755
--- a/llm/generate/gen_linux.sh
+++ b/llm/generate/gen_linux.sh
@@ -194,6 +194,12 @@ if [ -d "${ROCM_PATH}" ]; then
     for dep in $(ldd "${BUILD_DIR}/lib/libext_server.so" | grep "=>" | cut -f2 -d= | cut -f2 -d' ' | grep -e rocm -e amdgpu -e libtinfo ); do
         echo "${dep}" >> "${BUILD_DIR}/lib/deps.txt"
     done
+    # bomb out if for some reason we didn't get a few deps
+    if [ $(cat "${BUILD_DIR}/lib/deps.txt" | wc -l ) -lt 8 ] ; then
+        cat "${BUILD_DIR}/lib/deps.txt"
+        echo "ERROR: deps file short"
+        exit 1
+    fi
     compress_libs
 fi