文件历史

514 次代码提交

作者 SHA1 备注 提交日期
Xinyu Yao d9caeaaafc [GraphBolt] Update dataset names that exposed to users in examples. (#7280)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-04-08 17:06:49 +08:00
Muhammed Fatih BALIN 78df81015a [GraphBolt] Add optimized unique_and_compact_batched. (#7239) 2024-04-07 04:55:08 -04:00
Muhammed Fatih BALIN d3483fe1f1 [GraphBolt][CUDA] Optimize hetero sampling. (#7223) 2024-04-03 03:18:30 -04:00
yxy235 b743cdef71 [GraphBolt] Hack for original seed_nodes and node_pairs. (#7248)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-04-02 16:25:46 +08:00
Mingbang Wang 3df6e30180 [GraphBolt] Update __len__() and __getitem__() of ItemSet (#7253) 2024-04-02 11:04:08 +08:00
xiangyuzhi b725ee53d7 [Graphbolt] Disk feature fetcher (#7128)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-40-250.ap-northeast-1.compute.internal>
2024-04-01 16:01:43 +08:00
yxy235 fde1896f59 [GraphBolt] Fix to when using seeds. (#7245)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-03-28 13:38:14 +08:00
yxy235 3885da25f7 [GraphBolt] Modify _init_all_nodes_set to use seeds. (#7243)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-03-27 14:53:28 +08:00
yxy235 2f585940a8 [GraphBolt] Modify _seeds_preprocess in subgraph_sampler. (#7242)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-03-27 10:53:17 +08:00
yxy235 1ad78fba61 [GraphBolt] Update minibatch to to support seeds. (#7235)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-03-22 13:47:06 +08:00
yxy235 858c0b8632 [GraphBolt] Update to_pyg_data() to support get batch_size from seeds. (#7214)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-03-18 14:08:25 +08:00
Andrei Ivanov aed161e63f [GraphBolt] Fixing warning in test_to_pyg_data. (#7210) 2024-03-17 01:22:05 -04:00
Muhammed Fatih BALIN a272efed96 [GraphBolt] Implement labor dependent minibatching - python side. (#7208) 2024-03-13 13:05:19 -04:00
Andrei Ivanov b0982feb21 Reactivation of one of the GPU tests. (#7193) 2024-03-08 09:57:01 +08:00
Ramon Zhou 12fab559b5 [GraphBolt][PyG] Add more attributes in to_pyg_data (#7196) 2024-03-06 14:15:21 +08:00
Muhammed Fatih BALIN bc978736fc [GraphBolt] torch.compile() support for gb.expand_indptr. (#7188) 2024-03-05 00:26:54 -05:00
Andrei Ivanov 85c030f673 [GraphBolt] Combining test_OnDiskDataset_preprocess_yaml_content tests for Windows and Unix. (#7148) 2024-02-29 13:21:10 +08:00
Mingbang Wang 5d2d14538a [GraphBolt] Support saving graph and TVT_Sets in int32 (#7127)
Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com>
2024-02-28 21:54:19 -05:00
Rhett Ying e5263013a9 [CI] use torch 2.0.0, cu118, ubuntu2004, python310 (#7158) 2024-02-27 16:37:48 +08:00
Muhammed Fatih BALIN dbafbe4159 [GraphBolt] Fix scalar itemset dtype issue (#7147) 2024-02-26 12:39:02 -05:00
Muhammed Fatih BALIN 7c51cd1643 [GraphBolt] Cast sampled data to minimum dtype. (#7131) 2024-02-22 22:53:29 -05:00
yxy235 d95720b9b1 [Misc] Rename test files. (#7139)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-02-22 14:25:36 +08:00
yxy235 4440ac7548 [GraphBolt] Modify exclude_seed_edges to support seeds. (#7114)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-02-22 14:25:12 +08:00
Andrei Ivanov 364cb7186e Skip test when atomic operations are not supported on GPU. (#7117) 2024-02-21 16:14:58 +08:00
yxy235 bfd7cee194 [Misc] Move sample_utils tests to correct directory. (#7125)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-02-21 11:52:49 +08:00
Ramon Zhou 2df85862a5 [GraphBolt] Correct to_pyg_data (#7124) 2024-02-20 11:11:05 +08:00
Andrei Ivanov 13cbad32ce Suppressing warnings generated by the test_prop_nodes_topo test. (#6651)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2024-02-19 10:09:15 +08:00
Mingbang Wang b569e4beaf [Misc] Correct some typos (#7120) 2024-02-18 13:05:42 +08:00
Mingbang Wang 458b938c42 [GraphBolt] modify preprocess_ondisk_dataset() (#6986) 2024-02-18 01:04:39 +08:00
Andrei Ivanov 13e7c2faeb [GraphBolt] Improving subgraph_sampler tests. (#7047) 2024-02-14 23:36:00 +08:00
Ramon Zhou 8cf5ad84b6 [GraphBolt] Add to_pyg_data for MiniBatch (#7076) 2024-02-07 11:02:28 +08:00
yxy235 845864d2a0 [GraphBolt] Modify SubgraphSampler to support seeds. (#7049)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-02-06 14:26:06 +08:00
Muhammed Fatih BALIN a2e1c79618 [GraphBolt][CUDA] Pipelined sampling accuracy fix (#7088) 2024-02-05 20:01:31 +03:00
Muhammed Fatih BALIN badeaf19dc [GraphBolt][CUDA] Pipelined sampling optimization (#7039) 2024-02-05 12:20:15 +08:00
Muhammed Fatih BALIN 346197c477 [GraphBolt][CUDA] Add gb.index_select and fix example inferencing. (#7051) 2024-02-04 19:56:40 +03:00
Andrei Ivanov 95142b844e [GraphBolt] Fixing two issues in GraphBolt tests. (#7059)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2024-02-04 16:06:06 +08:00
yxy235 af0b63eda6 [GraphBolt] Fix gpu NegativeSampler for seeds. (#7068)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-02-04 13:24:34 +08:00
Muhammed Fatih BALIN 15695ed0ec [GraphBolt][CUDA] Handle edge case of %100 cache hit rate. (#7080) 2024-02-02 23:03:06 -08:00
Hongzhi (Steve), Chen e602ab1b56 [Misc] Move test_minibatch to the correct dir. (#7055)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2024-02-01 14:48:35 +08:00
Rhett Ying e117adac25 [GraphBolt] fix testcases on warning messages (#7054) 2024-02-01 11:41:05 +08:00
Mingbang Wang 942b17ab2e [GraphBolt] Add hardcode testcases for gb.preprocess_ondisk_dataset (#7035) 2024-01-31 23:36:01 +08:00
Muhammed Fatih BALIN f051a5cc69 [GraphBolt][CUDA] Add is_pinned method to TorchBasedFeatureStore. (#7041) 2024-01-31 14:57:24 +03:00
yxy235 e7c3454f12 [GraphBolt] Fix NegativeSampler seeds support. (#7043)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2024-01-31 15:14:24 +08:00
yxy235 cda8b38101 [GraphBolt] Add compacted_seeds to minibatch. (#7033)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-01-30 14:29:37 +08:00
Muhammed Fatih BALIN b085224fb9 [GraphBolt][CUDA] Dataloader feature overlap fix (#7036) 2024-01-30 09:24:22 +03:00
yxy235 11adb4e717 [GraphBolt] Refactor NegativeSampler. (#7001)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-01-26 14:01:42 +08:00
Rhett Ying 523bbb4cb1 [GraphBolt] fix preprocess issue for single ntype/etype graph (#7011) 2024-01-25 16:03:25 +08:00
Muhammed Fatih BALIN 3a79f02189 [GraphBolt][CUDA] Make nodes optional for sampling (#6993) 2024-01-24 12:58:54 -05:00
Muhammed Fatih BALIN 8da604d7ab [GraphBolt][CUDA] Refine pinning tests (#6994) 2024-01-23 16:27:33 +08:00
yxy235 d249744875 [GraphBolt] Add indexes to MiniBatch. (#6989)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
2024-01-23 13:08:12 +08:00