Commit Graph

2034 Commits

Author SHA1 Message Date
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
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