Commit Graph

2002 Commits

Author SHA1 Message Date
Muhammed Fatih BALIN
5b4635a3e6 [GraphBolt] Async feature fetch refactor (#7540) 2024-07-22 06:59:08 -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
c218b197f1 [GraphBolt] Avoid initializing CUDAContext in DataLoader workers. (#7560) 2024-07-22 02:05:28 -04:00
Muhammed Fatih BALIN
d775ab1dd1 [GraphBolt] Refactor and extend FeatureStore. (#7558) 2024-07-22 01:20:05 -04:00
Muhammed Fatih BALIN
4093f7934a [GraphBolt] CPUCachedFeature.read_async branch 2 [7] (#7552) 2024-07-20 16:31:43 -04:00
Muhammed Fatih BALIN
606693ba55 [GraphBolt] CPUCachedFeature.read_async branch 1 [6] (#7551) 2024-07-20 16:29:27 -04:00
Muhammed Fatih BALIN
78a86115f1 [GraphBolt] CPUCachedFeature.read_async branch 3 [8] (#7553) 2024-07-20 15:34:26 -04:00
Muhammed Fatih BALIN
68b49625ae [GraphBolt] CPUCachedFeature.read_async body. [4] (#7549) 2024-07-20 14:49:45 -04:00
Muhammed Fatih BALIN
ddef6ab7c5 [GraphBolt][CUDA] GPUCachedFeature.read_async. [3] (#7548) 2024-07-20 13:56:37 -04:00
Muhammed Fatih BALIN
42901740cf [GraphBolt] TorchBasedFeature.read_async. [2] (#7547) 2024-07-20 13:55:02 -04:00
Muhammed Fatih BALIN
8d770b6100 [GraphBolt] Add read_async to gb.Feature. [1] (#7546) 2024-07-20 13:06:07 -04:00
Muhammed Fatih BALIN
4d43d92a41 [GraphBolt] Add CPUCachedFeature. [2] (#7537) 2024-07-19 07:30:10 -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
472bef87b2 [GraphBolt][CUDA] Add stream for host to device copies. (#7534) 2024-07-17 05:34:57 -04:00
Muhammed Fatih BALIN
d37d516be0 [GraphBolt] Check return values of liburing calls and refactor. (#7518) 2024-07-17 04:50:35 -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
6e1a2df001 [GraphBolt][CUDA] In-place pinning should be out-of-place on WSL. (#7519) 2024-07-16 05:11:22 -04:00
Muhammed Fatih BALIN
8368f629e3 [GraphBolt] Add gb.numpy_save_aligned. (#7524) 2024-07-16 15:38:50 +08:00
Muhammed Fatih BALIN
0984ad9c39 [GraphBolt][CUDA] Add FeatureCache::IndexSelect. (#7526) 2024-07-16 03:30:14 -04:00
Muhammed Fatih BALIN
4d3318bcca [GraphBolt][CUDA] Force CUDA version if torch is. (#7514) 2024-07-16 01:24:44 -04:00
Muhammed Fatih BALIN
2ebd7eb609 [GraphBolt] Cache policy replace fix. (#7520) 2024-07-16 00:55:01 -04:00
Muhammed Fatih BALIN
4d497d1782 [GraphBolt] Fix issue on torch 2.3.1 (#7521) 2024-07-13 15:10:59 +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
3a8ce85dc5 [GraphBolt] Runtime warning for CUDA availability mismatch. (#7513) 2024-07-10 03:59:45 -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
Ayush Noori
025767c204 [Feature] Fixed sampler with limit on sampled nodes/edges in batch subgraph (#6668)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2024-07-09 14:00:36 +08: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
Rhett Ying
dfdfad2cdb Revert "[dev] limit numpy < 2.0.0 for compatibility" (#7496) 2024-07-01 11:33:51 +08:00
Muhammed Fatih BALIN
f9a34b6439 [GraphBolt] Fix deprecation warning (#7493) 2024-06-30 23:23:23 -04:00
Andrei Ivanov
65b949fdee [UserWarning] Fixing the warnings appearing in test_nn.py::test_group_rev_res. (#7486) 2024-07-01 11:01:16 +08:00
Rhett Ying
00f8feeee5 [release] bump version to 2.4 for nightly (#7489) 2024-06-27 11:55:38 +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
Rhett Ying
6294c4be16 [dev] limit numpy < 2.0.0 for compatibility (#7479) 2024-06-26 13:11:36 +08:00
Mingbang Wang
d3e4f4f375 [Misc] modify workaround for DILL_AVAILABLE in torch 2.3.0 (#7474) 2024-06-26 08:28:40 +08:00
peizhou001
b3e884e8fd [Temporal] Add time window support (#7467)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-21-218.ap-northeast-1.compute.internal>
2024-06-25 16:57:31 +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
Rhett Ying
ed50c170dd [DistGB] fix add edge attr request (#7460) 2024-06-14 13:29:56 +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
7f1d164392 [DistGB]enable prob/mask sampling on graphbolt partitions (#7458) 2024-06-13 17:25:22 +08:00
Rhett Ying
c29a2d01bc [DistGB] store original homo eids in default for GB partitions (#7455) 2024-06-12 11:22:00 +08:00
Rhett Ying
069a1527e6 [GraphBolt] enable to set single node/edge attribute (#7454) 2024-06-12 09:25:55 +08:00
Rhett Ying
e366260d43 [GraphBolt] pass tensor directly to probs or masks in sampling C++ API (#7449) 2024-06-06 14:47:38 +08:00
Rhett Ying
0fee9be329 [DistGB] exclude edges right after sampling (#7442) 2024-06-03 10:16:44 +08:00