d9c363f86f
- Add support for legacy CUDA versions; now compatible with CUDA 12.3 and newer - Add support for NVRTC compilation - Other fixes and code refactoring
26 lines
713 B
Bash
Executable File
26 lines
713 B
Bash
Executable File
# 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" .
|
|
else
|
|
echo "Error: No SO file found in build directory" >&2
|
|
exit 1
|
|
fi
|
|
|
|
# Open users' original directory
|
|
cd "$original_dir"
|