[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/torch_extension_cpu.cpp
+++ b/sgl-kernel/csrc/cpu/torch_extension_cpu.cpp
@@ -14,7 +14,7 @@ limitations under the License.
 ==============================================================================*/

 #include <ATen/ATen.h>
-#include <torch/extension.h>
+#include <torch/all.h>
 #include <torch/library.h>

 #include "shm.h"