Add CUDA kernel size analysis tool for sgl-kernel optimization (#14544)

2025-12-07 15:29:41 +08:00
parent 41d61faa99
commit e5135b73f4
3 changed files with 283 additions and 1 deletions
@@ -279,7 +279,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
    wheel \
    scikit-build-core \
    nixl \
-    py-spy
+    py-spy \
+    cubloaty

 # Build and install sgl-model-gateway (install Rust, build, then remove to save space)
 RUN --mount=type=cache,target=/root/.cache/pip \