dylanyunlong/project_6
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix(build): auto-find cutlass/cutlass.h under COREX_ROOT

Browse Source
This commit is contained in:
dylan
2026-08-15 12:02:23 +00:00
parent bfa18cd5b4
commit f41900c06b
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
qwen3_6_scripts/build_corex_batched_gemm.sh
View File

@@ -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