pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-03 19:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

[GraphBolt][CUDA] GPUGraphCache is being unnecessarily created. (#7577)

Browse Source
This commit is contained in:
Muhammed Fatih BALIN
2024-07-24 04:13:30 -04:00
committed by GitHub
parent 849f1ce4b2
commit 7551a5da2e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/dgl/graphbolt/dataloader.py
View File

@@ -209,7 +209,7 @@ class DataLoader(torch.utils.data.DataLoader):
executor = ThreadPoolExecutor(max_workers=1)
gpu_graph_cache = None
for sampler in samplers:
if gpu_graph_cache is None:
if num_gpu_cached_edges > 0 and gpu_graph_cache is None:
gpu_graph_cache = construct_gpu_graph_cache(
sampler, num_gpu_cached_edges, gpu_cache_threshold
)