Commit Graph

2040 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
8e9eb8b544 [GraphBolt][CUDA] sampled_edge_ids to fetch indices later. (#7664) 2024-08-07 00:32:50 -04:00
Muhammed Fatih BALIN
ae457160ba [GraphBolt] Remove unused return_eids option. (#7663) 2024-08-06 23:48:17 -04:00
Muhammed Fatih BALIN
3bea3e0c15 [GraphBolt][CUDA] Warning there are multiple CopyTo. (#7635) 2024-08-06 09:18:46 -04:00
Muhammed Fatih BALIN
40d32455be [GraphBolt][CUDA] Warn only when cuda is available. (#7643) 2024-08-06 09:18:06 -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
0b31bdd4e7 [GraphBolt] Refactor CachePolicy more. (#7649) 2024-08-03 23:12:34 -04:00
Muhammed Fatih BALIN
683a25a8ec [GraphBolt] Update CPUCachedFeature docstring. (#7646) 2024-08-03 11:12:51 -04:00
Muhammed Fatih BALIN
c5cace79e9 [GraphBolt] CPUCachedFeature no overlap fix. (#7648) 2024-08-03 11:12:23 -04:00
Renjie LIU
ea33b40efd [GraphBolt] Check data alignment before copying the file (#7641) 2024-08-02 19:29:02 -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
1a8cf7e0fd [GraphBolt][Doc] Fix rest of the read_async docs. (#7640) 2024-08-02 11:49:32 +08:00
pyynb
f1c28d3e9a [GraphBolt] Fix GPUCachedFeature docstring typo (#7639) 2024-08-02 11:28:45 +08:00
Muhammed Fatih BALIN
dfd491568d [GraphBolt][CUDA] Optimize CopyTo performance. (#7634) 2024-08-01 13:10:43 -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
65f85b50bb [GraphBolt] Use query_and_then_replace_async in read_async. (#7626) 2024-07-31 11:02:24 -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
Theodore Vasiloudis
ceb6672e63 [GraphBolt] Allow using multiple processes for GraphBolt partition conversion (#7497) 2024-07-30 13:28:29 +08:00
Muhammed Fatih BALIN
f989f9da14 [GraphBolt] Reuse CachePolicy::Query partition result. (#7608) 2024-07-29 03:03:41 -04:00
Muhammed Fatih BALIN
1202d22181 [GraphBolt] Optimize CachePolicy even further. (#7606) 2024-07-28 08:21:36 -04:00
Muhammed Fatih BALIN
ee20edb83d [GraphBolt][CUDA] Enable non_blocking copy_to in gb.DataLoader. (#7603) 2024-07-27 22:19:28 -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
781cc500c5 [GraphBolt] Move get_attributes utils. (#7601) 2024-07-27 20:28:19 -04:00
Muhammed Fatih BALIN
0af92c2303 [GraphBolt] recursive_apply_reduce_all. (#7600) 2024-07-27 19:55:23 -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
46f68e0bcc [GraphBolt] Use parallel_for and return replace partition out. (#7585) 2024-07-25 21:58:59 -04:00
Muhammed Fatih BALIN
b8604a5491 [GraphBolt] CachePolicy Writer lock for read_async correctness. (#7581) 2024-07-24 19:02:28 -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
7551a5da2e [GraphBolt][CUDA] GPUGraphCache is being unnecessarily created. (#7577) 2024-07-24 04:13:30 -04:00
Muhammed Fatih BALIN
849f1ce4b2 [GraphBolt][io_uring] Use the interop pool for io_uring workers. (#7575)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2024-07-23 23:23:45 -04:00
Muhammed Fatih BALIN
bed1d35261 [GraphBolt][CUDA] Eliminate memory leaks in FeatureFetcher. (#7573) 2024-07-23 07:51:41 -04:00
Rhett Ying
b5fe4737bb [dev] add more dependencies into setup (#7570) 2024-07-23 07:56:31 +08:00
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