[WIP] Added sgDMA operator for scatter kvcache communication.

2025-12-24 23:48:52 +08:00
parent 6ec1b23982
commit cf5e7df093
9 changed files with 1061 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,5 +1,5 @@
 [build-system]
-requires = ["setuptools>=61"]
+requires = ["setuptools>=61", "torch>=2.4.0"]
 build-backend = "setuptools.build_meta"

 [project]