[Graphbolt] Avoid transitive inclusion (include what you need) (#5759)

graphbolt/include/graphbolt/csc_sampling_graph.h

@@ -7,13 +7,12 @@
 #define GRAPHBOLT_CSC_SAMPLING_GRAPH_H_
 
 #include <graphbolt/sampled_subgraph.h>
+#include <graphbolt/shared_memory.h>
+#include <torch/torch.h>
 
 #include <string>
 #include <vector>
 
-#include "./sampled_subgraph.h"
-#include "./shared_memory.h"
-
 namespace graphbolt {
 namespace sampling {
 

graphbolt/include/graphbolt/serialize.h

@@ -7,13 +7,12 @@
 #ifndef GRAPHBOLT_SERIALIZE_H_
 #define GRAPHBOLT_SERIALIZE_H_
 
+#include <graphbolt/csc_sampling_graph.h>
 #include <torch/torch.h>
 
 #include <string>
 #include <vector>
 
-#include "csc_sampling_graph.h"
-
 /**
  * @brief Overload stream operator to enable `torch::save()` and `torch.load()`
  * for CSCSamplingGraph.

graphbolt/src/csc_sampling_graph.cc

@@ -6,6 +6,10 @@
 
 #include <graphbolt/csc_sampling_graph.h>
 #include <graphbolt/serialize.h>
+#include <torch/torch.h>
+
+#include <tuple>
+#include <vector>
 
 #include "./shared_memory_utils.h"
 

graphbolt/src/serialize.cc

@@ -5,6 +5,7 @@
  */
 
 #include <graphbolt/serialize.h>
+#include <torch/torch.h>
 
 namespace torch {
 

graphbolt/src/shared_memory_utils.cc

@@ -8,6 +8,12 @@
 
 #include <graphbolt/serialize.h>
 #include <graphbolt/shared_memory.h>
+#include <torch/torch.h>
+
+#include <cstring>
+#include <string>
+#include <tuple>
+#include <vector>
 
 namespace graphbolt {
 namespace sampling {

graphbolt/src/shared_memory_utils.h

@@ -13,6 +13,7 @@
 #include <memory>
 #include <sstream>
 #include <string>
+#include <tuple>
 #include <vector>
 
 namespace graphbolt {