├── .github ├── ISSUE_TEMPLATE │ ├── bug-report.yml │ ├── config.yml │ ├── documentation.yml │ └── feature-request.yml └── workflows │ ├── glt-ci.yml │ ├── glt-v6d-ci.yml │ ├── manylinux-cd.yml │ └── scripts │ ├── build.sh │ ├── build_wheel.sh │ └── upload_pypi.sh ├── .gitignore ├── .gitmodules ├── .readthedocs.yaml ├── CMakeLists.txt ├── LICENSE ├── README.md ├── benchmarks └── api │ ├── README.md │ ├── bench_dist_config.yml │ ├── bench_dist_neighbor_loader.py │ ├── bench_feature.py │ ├── bench_sampler.py │ └── run_dist_bench.py ├── dockerfiles ├── graphlearn-torch-dev.Dockerfile ├── graphlearn-torch-wheel.Dockerfile └── gs-graphlearn-torch-dev.Dockerfile ├── docs ├── Makefile ├── README.md ├── apis │ ├── channel.rst │ ├── data.rst │ ├── distributed.rst │ ├── loader.rst │ ├── modules.rst │ ├── partition.rst │ ├── sampler.rst │ └── utils.rst ├── conf.py ├── contribute.md ├── faq.md ├── figures │ ├── arch.png │ ├── dist_arch_server_client_mode.png │ ├── dist_arch_worker_mode.png │ ├── glt_dgl.png │ ├── replica_per_dg.png │ ├── replica_per_gpu.png │ ├── scale_out.png │ ├── scale_up.png │ └── uni_tensor.png ├── get_started │ ├── dist_train.md │ ├── graph_basic.md │ ├── hetero.md │ ├── link_pred.md │ └── node_class.md ├── index.rst ├── install │ └── install.md ├── make.bat ├── requirements.txt └── tutorial │ ├── aliyun.md │ ├── basic_object.md │ ├── dist.md │ └── graph_ops.md ├── examples ├── README.md ├── distributed │ ├── README.md │ ├── dist_sage_unsup │ │ ├── dist_sage_unsup.py │ │ └── preprocess_template.py │ ├── dist_train_sage_sup_config.yml │ ├── dist_train_sage_supervised.py │ ├── partition_ogbn_dataset.py │ ├── run_dist_train_sage_sup.py │ └── server_client_mode │ │ ├── sage_supervised_client.py │ │ └── sage_supervised_server.py ├── feature_mp.py ├── gpt │ ├── README.md │ ├── arxiv.py │ └── utils.py ├── graph_sage_unsup_ppi.py ├── hetero │ ├── bipartite_sage_unsup.py │ ├── hierarchical_sage.py │ ├── train_hgt_mag.py │ └── train_hgt_mag_mp.py ├── igbh │ ├── Dockerfile │ ├── README.md │ ├── __init__.py │ ├── build_partition_feature.py │ ├── compress_graph.py │ ├── dataset.py │ ├── dist_train_rgnn.py │ ├── download.py │ ├── download_igbh_full.sh │ ├── mlperf_logging_utils.py │ ├── partition.py │ ├── rgnn.py │ ├── split_seeds.py │ ├── train_rgnn_multi_gpu.py │ └── utilities.py ├── multi_gpu │ └── train_sage_ogbn_papers100m.py ├── pai │ ├── README.md │ ├── ogbn_products │ │ ├── README.md │ │ ├── data_preprocess.py │ │ ├── dist_train_products_sage.py │ │ └── train_products_sage.py │ └── requirements.txt ├── seal_link_pred.py ├── train_sage_ogbn_products.py └── train_sage_prod_with_trim.py ├── graphlearn_torch ├── csrc │ ├── cpu │ │ ├── graph.cc │ │ ├── inducer.cc │ │ ├── inducer.h │ │ ├── random_negative_sampler.cc │ │ ├── random_negative_sampler.h │ │ ├── random_sampler.cc │ │ ├── random_sampler.h │ │ ├── stitch_sample_results.cc │ │ ├── subgraph_op.cc │ │ ├── subgraph_op.h │ │ ├── weighted_sampler.cc │ │ └── weighted_sampler.h │ ├── cuda │ │ ├── graph.cu │ │ ├── hash_table.cu │ │ ├── inducer.cu │ │ ├── inducer.cuh │ │ ├── random_negative_sampler.cu │ │ ├── random_negative_sampler.cuh │ │ ├── random_sampler.cu │ │ ├── random_sampler.cuh │ │ ├── stitch_sample_results.cu │ │ ├── subgraph_op.cu │ │ ├── subgraph_op.cuh │ │ ├── unified_tensor.cu │ │ └── weighted_sampler.cuh │ ├── sample_queue.cc │ ├── shm_queue.cc │ └── tensor_map.cc ├── include │ ├── common.cuh │ ├── common.h │ ├── func_factory.h │ ├── graph.h │ ├── hash_table.cuh │ ├── inducer_base.h │ ├── negative_sampler.h │ ├── sample_queue.h │ ├── sampler.h │ ├── shm_queue.h │ ├── stitch_sample_results.h │ ├── subgraph_op_base.h │ ├── tensor_map.h │ ├── types.h │ └── unified_tensor.cuh ├── python │ ├── __init__.py │ ├── channel │ │ ├── __init__.py │ │ ├── base.py │ │ ├── mp_channel.py │ │ ├── remote_channel.py │ │ └── shm_channel.py │ ├── data │ │ ├── __init__.py │ │ ├── dataset.py │ │ ├── feature.py │ │ ├── graph.py │ │ ├── reorder.py │ │ ├── table_dataset.py │ │ ├── unified_tensor.py │ │ └── vineyard_utils.py │ ├── distributed │ │ ├── __init__.py │ │ ├── dist_client.py │ │ ├── dist_context.py │ │ ├── dist_dataset.py │ │ ├── dist_feature.py │ │ ├── dist_graph.py │ │ ├── dist_link_neighbor_loader.py │ │ ├── dist_loader.py │ │ ├── dist_neighbor_loader.py │ │ ├── dist_neighbor_sampler.py │ │ ├── dist_options.py │ │ ├── dist_random_partitioner.py │ │ ├── dist_sampling_producer.py │ │ ├── dist_server.py │ │ ├── dist_subgraph_loader.py │ │ ├── dist_table_dataset.py │ │ ├── event_loop.py │ │ └── rpc.py │ ├── loader │ │ ├── __init__.py │ │ ├── link_loader.py │ │ ├── link_neighbor_loader.py │ │ ├── neighbor_loader.py │ │ ├── node_loader.py │ │ ├── subgraph_loader.py │ │ └── transform.py │ ├── partition │ │ ├── __init__.py │ │ ├── base.py │ │ ├── frequency_partitioner.py │ │ ├── partition_book.py │ │ └── random_partitioner.py │ ├── py_export_glt.cc │ ├── py_export_v6d.cc │ ├── sampler │ │ ├── __init__.py │ │ ├── base.py │ │ ├── negative_sampler.py │ │ └── neighbor_sampler.py │ ├── typing.py │ └── utils │ │ ├── __init__.py │ │ ├── build_glt.py │ │ ├── common.py │ │ ├── device.py │ │ ├── exit_status.py │ │ ├── mixin.py │ │ ├── singleton.py │ │ ├── tensor.py │ │ ├── topo.py │ │ └── units.py └── v6d │ ├── vineyard_utils.cc │ └── vineyard_utils.h ├── install_dependencies.sh ├── scripts ├── build_wheel.sh ├── launch_container.sh ├── remove_container.sh ├── run_cpp_ut.sh └── run_python_ut.sh ├── setup.py ├── test ├── cpp │ ├── test_graph.cu │ ├── test_hash_table.cu │ ├── test_hetero_inducer.cu │ ├── test_inducer.cu │ ├── test_random_negative_sampler.cu │ ├── test_random_sampler.cu │ ├── test_shm_queue.cu │ ├── test_stitch_sample_results.cu │ ├── test_subgraph.cu │ ├── test_tensor_map_serializer.cu │ ├── test_unified_tensor.cu │ ├── test_vineyard.cc │ └── test_vineyard.cu └── python │ ├── dist_test_utils.py │ ├── test_dist_feature.py │ ├── test_dist_link_loader.py │ ├── test_dist_neighbor_loader.py │ ├── test_dist_random_partitioner.py │ ├── test_dist_subgraph_loader.py │ ├── test_feature.py │ ├── test_graph.py │ ├── test_hetero_neighbor_sampler.py │ ├── test_link_loader.py │ ├── test_neighbor_sampler.py │ ├── test_partition.py │ ├── test_pyg_remote_backend.py │ ├── test_sample_prob.py │ ├── test_shm_channel.py │ ├── test_subgraph.py │ ├── test_unified_tensor.py │ ├── test_vineyard.py │ └── vineyard_data │ └── modern_graph │ ├── created.csv │ ├── knows.csv │ ├── person.csv │ └── software.csv └── third_party └── googletest └── build.sh /.github/ISSUE_TEMPLATE/bug-report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/ISSUE_TEMPLATE/bug-report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/ISSUE_TEMPLATE/documentation.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/ISSUE_TEMPLATE/feature-request.yml -------------------------------------------------------------------------------- /.github/workflows/glt-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/workflows/glt-ci.yml -------------------------------------------------------------------------------- /.github/workflows/glt-v6d-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/workflows/glt-v6d-ci.yml -------------------------------------------------------------------------------- /.github/workflows/manylinux-cd.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/workflows/manylinux-cd.yml -------------------------------------------------------------------------------- /.github/workflows/scripts/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/workflows/scripts/build.sh -------------------------------------------------------------------------------- /.github/workflows/scripts/build_wheel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/workflows/scripts/build_wheel.sh -------------------------------------------------------------------------------- /.github/workflows/scripts/upload_pypi.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.github/workflows/scripts/upload_pypi.sh -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.gitmodules -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/benchmarks/api/README.md -------------------------------------------------------------------------------- /benchmarks/api/bench_dist_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/benchmarks/api/bench_dist_config.yml -------------------------------------------------------------------------------- /benchmarks/api/bench_dist_neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/benchmarks/api/bench_dist_neighbor_loader.py -------------------------------------------------------------------------------- /benchmarks/api/bench_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/benchmarks/api/bench_feature.py -------------------------------------------------------------------------------- /benchmarks/api/bench_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/benchmarks/api/bench_sampler.py -------------------------------------------------------------------------------- /benchmarks/api/run_dist_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/benchmarks/api/run_dist_bench.py -------------------------------------------------------------------------------- /dockerfiles/graphlearn-torch-dev.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/dockerfiles/graphlearn-torch-dev.Dockerfile -------------------------------------------------------------------------------- /dockerfiles/graphlearn-torch-wheel.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/dockerfiles/graphlearn-torch-wheel.Dockerfile -------------------------------------------------------------------------------- /dockerfiles/gs-graphlearn-torch-dev.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/dockerfiles/gs-graphlearn-torch-dev.Dockerfile -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/apis/channel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/apis/channel.rst -------------------------------------------------------------------------------- /docs/apis/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/apis/data.rst -------------------------------------------------------------------------------- /docs/apis/distributed.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/apis/distributed.rst -------------------------------------------------------------------------------- /docs/apis/loader.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/apis/loader.rst -------------------------------------------------------------------------------- /docs/apis/modules.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/apis/modules.rst -------------------------------------------------------------------------------- /docs/apis/partition.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/apis/partition.rst -------------------------------------------------------------------------------- /docs/apis/sampler.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/apis/sampler.rst -------------------------------------------------------------------------------- /docs/apis/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/apis/utils.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/contribute.md: -------------------------------------------------------------------------------- 1 | # Contribute to GLT -------------------------------------------------------------------------------- /docs/faq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/faq.md -------------------------------------------------------------------------------- /docs/figures/arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/arch.png -------------------------------------------------------------------------------- /docs/figures/dist_arch_server_client_mode.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/dist_arch_server_client_mode.png -------------------------------------------------------------------------------- /docs/figures/dist_arch_worker_mode.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/dist_arch_worker_mode.png -------------------------------------------------------------------------------- /docs/figures/glt_dgl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/glt_dgl.png -------------------------------------------------------------------------------- /docs/figures/replica_per_dg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/replica_per_dg.png -------------------------------------------------------------------------------- /docs/figures/replica_per_gpu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/replica_per_gpu.png -------------------------------------------------------------------------------- /docs/figures/scale_out.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/scale_out.png -------------------------------------------------------------------------------- /docs/figures/scale_up.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/scale_up.png -------------------------------------------------------------------------------- /docs/figures/uni_tensor.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/figures/uni_tensor.png -------------------------------------------------------------------------------- /docs/get_started/dist_train.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/get_started/dist_train.md -------------------------------------------------------------------------------- /docs/get_started/graph_basic.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/get_started/graph_basic.md -------------------------------------------------------------------------------- /docs/get_started/hetero.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/get_started/hetero.md -------------------------------------------------------------------------------- /docs/get_started/link_pred.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/get_started/link_pred.md -------------------------------------------------------------------------------- /docs/get_started/node_class.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/get_started/node_class.md -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/install/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/install/install.md -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/tutorial/aliyun.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/tutorial/aliyun.md -------------------------------------------------------------------------------- /docs/tutorial/basic_object.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/tutorial/basic_object.md -------------------------------------------------------------------------------- /docs/tutorial/dist.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/tutorial/dist.md -------------------------------------------------------------------------------- /docs/tutorial/graph_ops.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/docs/tutorial/graph_ops.md -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/distributed/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/README.md -------------------------------------------------------------------------------- /examples/distributed/dist_sage_unsup/dist_sage_unsup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/dist_sage_unsup/dist_sage_unsup.py -------------------------------------------------------------------------------- /examples/distributed/dist_sage_unsup/preprocess_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/dist_sage_unsup/preprocess_template.py -------------------------------------------------------------------------------- /examples/distributed/dist_train_sage_sup_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/dist_train_sage_sup_config.yml -------------------------------------------------------------------------------- /examples/distributed/dist_train_sage_supervised.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/dist_train_sage_supervised.py -------------------------------------------------------------------------------- /examples/distributed/partition_ogbn_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/partition_ogbn_dataset.py -------------------------------------------------------------------------------- /examples/distributed/run_dist_train_sage_sup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/run_dist_train_sage_sup.py -------------------------------------------------------------------------------- /examples/distributed/server_client_mode/sage_supervised_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/server_client_mode/sage_supervised_client.py -------------------------------------------------------------------------------- /examples/distributed/server_client_mode/sage_supervised_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/distributed/server_client_mode/sage_supervised_server.py -------------------------------------------------------------------------------- /examples/feature_mp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/feature_mp.py -------------------------------------------------------------------------------- /examples/gpt/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/gpt/README.md -------------------------------------------------------------------------------- /examples/gpt/arxiv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/gpt/arxiv.py -------------------------------------------------------------------------------- /examples/gpt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/gpt/utils.py -------------------------------------------------------------------------------- /examples/graph_sage_unsup_ppi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/graph_sage_unsup_ppi.py -------------------------------------------------------------------------------- /examples/hetero/bipartite_sage_unsup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/hetero/bipartite_sage_unsup.py -------------------------------------------------------------------------------- /examples/hetero/hierarchical_sage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/hetero/hierarchical_sage.py -------------------------------------------------------------------------------- /examples/hetero/train_hgt_mag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/hetero/train_hgt_mag.py -------------------------------------------------------------------------------- /examples/hetero/train_hgt_mag_mp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/hetero/train_hgt_mag_mp.py -------------------------------------------------------------------------------- /examples/igbh/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/Dockerfile -------------------------------------------------------------------------------- /examples/igbh/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/README.md -------------------------------------------------------------------------------- /examples/igbh/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/igbh/build_partition_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/build_partition_feature.py -------------------------------------------------------------------------------- /examples/igbh/compress_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/compress_graph.py -------------------------------------------------------------------------------- /examples/igbh/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/dataset.py -------------------------------------------------------------------------------- /examples/igbh/dist_train_rgnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/dist_train_rgnn.py -------------------------------------------------------------------------------- /examples/igbh/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/download.py -------------------------------------------------------------------------------- /examples/igbh/download_igbh_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/download_igbh_full.sh -------------------------------------------------------------------------------- /examples/igbh/mlperf_logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/mlperf_logging_utils.py -------------------------------------------------------------------------------- /examples/igbh/partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/partition.py -------------------------------------------------------------------------------- /examples/igbh/rgnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/rgnn.py -------------------------------------------------------------------------------- /examples/igbh/split_seeds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/split_seeds.py -------------------------------------------------------------------------------- /examples/igbh/train_rgnn_multi_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/train_rgnn_multi_gpu.py -------------------------------------------------------------------------------- /examples/igbh/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/igbh/utilities.py -------------------------------------------------------------------------------- /examples/multi_gpu/train_sage_ogbn_papers100m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/multi_gpu/train_sage_ogbn_papers100m.py -------------------------------------------------------------------------------- /examples/pai/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/pai/README.md -------------------------------------------------------------------------------- /examples/pai/ogbn_products/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/pai/ogbn_products/README.md -------------------------------------------------------------------------------- /examples/pai/ogbn_products/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/pai/ogbn_products/data_preprocess.py -------------------------------------------------------------------------------- /examples/pai/ogbn_products/dist_train_products_sage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/pai/ogbn_products/dist_train_products_sage.py -------------------------------------------------------------------------------- /examples/pai/ogbn_products/train_products_sage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/pai/ogbn_products/train_products_sage.py -------------------------------------------------------------------------------- /examples/pai/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/pai/requirements.txt -------------------------------------------------------------------------------- /examples/seal_link_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/seal_link_pred.py -------------------------------------------------------------------------------- /examples/train_sage_ogbn_products.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/train_sage_ogbn_products.py -------------------------------------------------------------------------------- /examples/train_sage_prod_with_trim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/examples/train_sage_prod_with_trim.py -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/graph.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/graph.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/inducer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/inducer.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/inducer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/inducer.h -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/random_negative_sampler.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/random_negative_sampler.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/random_negative_sampler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/random_negative_sampler.h -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/random_sampler.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/random_sampler.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/random_sampler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/random_sampler.h -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/stitch_sample_results.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/stitch_sample_results.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/subgraph_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/subgraph_op.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/subgraph_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/subgraph_op.h -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/weighted_sampler.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/weighted_sampler.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cpu/weighted_sampler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cpu/weighted_sampler.h -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/graph.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/graph.cu -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/hash_table.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/hash_table.cu -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/inducer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/inducer.cu -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/inducer.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/inducer.cuh -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/random_negative_sampler.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/random_negative_sampler.cu -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/random_negative_sampler.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/random_negative_sampler.cuh -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/random_sampler.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/random_sampler.cu -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/random_sampler.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/random_sampler.cuh -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/stitch_sample_results.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/stitch_sample_results.cu -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/subgraph_op.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/subgraph_op.cu -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/subgraph_op.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/subgraph_op.cuh -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/unified_tensor.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/unified_tensor.cu -------------------------------------------------------------------------------- /graphlearn_torch/csrc/cuda/weighted_sampler.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/cuda/weighted_sampler.cuh -------------------------------------------------------------------------------- /graphlearn_torch/csrc/sample_queue.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/sample_queue.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/shm_queue.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/shm_queue.cc -------------------------------------------------------------------------------- /graphlearn_torch/csrc/tensor_map.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/csrc/tensor_map.cc -------------------------------------------------------------------------------- /graphlearn_torch/include/common.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/common.cuh -------------------------------------------------------------------------------- /graphlearn_torch/include/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/common.h -------------------------------------------------------------------------------- /graphlearn_torch/include/func_factory.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/func_factory.h -------------------------------------------------------------------------------- /graphlearn_torch/include/graph.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/graph.h -------------------------------------------------------------------------------- /graphlearn_torch/include/hash_table.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/hash_table.cuh -------------------------------------------------------------------------------- /graphlearn_torch/include/inducer_base.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/inducer_base.h -------------------------------------------------------------------------------- /graphlearn_torch/include/negative_sampler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/negative_sampler.h -------------------------------------------------------------------------------- /graphlearn_torch/include/sample_queue.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/sample_queue.h -------------------------------------------------------------------------------- /graphlearn_torch/include/sampler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/sampler.h -------------------------------------------------------------------------------- /graphlearn_torch/include/shm_queue.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/shm_queue.h -------------------------------------------------------------------------------- /graphlearn_torch/include/stitch_sample_results.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/stitch_sample_results.h -------------------------------------------------------------------------------- /graphlearn_torch/include/subgraph_op_base.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/subgraph_op_base.h -------------------------------------------------------------------------------- /graphlearn_torch/include/tensor_map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/tensor_map.h -------------------------------------------------------------------------------- /graphlearn_torch/include/types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/types.h -------------------------------------------------------------------------------- /graphlearn_torch/include/unified_tensor.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/include/unified_tensor.cuh -------------------------------------------------------------------------------- /graphlearn_torch/python/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/__init__.py -------------------------------------------------------------------------------- /graphlearn_torch/python/channel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/channel/__init__.py -------------------------------------------------------------------------------- /graphlearn_torch/python/channel/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/channel/base.py -------------------------------------------------------------------------------- /graphlearn_torch/python/channel/mp_channel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/channel/mp_channel.py -------------------------------------------------------------------------------- /graphlearn_torch/python/channel/remote_channel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/channel/remote_channel.py -------------------------------------------------------------------------------- /graphlearn_torch/python/channel/shm_channel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/channel/shm_channel.py -------------------------------------------------------------------------------- /graphlearn_torch/python/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/data/__init__.py -------------------------------------------------------------------------------- /graphlearn_torch/python/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/data/dataset.py -------------------------------------------------------------------------------- /graphlearn_torch/python/data/feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/data/feature.py -------------------------------------------------------------------------------- /graphlearn_torch/python/data/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/data/graph.py -------------------------------------------------------------------------------- /graphlearn_torch/python/data/reorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/data/reorder.py -------------------------------------------------------------------------------- /graphlearn_torch/python/data/table_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/data/table_dataset.py -------------------------------------------------------------------------------- /graphlearn_torch/python/data/unified_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/data/unified_tensor.py -------------------------------------------------------------------------------- /graphlearn_torch/python/data/vineyard_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/data/vineyard_utils.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/__init__.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_client.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_context.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_dataset.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_feature.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_graph.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_link_neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_link_neighbor_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_neighbor_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_neighbor_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_neighbor_sampler.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_options.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_random_partitioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_random_partitioner.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_sampling_producer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_sampling_producer.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_server.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_subgraph_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_subgraph_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/dist_table_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/dist_table_dataset.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/event_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/event_loop.py -------------------------------------------------------------------------------- /graphlearn_torch/python/distributed/rpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/distributed/rpc.py -------------------------------------------------------------------------------- /graphlearn_torch/python/loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/loader/__init__.py -------------------------------------------------------------------------------- /graphlearn_torch/python/loader/link_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/loader/link_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/loader/link_neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/loader/link_neighbor_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/loader/neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/loader/neighbor_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/loader/node_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/loader/node_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/loader/subgraph_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/loader/subgraph_loader.py -------------------------------------------------------------------------------- /graphlearn_torch/python/loader/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/loader/transform.py -------------------------------------------------------------------------------- /graphlearn_torch/python/partition/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/partition/__init__.py -------------------------------------------------------------------------------- /graphlearn_torch/python/partition/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/partition/base.py -------------------------------------------------------------------------------- /graphlearn_torch/python/partition/frequency_partitioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/partition/frequency_partitioner.py -------------------------------------------------------------------------------- /graphlearn_torch/python/partition/partition_book.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/partition/partition_book.py -------------------------------------------------------------------------------- /graphlearn_torch/python/partition/random_partitioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/partition/random_partitioner.py -------------------------------------------------------------------------------- /graphlearn_torch/python/py_export_glt.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/py_export_glt.cc -------------------------------------------------------------------------------- /graphlearn_torch/python/py_export_v6d.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/py_export_v6d.cc -------------------------------------------------------------------------------- /graphlearn_torch/python/sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/sampler/__init__.py -------------------------------------------------------------------------------- /graphlearn_torch/python/sampler/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/sampler/base.py -------------------------------------------------------------------------------- /graphlearn_torch/python/sampler/negative_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/sampler/negative_sampler.py -------------------------------------------------------------------------------- /graphlearn_torch/python/sampler/neighbor_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/sampler/neighbor_sampler.py -------------------------------------------------------------------------------- /graphlearn_torch/python/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/typing.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/__init__.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/build_glt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/build_glt.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/common.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/device.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/exit_status.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/exit_status.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/mixin.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/singleton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/singleton.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/tensor.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/topo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/topo.py -------------------------------------------------------------------------------- /graphlearn_torch/python/utils/units.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/python/utils/units.py -------------------------------------------------------------------------------- /graphlearn_torch/v6d/vineyard_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/v6d/vineyard_utils.cc -------------------------------------------------------------------------------- /graphlearn_torch/v6d/vineyard_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/graphlearn_torch/v6d/vineyard_utils.h -------------------------------------------------------------------------------- /install_dependencies.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/install_dependencies.sh -------------------------------------------------------------------------------- /scripts/build_wheel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/scripts/build_wheel.sh -------------------------------------------------------------------------------- /scripts/launch_container.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/scripts/launch_container.sh -------------------------------------------------------------------------------- /scripts/remove_container.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/scripts/remove_container.sh -------------------------------------------------------------------------------- /scripts/run_cpp_ut.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/scripts/run_cpp_ut.sh -------------------------------------------------------------------------------- /scripts/run_python_ut.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/scripts/run_python_ut.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/setup.py -------------------------------------------------------------------------------- /test/cpp/test_graph.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_graph.cu -------------------------------------------------------------------------------- /test/cpp/test_hash_table.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_hash_table.cu -------------------------------------------------------------------------------- /test/cpp/test_hetero_inducer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_hetero_inducer.cu -------------------------------------------------------------------------------- /test/cpp/test_inducer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_inducer.cu -------------------------------------------------------------------------------- /test/cpp/test_random_negative_sampler.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_random_negative_sampler.cu -------------------------------------------------------------------------------- /test/cpp/test_random_sampler.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_random_sampler.cu -------------------------------------------------------------------------------- /test/cpp/test_shm_queue.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_shm_queue.cu -------------------------------------------------------------------------------- /test/cpp/test_stitch_sample_results.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_stitch_sample_results.cu -------------------------------------------------------------------------------- /test/cpp/test_subgraph.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_subgraph.cu -------------------------------------------------------------------------------- /test/cpp/test_tensor_map_serializer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_tensor_map_serializer.cu -------------------------------------------------------------------------------- /test/cpp/test_unified_tensor.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_unified_tensor.cu -------------------------------------------------------------------------------- /test/cpp/test_vineyard.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_vineyard.cc -------------------------------------------------------------------------------- /test/cpp/test_vineyard.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/cpp/test_vineyard.cu -------------------------------------------------------------------------------- /test/python/dist_test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/dist_test_utils.py -------------------------------------------------------------------------------- /test/python/test_dist_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_dist_feature.py -------------------------------------------------------------------------------- /test/python/test_dist_link_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_dist_link_loader.py -------------------------------------------------------------------------------- /test/python/test_dist_neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_dist_neighbor_loader.py -------------------------------------------------------------------------------- /test/python/test_dist_random_partitioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_dist_random_partitioner.py -------------------------------------------------------------------------------- /test/python/test_dist_subgraph_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_dist_subgraph_loader.py -------------------------------------------------------------------------------- /test/python/test_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_feature.py -------------------------------------------------------------------------------- /test/python/test_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_graph.py -------------------------------------------------------------------------------- /test/python/test_hetero_neighbor_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_hetero_neighbor_sampler.py -------------------------------------------------------------------------------- /test/python/test_link_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_link_loader.py -------------------------------------------------------------------------------- /test/python/test_neighbor_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_neighbor_sampler.py -------------------------------------------------------------------------------- /test/python/test_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_partition.py -------------------------------------------------------------------------------- /test/python/test_pyg_remote_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_pyg_remote_backend.py -------------------------------------------------------------------------------- /test/python/test_sample_prob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_sample_prob.py -------------------------------------------------------------------------------- /test/python/test_shm_channel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_shm_channel.py -------------------------------------------------------------------------------- /test/python/test_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_subgraph.py -------------------------------------------------------------------------------- /test/python/test_unified_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_unified_tensor.py -------------------------------------------------------------------------------- /test/python/test_vineyard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/test_vineyard.py -------------------------------------------------------------------------------- /test/python/vineyard_data/modern_graph/created.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/vineyard_data/modern_graph/created.csv -------------------------------------------------------------------------------- /test/python/vineyard_data/modern_graph/knows.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/vineyard_data/modern_graph/knows.csv -------------------------------------------------------------------------------- /test/python/vineyard_data/modern_graph/person.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/vineyard_data/modern_graph/person.csv -------------------------------------------------------------------------------- /test/python/vineyard_data/modern_graph/software.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/test/python/vineyard_data/modern_graph/software.csv -------------------------------------------------------------------------------- /third_party/googletest/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alibaba/graphlearn-for-pytorch/HEAD/third_party/googletest/build.sh --------------------------------------------------------------------------------