mirror of
https://github.com/kvcache-ai/sglang.git
synced 2026-06-30 19:57:52 +00:00
87 lines
3.3 KiB
Makefile
87 lines
3.3 KiB
Makefile
.PHONY: help check-deps install-deps tree ln submodule install build clean rebuild test format update
|
|
|
|
# ---------------------------
|
|
# Build resource controls
|
|
# ---------------------------
|
|
# By default, build uses all available CPU cores, but users can override:
|
|
# make build MAX_JOBS=2 CMAKE_BUILD_PARALLEL_LEVEL=2 CMAKE_ARGS="-DSGL_KERNEL_COMPILE_THREADS=1"
|
|
NPROC ?= $(shell nproc 2>/dev/null || echo 1)
|
|
MAX_JOBS ?= $(NPROC)
|
|
CMAKE_BUILD_PARALLEL_LEVEL ?= $(MAX_JOBS)
|
|
UV_BUILD_DIR ?= build
|
|
CMAKE_POLICY_VERSION_MINIMUM ?= 3.5
|
|
|
|
# Show help for each target
|
|
help: ## Show this help message
|
|
@echo "Available targets:"
|
|
@grep -E '^[a-zA-Z0-9_-]+:.*?## .*$$' $(MAKEFILE_LIST) | sort | awk 'BEGIN {FS = ":.*?## "}; {printf "\033[36m%-20s\033[0m %s\n", $$1, $$2}'
|
|
|
|
check-deps: ## Check and install required Python formatting dependencies
|
|
@command -v isort >/dev/null 2>&1 || (echo "Installing isort..." && pip install isort)
|
|
@command -v black >/dev/null 2>&1 || (echo "Installing black..." && pip install black)
|
|
|
|
install-deps: ## Install Python formatting tools (isort and black)
|
|
pip install scikit-build-core isort black
|
|
|
|
tree: ## Show project directory structure
|
|
@tree --prune -I "__pycache__|*.egg-info|*.so|build|3rdparty|dist"
|
|
|
|
submodule: ## Initialize and update git submodules
|
|
@git submodule update --init --recursive
|
|
|
|
ln: submodule ## Create compilation database
|
|
@rm -rf build && mkdir build && cd build && cmake .. -DCMAKE_EXPORT_COMPILE_COMMANDS=YES -DCMAKE_POLICY_VERSION_MINIMUM=3.5
|
|
|
|
install: submodule ## Install package in development mode
|
|
@pip install -e . --no-build-isolation
|
|
|
|
build: install-deps submodule ## Build and install wheel package
|
|
@rm -rf dist/* || true && \
|
|
CMAKE_POLICY_VERSION_MINIMUM=$(CMAKE_POLICY_VERSION_MINIMUM) \
|
|
MAX_JOBS=$(MAX_JOBS) \
|
|
CMAKE_BUILD_PARALLEL_LEVEL=$(CMAKE_BUILD_PARALLEL_LEVEL) \
|
|
CMAKE_ARGS="$(CMAKE_ARGS)" \
|
|
uv build --wheel -Cbuild-dir=$(UV_BUILD_DIR) . --verbose --color=always --no-build-isolation && \
|
|
pip3 install dist/*whl --force-reinstall --no-deps
|
|
|
|
clean: ## Remove build artifacts
|
|
@rm -rf build dist *.egg-info
|
|
|
|
rebuild: clean submodule build ## Clean and rebuild the project
|
|
@echo "Succeed to rebuild"
|
|
|
|
test: ## Run all tests
|
|
@find tests -name "test_*.py" | xargs -n 1 python3
|
|
|
|
format: check-deps ## Format all source files
|
|
@echo "Formatting source files..."
|
|
@find csrc tests -name '*.cc' -o -name '*.cu' -o -name '*.cuh' -o -name '*.h' -o -name '*.hpp' | xargs clang-format -i
|
|
@find python tests -name '*.py' | xargs isort
|
|
@find python tests -name '*.py' | xargs black
|
|
@pre-commit run --all-files
|
|
|
|
FILES_TO_UPDATE = python/sgl_kernel/version.py \
|
|
pyproject.toml \
|
|
pyproject_rocm.toml \
|
|
pyproject_cpu.toml
|
|
|
|
update: ## Update version numbers across project files. Usage: make update <new_version>
|
|
@if [ -z "$(filter-out $@,$(MAKECMDGOALS))" ]; then \
|
|
echo "Version required. Usage: make update <new_version>"; \
|
|
exit 1; \
|
|
fi
|
|
@OLD_VERSION=$$(grep "version" python/sgl_kernel/version.py | cut -d '"' -f2); \
|
|
NEW_VERSION=$(filter-out $@,$(MAKECMDGOALS)); \
|
|
echo "Updating version from $$OLD_VERSION to $$NEW_VERSION"; \
|
|
for file in $(FILES_TO_UPDATE); do \
|
|
if [ "$(shell uname)" = "Darwin" ]; then \
|
|
sed -i '' -e "s/$$OLD_VERSION/$$NEW_VERSION/g" $$file; \
|
|
else \
|
|
sed -i -e "s/$$OLD_VERSION/$$NEW_VERSION/g" $$file; \
|
|
fi \
|
|
done; \
|
|
echo "Version update complete"
|
|
|
|
%:
|
|
@:
|