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][Example] Minor bug fix. (#7566)

Browse Source
This commit is contained in:
Muhammed Fatih BALIN
2024-07-22 15:32:10 -04:00
committed by GitHub
parent c83acdddd6
commit 91c7a25755
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/graphbolt/pyg/labor/node_classification.py
View File

@@ -402,8 +402,13 @@ def main():
num_classes = dataset.tasks[0].metadata["num_classes"]
feature_index_device = (
args.feature_device if args.feature_device != "pinned" else None
)
feature_num_bytes = (
features[("node", None, "feat")].read(torch.zeros(1).long()).nbytes
features[("node", None, "feat")]
# Read a single row to query its size in bytes.
.read(torch.zeros(1, device=feature_index_device).long()).nbytes
)
if args.num_cpu_cached_features > 0 and isinstance(
features[("node", None, "feat")], gb.DiskBasedFeature