[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/csrc/cpu/interface.cpp
+++ b/sgl-kernel/csrc/cpu/interface.cpp
@@ -1,5 +1,5 @@
 #include <ATen/record_function.h>
-#include <torch/extension.h>
+#include <torch/all.h>

 #include "shm.h"