fix sfpi issue

Author: ambroseling

install_tt.sh

@@ -31,8 +31,59 @@ apt-get update && apt-get install -y \
     libapt-pkg-dev \
     build-essential
 
+#
+
+# ----------------------------
+# Upgrade pip and install gdown if missing
 # ----------------------------
-# Install Tenstorrent SFPI runtime (.deb)
+python -m pip install --upgrade pip
+if ! python -m pip show gdown &> /dev/null; then
+    python -m pip install gdown --no-cache-dir
+fi
+
+# ----------------------------
+# Download & unzip Google Drive files
+# ----------------------------
+download_and_unzip() {
+    FILE_ID="$1"
+    DEST="$2"
+    ZIP_NAME="${DEST}.zip"
+
+    remove_path "$DEST"
+    remove_path "$ZIP_NAME"
+
+    echo "Downloading $ZIP_NAME..."
+    gdown "$FILE_ID" -O "$ZIP_NAME"
+
+    if unzip -tq "$ZIP_NAME" &> /dev/null; then
+        echo "Unzipping $ZIP_NAME..."
+        unzip -q "$ZIP_NAME"
+        rm "$ZIP_NAME"
+    else
+        echo "❌ $ZIP_NAME is not a valid zip file!"
+        exit 1
+    fi
+}
+
+download_and_unzip 1F2MJQ5enUPVtyi3s410PUuv8LiWr8qCz assets
+download_and_unzip 1LAOL8sYCUfsCk3TEA3vvyJCLSl0EdwYB attacks
+
+# ----------------------------
+# Install Python dependencies
+# ----------------------------
+if [ -f requirements.txt ]; then
+    echo "Installing Python dependencies..."
+    python3 -m pip install -r requirements.txt
+else
+    echo "No requirements.txt found. Skipping Python dependency installation."
+fi
+
+python -m pip install ttnn
+
+python -m pip install numpy==2.1.1
+
+# ----------------------------
+# Install Tenstorrent SFPI runtime (.deb) AFTER ttnn, then normalize paths
 # ----------------------------
 SFPI_DEB_URL="https://github.com/tenstorrent/sfpi/releases/download/v6.21.0/sfpi_6.21.0_x86_64.deb"
 SFPI_DEB_PATH="/tmp/sfpi_6.21.0_x86_64.deb"
@@ -41,10 +92,7 @@ echo "Downloading SFPI package from: $SFPI_DEB_URL"
 curl -L "$SFPI_DEB_URL" -o "$SFPI_DEB_PATH"
 
 echo "Installing SFPI package..."
-# Use dpkg then fix any missing dependencies
 dpkg -i "$SFPI_DEB_PATH" || apt-get -f install -y
-# Optionally, re-run dpkg if needed (usually not required after -f install)
-# dpkg -i "$SFPI_DEB_PATH" || true
 
 # Normalize SFPI location and ensure both expected lookup paths exist without symlink loops
 SITE_PKGS="$(python - <<'PY'
@@ -95,14 +143,12 @@ echo "Python runtime target: ${SITE_SFPI:-<none>}"
 
 # If SFPI located inside site-packages, move it to canonical location to avoid symlink loops
 if [ -n "$SITE_SFPI" ] && [ "$SFPI_SRC_DIR" = "$SITE_SFPI" ]; then
-    # Resolve symlink if site sfpi is a link
     if [ -L "$SITE_SFPI" ]; then
         RESOLVED="$(readlink -f "$SITE_SFPI" || true)"
         if [ -n "$RESOLVED" ] && [ -d "$RESOLVED" ]; then
             SFPI_SRC_DIR="$RESOLVED"
         fi
     fi
-    # If still the same path (real dir), move to /opt
     if [ "$SFPI_SRC_DIR" = "$SITE_SFPI" ]; then
         rm -rf "$SFPI_CANON"
         mkdir -p "$(dirname "$SFPI_CANON")"
@@ -128,55 +174,19 @@ if [ -n "$SITE_TTNN_RUNTIME" ]; then
     echo "SFPI linked into Python runtime at: $SITE_SFPI -> $SFPI_CANON"
 fi
 
-# ----------------------------
-# Upgrade pip and install gdown if missing
-# ----------------------------
-python -m pip install --upgrade pip
-if ! python -m pip show gdown &> /dev/null; then
-    python -m pip install gdown --no-cache-dir
+# Verify presence
+if [ ! -d "$SFPI_CANON" ]; then
+    echo "❌ SFPI canonical directory missing: $SFPI_CANON"
+    exit 1
 fi
-
-# ----------------------------
-# Download & unzip Google Drive files
-# ----------------------------
-download_and_unzip() {
-    FILE_ID="$1"
-    DEST="$2"
-    ZIP_NAME="${DEST}.zip"
-
-    remove_path "$DEST"
-    remove_path "$ZIP_NAME"
-
-    echo "Downloading $ZIP_NAME..."
-    gdown "$FILE_ID" -O "$ZIP_NAME"
-
-    if unzip -tq "$ZIP_NAME" &> /dev/null; then
-        echo "Unzipping $ZIP_NAME..."
-        unzip -q "$ZIP_NAME"
-        rm "$ZIP_NAME"
-    else
-        echo "❌ $ZIP_NAME is not a valid zip file!"
-        exit 1
-    fi
-}
-
-download_and_unzip 1F2MJQ5enUPVtyi3s410PUuv8LiWr8qCz assets
-download_and_unzip 1LAOL8sYCUfsCk3TEA3vvyJCLSl0EdwYB attacks
-
-# ----------------------------
-# Install Python dependencies
-# ----------------------------
-if [ -f requirements.txt ]; then
-    echo "Installing Python dependencies..."
-    python3 -m pip install -r requirements.txt
-else
-    echo "No requirements.txt found. Skipping Python dependency installation."
+if [ -n "$SITE_SFPI" ] && [ ! -e "$SITE_SFPI/compiler/bin/riscv-tt-elf-g++" ]; then
+    echo "❌ SFPI compiler not found at $SITE_SFPI/compiler/bin/riscv-tt-elf-g++"
+    echo "Contents of $SITE_SFPI:"
+    ls -la "$SITE_SFPI" || true
+    echo "Contents of $SFPI_CANON:"
+    ls -la "$SFPI_CANON" || true
+    exit 1
 fi
 
-python -m pip install ttnn
-
-python -m pip install numpy==2.1.1
-
-
 
 echo "✅ Installation complete."