Embed DeepGEMM source (not submodule) for SM100 raw CUDA GEMM primitives
This commit is contained in:
25
third_party/DeepGEMM/develop.sh
vendored
Executable file
25
third_party/DeepGEMM/develop.sh
vendored
Executable 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"
|
||||
Reference in New Issue
Block a user