Embed DeepGEMM source (not submodule) for SM100 raw CUDA GEMM primitives

This commit is contained in:
2026-06-01 07:39:40 +00:00
parent dae83723a3
commit e3ea609ddd
145 changed files with 27360 additions and 1 deletions

25
third_party/DeepGEMM/develop.sh vendored Executable file
View File

@@ -0,0 +1,25 @@
# Change current directory into project root
original_dir=$(pwd)
script_dir=$(realpath "$(dirname "$0")")
cd "$script_dir"
# Link CUTLASS includes
ln -sf $script_dir/third-party/cutlass/include/cutlass deep_gemm/include
ln -sf $script_dir/third-party/cutlass/include/cute deep_gemm/include
# Remove old dist file, build files, and build
rm -rf build dist
rm -rf *.egg-info
python setup.py build
# Find the .so file in build directory and create symlink in current directory
so_file=$(find build -name "*.so" -type f | head -n 1)
if [ -n "$so_file" ]; then
ln -sf "../$so_file" deep_gemm/
else
echo "Error: No SO file found in build directory" >&2
exit 1
fi
# Open users' original directory
cd "$original_dir"