[CPU] Fix build issue (#6419)

2025-05-22 02:17:10 +08:00
parent d4c038daed
commit cfe48c5902
14 changed files with 157 additions and 143 deletions
--- a/sgl-kernel/csrc/cpu/shm.h
+++ b/sgl-kernel/csrc/cpu/shm.h
@@ -1,4 +1,4 @@
-#include <torch/torch.h>
+#include <torch/all.h>

 #include <torch/csrc/distributed/c10d/ProcessGroup.hpp>