Merge branch 'master' into gb_pyg_link_prediction_example

dmlc · Aug 27, 2024 · 526aeda · 526aeda
2 parents 46f2308 + 240a49e
commit 526aeda
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/dgl/graphbolt/impl/cpu_cached_feature.py b/python/dgl/graphbolt/impl/cpu_cached_feature.py
@@ -17,7 +17,7 @@
 
 
 class CPUCachedFeature(Feature):
-    r"""CPU cached feature wrapping a fallback feature. Use `cpu_feature_cache`
+    r"""CPU cached feature wrapping a fallback feature. Use `cpu_cached_feature`
     to construct an instance of this class.
 
     Parameters

diff --git a/python/dgl/graphbolt/impl/gpu_cached_feature.py b/python/dgl/graphbolt/impl/gpu_cached_feature.py
@@ -18,7 +18,7 @@
 class GPUCachedFeature(Feature):
     r"""GPU cached feature wrapping a fallback feature. It uses the least
     recently used (LRU) algorithm as the cache eviction policy. Use
-    `gpu_feature_cache` to construct an instance of this class.
+    `gpu_cached_feature` to construct an instance of this class.
 
     Places the GPU cache to torch.cuda.current_device().