Commit Graph

481 Commits

Author SHA1 Message Date
Muhammed Fatih BALIN
b5ee45fd1a [GraphBolt][CUDA] Fetch indices for NS later. (#7665) 2024-08-07 10:52:06 -04:00
Muhammed Fatih BALIN
b79d7430ce [GraphBolt] Remove unnecessary dispatch and refine. (#7660) 2024-08-06 07:29:43 -04:00
Hongzhi (Steve), Chen
26ff09fdbf Remove dependency on torchdata. (#7638)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2024-08-06 11:20:26 +08:00
Muhammed Fatih BALIN
c5cace79e9 [GraphBolt] CPUCachedFeature no overlap fix. (#7648) 2024-08-03 11:12:23 -04:00
Muhammed Fatih BALIN
84c2cf03e1 [GraphBolt] Keep references valid in CachePolicy. (#7644) 2024-08-02 17:17:42 -04:00
Muhammed Fatih BALIN
56a1e64945 [GraphBolt][CUDA] Use better memory allocation algorithm to avoid OOM. (#7618) 2024-08-02 07:25:06 -04:00
Muhammed Fatih BALIN
5c902cd59e [GraphBolt][CUDA] Remove python thread from overlap_graph_fetch. (#7632) 2024-08-01 11:55:09 -04:00
Muhammed Fatih BALIN
489ab1a81e [GraphBolt] QueryAndReplace replaces QueryAndThenReplace. (#7628) 2024-08-01 10:23:18 -04:00
Hongzhi (Steve), Chen
ea3716d562 Use torch core instead of torchdata modules. (#7609)
Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com>
2024-08-01 07:13:09 -04:00
Muhammed Fatih BALIN
f37741c9ad [GraphBolt] Test query_and_then_replace more rigorously. (#7625) 2024-07-31 10:20:21 -04:00
Muhammed Fatih BALIN
cfe0541802 [GraphBolt] CachePolicy::QueryAndThenReplace for num_parts>1. (#7621) 2024-07-31 02:41:00 -04:00
Muhammed Fatih BALIN
87ea76b02a [GraphBolt] Test CachePolicy::QueryAndThenReplace for num_parts==1. (#7620) 2024-07-30 15:16:23 -04:00
Muhammed Fatih BALIN
f989f9da14 [GraphBolt] Reuse CachePolicy::Query partition result. (#7608) 2024-07-29 03:03:41 -04:00
Muhammed Fatih BALIN
a004a2535b [GraphBolt][CUDA] Add non_blocking option to CopyTo. (#7602) 2024-07-27 21:03:14 -04:00
Muhammed Fatih BALIN
db13f05f58 [GraphBolt][CUDA] Fix overlap_graph_fetch edge_ids cached case. (#7595) 2024-07-27 01:04:40 -04:00
Muhammed Fatih BALIN
bea83ed738 [GraphBolt][CUDA] Fix overlap_graph_fetch edge_ids. (#7594) 2024-07-26 19:19:36 -04:00
Muhammed Fatih BALIN
ea281961f6 [GraphBolt][CUDA] gb.indptr_edge_ids python API and tests. (#7593) 2024-07-26 17:58:41 -04:00
Muhammed Fatih BALIN
3299bbd126 [GraphBolt] Edge feature fetch fix. (#7589) 2024-07-26 08:10:12 -04:00
Muhammed Fatih BALIN
b737f18e07 [GraphBolt] Option to use DiskBasedFeature to OnDiskDataset.load() (#7541) 2024-07-24 05:11:34 -04:00
Muhammed Fatih BALIN
2074cbf556 [GraphBolt] Refactor FeatureFetcher to use read_async. (#7559) 2024-07-22 06:39:27 -04:00
Muhammed Fatih BALIN
d775ab1dd1 [GraphBolt] Refactor and extend FeatureStore. (#7558) 2024-07-22 01:20:05 -04:00
Muhammed Fatih BALIN
a4418a2ab7 [GraphBolt][CUDA] GPUCachedFeature.read_async tests 2. [10] (#7555) 2024-07-20 17:06:15 -04:00
Muhammed Fatih BALIN
8fb50d7b8d [GraphBolt] CPUCachedFeature.read_async tests [9] (#7554) 2024-07-20 17:05:37 -04:00
Muhammed Fatih BALIN
b902171813 [GraphBolt][CUDA] GPUCachedFeature read_async tests 1. [5] (#7550) 2024-07-20 14:38:46 -04:00
Muhammed Fatih BALIN
b66e62ef07 [GraphBolt] scatter_async. (#7545) 2024-07-20 11:46:01 -04:00
Muhammed Fatih BALIN
117df61fc0 [GraphBolt] CPUCachedFeature tests. [3] (#7538) 2024-07-19 09:32:50 -04:00
Muhammed Fatih BALIN
41bb3b6491 [GraphBolt] Rename to CPUFeatureCache and bug fixes. [1] (#7539) 2024-07-18 23:50:21 -04:00
Muhammed Fatih BALIN
4a77aeadbe [GraphBolt] Add index_select_async. (#7535) 2024-07-17 05:46:53 -04:00
Muhammed Fatih BALIN
b0706d783d [GraphBolt] Add additional capabilities to gb.index_select. (#7532) 2024-07-16 21:52:55 -04:00
Muhammed Fatih BALIN
c3b774d6ab [GraphBolt] Refine FeatureCache and increase test coverage. (#7531) 2024-07-16 13:51:43 -04:00
Muhammed Fatih BALIN
2a5c884a56 [GraphBolt] Improve DiskBasedFeature tests. (#7530) 2024-07-16 05:11:31 -04:00
Muhammed Fatih BALIN
8368f629e3 [GraphBolt] Add gb.numpy_save_aligned. (#7524) 2024-07-16 15:38:50 +08:00
Muhammed Fatih BALIN
945247e895 [GraphBolt][CUDA] Skip in-place pinning tests on WSL. (#7517) 2024-07-12 03:50:08 -04:00
Muhammed Fatih BALIN
10e9f889fe [GraphBolt] index_select_csc_batched. (#7511) 2024-07-10 01:46:56 -04:00
Muhammed Fatih BALIN
5434fc35db [GraphBolt][Standalone] Reduce and mark dependencies on DGL. (#7499) 2024-07-09 05:58:20 -04:00
Muhammed Fatih BALIN
9e546f3075 [GraphBolt] gb.TemporalLayerNeighborSampler (#7503) 2024-07-09 16:51:22 +08:00
Muhammed Fatih BALIN
52651379a0 [GraphBolt][io_uring] Refactor and enable tests (#7506) 2024-07-09 04:06:09 -04:00
Muhammed Fatih BALIN
de33ee1ad5 [GraphBolt] Refactor S3-FIFO and add SIEVE, LRU and CLOCK. (#7508) 2024-07-09 01:37:30 -04:00
Muhammed Fatih BALIN
25c29537b9 [GraphBolt] Partitioned S3-FIFO Feature Cache implementation (#7492) 2024-07-04 22:24:15 -04:00
Muhammed Fatih BALIN
fb0547d613 [GraphBolt] Add temporal labor sampling to graph. (#7500) 2024-07-04 12:03:16 -04:00
pyynb
713ffb5714 [GraphBolt] Fix DiskBasedFeature Memory Leak Problem (#7349)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-128.us-west-2.compute.internal>
2024-07-03 10:12:16 +08:00
Andrei Ivanov
6b222af6f3 [DeprecationWarning] Fixing the warnings appearing in test_traversal.py::test_bfs. (#7484)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2024-07-01 15:02:49 +08:00
Muhammed Fatih BALIN
95dc96a631 [GraphBolt][CUDA] Incremental GPU graph cache into gb.Dataloader. (#7475)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2024-06-26 15:38:31 -04:00
Muhammed Fatih BALIN
c822bc16f3 [GraphBolt][CUDA] Incremental GPU graph caching. (#7470)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2024-06-26 13:26:11 +08:00
Mingbang Wang
7c3d418e88 [GraphBolt] Alias ItemSetDict to HeteroItemSet (#7466)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2024-06-20 10:29:38 +08:00
Xinyu Yao
e489e3843a [GraphBolt] Fix blocks in minibatch when facing with empty edges in subgraph. (#7413)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-30-178.ap-northeast-1.compute.internal>
Co-authored-by: Skeleton003 <799284168.qq.com>
Co-authored-by: Skeleton003 <799284168@qq.com>
Co-authored-by: Mingbang Wang <100203018+Skeleton003@users.noreply.github.com>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2024-06-13 17:36:38 +08:00
Rhett Ying
069a1527e6 [GraphBolt] enable to set single node/edge attribute (#7454) 2024-06-12 09:25:55 +08:00
Mingbang Wang
ece1210817 [GraphBolt] Rename ItemSetDict as HeteroItemSet (#7424) 2024-05-24 13:39:27 +08:00
Mingbang Wang
3574bff42c [GraphBolt] Update ItemSampler (#7408) 2024-05-21 14:48:47 +08:00
Mingbang Wang
2ee440a68f [GraphBolt] Modify ItemSet/Dict requiring items to be tensors (#7394)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-30-178.ap-northeast-1.compute.internal>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2024-05-15 11:48:56 +08:00