Merge branch 'dmlc:master' into add-igb-to-sage

dmlc · Aug 26, 2024 · 10c86e6 · 10c86e6
2 parents 81ba919 + 0331009
commit 10c86e6
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/docs/source/api/python/dgl.graphbolt.rst b/docs/source/api/python/dgl.graphbolt.rst
@@ -56,9 +56,7 @@ collection of features.
     TorchBasedFeature
     TorchBasedFeatureStore
     DiskBasedFeature
-    cpu_cached_feature
     CPUCachedFeature
-    gpu_cached_feature
     GPUCachedFeature
 
 
@@ -186,6 +184,8 @@ Utilities
     :toctree: ../../generated/
     :nosignatures:
 
+    cpu_cached_feature
+    gpu_cached_feature
     fused_csc_sampling_graph
     load_from_shared_memory
     from_dglgraph
@@ -195,9 +195,10 @@ Utilities
     seed
     index_select
     expand_indptr
+    indptr_edge_ids
     add_reverse_edges
     exclude_seed_edges
     compact_csc_format
     unique_and_compact
     unique_and_compact_csc_formats
-
+    numpy_save_aligned
diff --git a/python/dgl/graphbolt/impl/cpu_cached_feature.py b/python/dgl/graphbolt/impl/cpu_cached_feature.py
@@ -81,7 +81,7 @@ def read_async(self, ids: torch.Tensor):
         -------
         A generator object.
             The returned generator object returns a future on
-            ``read_async_num_stages(ids.device)``th invocation. The return result
+            ``read_async_num_stages(ids.device)``\ th invocation. The return result
             can be accessed by calling ``.wait()``. on the returned future object.
             It is undefined behavior to call ``.wait()`` more than once.
 

diff --git a/python/dgl/graphbolt/impl/gpu_cached_feature.py b/python/dgl/graphbolt/impl/gpu_cached_feature.py
@@ -113,7 +113,7 @@ def read_async(self, ids: torch.Tensor):
         -------
         A generator object.
             The returned generator object returns a future on
-            ``read_async_num_stages(ids.device)``th invocation. The return result
+            ``read_async_num_stages(ids.device)``\ th invocation. The return result
             can be accessed by calling ``.wait()``. on the returned future object.
             It is undefined behavior to call ``.wait()`` more than once.
 

diff --git a/python/dgl/graphbolt/impl/torch_based_feature_store.py b/python/dgl/graphbolt/impl/torch_based_feature_store.py
@@ -151,7 +151,7 @@ def read_async(self, ids: torch.Tensor):
         -------
         A generator object.
             The returned generator object returns a future on
-            ``read_async_num_stages(ids.device)``th invocation. The return result
+            ``read_async_num_stages(ids.device)``\ th invocation. The return result
             can be accessed by calling ``.wait()``. on the returned future object.
             It is undefined behavior to call ``.wait()`` more than once.
 
@@ -434,7 +434,7 @@ def read_async(self, ids: torch.Tensor):
         -------
         A generator object.
             The returned generator object returns a future on
-            ``read_async_num_stages(ids.device)``th invocation. The return result
+            ``read_async_num_stages(ids.device)``\ th invocation. The return result
             can be accessed by calling ``.wait()``. on the returned future object.
             It is undefined behavior to call ``.wait()`` more than once.