test: cat tensorop GEMM example + arch.h + cutlass.h from corex-samples
This commit is contained in:
43
qwen3_6_scripts/cat_cutlass_cu10_part2.sh
Executable file
43
qwen3_6_scripts/cat_cutlass_cu10_part2.sh
Executable file
@@ -0,0 +1,43 @@
|
||||
#!/bin/bash
|
||||
# cat_cutlass_cu10_part2.sh — Cat tensorop examples and arch files
|
||||
SAMPLES="/usr/local/corex-samples-3.2.3_x86_64/samples/cutlass"
|
||||
OUTDIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)/cat_files"
|
||||
mkdir -p "$OUTDIR"
|
||||
|
||||
cp "${SAMPLES}/examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu" "$OUTDIR/turing_tensorop_gemm.cu"
|
||||
echo "✓ turing_tensorop_gemm.cu"
|
||||
|
||||
cp "${SAMPLES}/examples/00_basic_gemm/basic_gemm.cu" "$OUTDIR/basic_gemm.cu"
|
||||
echo "✓ basic_gemm.cu"
|
||||
|
||||
cp "${SAMPLES}/include/cutlass/arch/arch.h" "$OUTDIR/arch.h" 2>/dev/null
|
||||
echo "✓ arch.h"
|
||||
|
||||
# Get the Cu10 arch tag definition
|
||||
grep -rl "struct Cu10" "${SAMPLES}/include/" 2>/dev/null | while read f; do
|
||||
base=$(basename "$f")
|
||||
cp "$f" "$OUTDIR/arch_${base}"
|
||||
echo "✓ arch_${base} (contains Cu10 definition)"
|
||||
done
|
||||
|
||||
# Get the cutlass.h to see CUTLASS_ARCH_CU10_SUPPORTED
|
||||
cp "${SAMPLES}/include/cutlass/cutlass.h" "$OUTDIR/cutlass.h"
|
||||
echo "✓ cutlass.h"
|
||||
|
||||
# Get gemm_batched.h full (we only had head before)
|
||||
cp "${SAMPLES}/include/cutlass/gemm/device/gemm_batched.h" "$OUTDIR/gemm_batched_full.h"
|
||||
echo "✓ gemm_batched_full.h"
|
||||
|
||||
# Get gemm.h (device level)
|
||||
cp "${SAMPLES}/include/cutlass/gemm/device/gemm.h" "$OUTDIR/gemm_device.h"
|
||||
echo "✓ gemm_device.h"
|
||||
|
||||
# Get the CMakeLists for batched_gemm and tensorop examples
|
||||
cp "${SAMPLES}/examples/05_batched_gemm/CMakeLists.txt" "$OUTDIR/CMakeLists_batched_gemm.txt"
|
||||
cp "${SAMPLES}/examples/08_turing_tensorop_gemm/CMakeLists.txt" "$OUTDIR/CMakeLists_tensorop_gemm.txt"
|
||||
echo "✓ CMakeLists"
|
||||
|
||||
echo ""
|
||||
ls -lh "$OUTDIR/"
|
||||
echo ""
|
||||
echo "git add cat_files/ && git commit -m 'data: Cu10 CUTLASS part 2' && git push"
|
||||
Reference in New Issue
Block a user