[Kernel] Move sgl-kernel under sglang.kernels.aot (#32648)
This commit is contained in:
@@ -50,11 +50,11 @@ cd sglang
|
||||
|
||||
# Compile sgl-kernel
|
||||
pip install --upgrade pip
|
||||
cd sgl-kernel
|
||||
cd python/sglang/kernels/aot
|
||||
python setup_rocm.py install
|
||||
|
||||
# Install sglang python package along with diffusion support
|
||||
cd ..
|
||||
cd ../../../..
|
||||
rm -rf python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml
|
||||
pip install -e "python[all_hip]"
|
||||
```
|
||||
|
||||
@@ -36,7 +36,7 @@ source sglang-metal/bin/activate
|
||||
|
||||
# (Optional) Compile sgl-kernel
|
||||
uv pip install --upgrade pip
|
||||
uv run sgl-kernel/setup_metal.py install
|
||||
uv run python/sglang/kernels/aot/setup_metal.py install
|
||||
|
||||
# Install sglang python package along with diffusion support
|
||||
rm -f python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml
|
||||
|
||||
@@ -126,7 +126,7 @@ uv pip install --upgrade pip setuptools
|
||||
uv pip install .
|
||||
|
||||
# Build the CPU backend kernels
|
||||
cd ../sgl-kernel
|
||||
cd sglang/kernels/aot
|
||||
cp pyproject_cpu.toml pyproject.toml
|
||||
uv pip install .
|
||||
```
|
||||
|
||||
@@ -19,11 +19,11 @@ cd sglang
|
||||
|
||||
# Compile sgl-kernel
|
||||
pip install --upgrade pip
|
||||
cd sgl-kernel
|
||||
cd python/sglang/kernels/aot
|
||||
python setup_musa.py install
|
||||
|
||||
# Install sglang python package along with diffusion support
|
||||
cd ..
|
||||
cd ../../../..
|
||||
rm -f python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml
|
||||
pip install -e "python[all_musa]"
|
||||
```
|
||||
|
||||
Reference in New Issue
Block a user