9da4a23561
* Add more GPU architectures support * Update layout.py * Optimize performance, Add SM90 support, Add 1D2D SM100 support * Add fmtlib submodule at commit 553ec11 --------- Co-authored-by: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
26 lines
707 B
Bash
Executable File
26 lines
707 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, 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"
|