[AMD][Diffusion] support timestep embedding kernel for AMD GPUs (#16766)

2026-01-12 22:17:07 -08:00
parent ff3ddb9d9b
commit 8716589826
9 changed files with 40 additions and 19 deletions
@@ -52,10 +52,10 @@ pip install --upgrade pip
 cd sgl-kernel
 python setup_rocm.py install

-# Install sglang python package
+# Install sglang python package along with diffusion support
 cd ..
 rm -rf python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml
-pip install -e "python[all_hip]"
+pip install -e "python[all_hip,diffusion]"
 ```

 ### Install Using Docker (Recommended)