From 8f5233de10959d60b72d322bfb1da9494843f98e Mon Sep 17 00:00:00 2001
From: leejet <leejet714@gmail.com>
Date: Sun, 1 Nov 2026 13:25:35 +0800
Subject: [PATCH] fix: include backend extensions in formatting scripts

---
 scripts/format-code.ps1 | 15 +++++++++------
 scripts/format-code.sh  |  7 +++++--
 2 files changed, 14 insertions(+), 8 deletions(-)

diff --git a/scripts/format-code.ps1 b/scripts/format-code.ps1
index 91c5f3b0..479433b3 100644
--- a/scripts/format-code.ps1
+++ b/scripts/format-code.ps1
@@ -8,8 +8,17 @@ $patterns = @(
     "src/core/*.cpp"
     "src/core/*.h"
     "src/core/*.hpp"
+    "src/core/cpu/*.cpp"
+    "src/core/cpu/*.h"
+    "src/core/cuda/*.cu"
+    "src/core/cuda/*.cuh"
+    "src/core/cuda/*.h"
+    "src/core/cuda/sageattention/*.cuh"
     "src/core/probe/*.cpp"
     "src/core/probe/*.h"
+    "src/core/vulkan/*.cpp"
+    "src/core/vulkan/*.h"
+    "src/core/vulkan/shaders/*.comp"
     "src/extensions/*.cpp"
     "src/extensions/*.h"
     "src/extensions/*.hpp"
@@ -23,8 +32,6 @@ $patterns = @(
     "src/model/*/*.hpp"
     "src/tokenizers/*.h"
     "src/tokenizers/*.cpp"
-    "src/tokenizers/vocab/*.h"
-    "src/tokenizers/vocab/*.cpp"
     "src/model_io/*.h"
     "src/model_io/*.cpp"
     "examples/cli/*.cpp"
@@ -46,10 +53,6 @@ try {
         foreach ($file in $files) {
             $relativePath = $file.FullName.Substring($root.Length).TrimStart('\', '/') -replace '\\', '/'
 
-            if ($relativePath -like "vocab*") {
-                continue
-            }
-
             Write-Host "formatting '$relativePath'"
 
             # if ($relativePath -ne "stable-diffusion.h") {
diff --git a/scripts/format-code.sh b/scripts/format-code.sh
index 568a0e54..29b3313a 100644
--- a/scripts/format-code.sh
+++ b/scripts/format-code.sh
@@ -8,16 +8,19 @@ cd "$repo_root"
 for f in src/*.cpp src/*.h src/*.hpp \
          src/conditioning/*.cpp src/conditioning/*.h src/conditioning/*.hpp \
          src/core/*.cpp src/core/*.h src/core/*.hpp \
+         src/core/cpu/*.cpp src/core/cpu/*.h \
+         src/core/cuda/*.cu src/core/cuda/*.cuh src/core/cuda/*.h \
+         src/core/cuda/sageattention/*.cuh \
          src/core/probe/*.cpp src/core/probe/*.h \
+         src/core/vulkan/*.cpp src/core/vulkan/*.h src/core/vulkan/shaders/*.comp \
          src/extensions/*.cpp src/extensions/*.h src/extensions/*.hpp \
          src/pipeline/*.cpp src/pipeline/*.h \
          src/runtime/*.cpp src/runtime/*.h src/runtime/*.hpp \
          src/model/*/*.cpp src/model/*/*.h src/model/*/*.hpp \
-         src/tokenizers/*.h src/tokenizers/*.cpp src/tokenizers/vocab/*.h src/tokenizers/vocab/*.cpp \
+         src/tokenizers/*.h src/tokenizers/*.cpp \
          src/model_io/*.h src/model_io/*.cpp examples/cli/*.cpp examples/cli/*.h examples/server/*.cpp \
          examples/common/*.hpp examples/common/*.h examples/common/*.cpp; do
   [[ -e "$f" ]] || continue
-  [[ "$f" == vocab* ]] && continue
   echo "formatting '$f'"
   # if [ "$f" != "stable-diffusion.h" ]; then
   #   clang-tidy -fix -p build_linux/ "$f"