Move rope and bmm into sgl-kernel (#4241)

2025-03-09 18:38:15 -07:00
parent 9dfafa743c
commit eb06dbcbf8
5 changed files with 183 additions and 13 deletions
--- a/sgl-kernel/pyproject.toml
+++ b/sgl-kernel/pyproject.toml
@@ -1,5 +1,10 @@
 [build-system]
-requires = ["setuptools>=61.0", "wheel", "torch"]
+requires = [
+  "setuptools>=61.0",
+  "scikit-build-core>=0.10",
+  "torch==2.5.1",
+  "wheel",
+]
 build-backend = "setuptools.build_meta"

 [project]