From f41900c06b79b0f2da3c63bdd0c00bfb5e738de9 Mon Sep 17 00:00:00 2001
From: dylan <dylan@project6.dev>
Date: Sat, 15 Aug 2026 12:02:23 +0000
Subject: [PATCH] fix(build): auto-find cutlass/cutlass.h under COREX_ROOT

---
 qwen3_6_scripts/build_corex_batched_gemm.sh | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/qwen3_6_scripts/build_corex_batched_gemm.sh b/qwen3_6_scripts/build_corex_batched_gemm.sh
index 6780517c..ade4fb86 100755
--- a/qwen3_6_scripts/build_corex_batched_gemm.sh
+++ b/qwen3_6_scripts/build_corex_batched_gemm.sh
@@ -20,7 +20,19 @@ fi
 
 CUTLASS_INCLUDE="$COREX_ROOT/samples/cutlass/include"
 if [ ! -d "$CUTLASS_INCLUDE/cutlass" ]; then
-    CUTLASS_INCLUDE="$COREX_ROOT/include"
+    CUTLASS_INCLUDE="$COREX_ROOT/include/cutlass"
+fi
+if [ ! -d "$CUTLASS_INCLUDE/cutlass" ] && [ -d "$CUTLASS_INCLUDE" ]; then
+    # Maybe cutlass.h is directly under this path
+    true
+fi
+# Last resort: find it
+if [ ! -f "$CUTLASS_INCLUDE/cutlass/cutlass.h" ] && [ ! -f "$CUTLASS_INCLUDE/cutlass.h" ]; then
+    CUTLASS_INCLUDE=$(find "$COREX_ROOT" -path "*/cutlass/cutlass.h" -printf '%h\n' 2>/dev/null | head -1 | sed 's|/cutlass$||')
+    if [ -z "$CUTLASS_INCLUDE" ]; then
+        echo "[build] ERROR: cannot find cutlass/cutlass.h under $COREX_ROOT"
+        exit 1
+    fi
 fi
 
 # Output path