[CPU] Add CMakeLists.txt for sgl-kernel (#6115)

2025-05-14 06:30:37 +08:00
parent 16267d4fa7
commit 0f75b907c6
6 changed files with 132 additions and 3 deletions
--- a/sgl-kernel/setup_cpu.py
+++ b/sgl-kernel/setup_cpu.py
@@ -64,6 +64,7 @@ sources = [
    "csrc/cpu/topk.cpp",
    "csrc/cpu/interface.cpp",
    "csrc/cpu/shm.cpp",
+    "csrc/cpu/rope.cpp",
    "csrc/cpu/torch_extension_cpu.cpp",
 ]