diff --git a/benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py b/benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py index 0a7c7fa8cb7..5c1035799b2 100644 --- a/benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py +++ b/benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py @@ -108,7 +108,7 @@ def x_dict(self) -> Dict[str, torch.Tensor]: full_path = os.path.join(node_type_path, 'node_feat.npy') else: full_path = os.path.join(node_type_path, f'node_feat_{self.__replication_factor}x.npy') - + self.__disk_x = {'paper': np.load( full_path, mmap_mode='r'