├── .devcontainer ├── Dockerfile ├── README.md ├── cuda12.9-conda │ └── devcontainer.json ├── cuda12.9-pip │ └── devcontainer.json ├── cuda13.0-conda │ └── devcontainer.json └── cuda13.0-pip │ └── devcontainer.json ├── .github ├── CODEOWNERS ├── copy-pr-bot.yaml ├── ops-bot.yaml ├── release.yml └── workflows │ ├── build.yaml │ ├── pr.yaml │ ├── test.yaml │ └── trigger-breaking-change-alert.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── CHANGELOG.md ├── LICENSE ├── RAPIDS_BRANCH ├── README.md ├── VERSION ├── ci ├── build_cpp.sh ├── build_docs.sh ├── build_python.sh ├── build_wheel.sh ├── build_wheel_cugraph-pyg.sh ├── build_wheel_libwholegraph.sh ├── build_wheel_pylibwholegraph.sh ├── check_style.sh ├── notebook_list.py ├── release │ └── update-version.sh ├── run_ctests.sh ├── run_cugraph_pyg_pytests.sh ├── run_pylibwholegraph_pytests.sh ├── test.sh ├── test_cpp.sh ├── test_notebooks.sh ├── test_python.sh ├── test_wheel_cugraph-pyg.sh ├── test_wheel_pylibwholegraph.sh ├── utils │ ├── git_helpers.py │ ├── nbtest.sh │ └── nbtestlog2junitxml.py └── validate_wheel.sh ├── cmake ├── RAPIDS.cmake └── rapids_config.cmake ├── codecov.yml ├── conda ├── environments │ ├── all_cuda-129_arch-aarch64.yaml │ ├── all_cuda-129_arch-x86_64.yaml │ ├── all_cuda-130_arch-aarch64.yaml │ └── all_cuda-130_arch-x86_64.yaml └── recipes │ ├── cugraph-pyg │ └── recipe.yaml │ ├── libwholegraph │ ├── conda_build_config.yaml │ └── recipe.yaml │ └── pylibwholegraph │ ├── conda_build_config.yaml │ └── recipe.yaml ├── cpp ├── .clang-format ├── .clang-tidy ├── CMakeLists.txt ├── Doxyfile ├── bench │ ├── CMakeLists.txt │ ├── common │ │ ├── wholegraph_benchmark.cpp │ │ └── wholegraph_benchmark.hpp │ └── wholememory_ops │ │ └── gather_scatter_bench.cu ├── cmake │ ├── config.json │ └── thirdparty │ │ ├── get_nccl.cmake │ │ ├── get_nvshmem.cmake │ │ └── get_raft.cmake ├── docs │ └── DEVELOPER_GUIDE.md ├── doxygen │ ├── header.html │ └── main_page.md ├── include │ └── wholememory │ │ ├── device_reference.cuh │ │ ├── embedding.h │ │ ├── env_func_ptrs.h │ │ ├── global_reference.h │ │ ├── graph_op.h │ │ ├── tensor_description.h │ │ ├── wholegraph_op.h │ │ ├── wholememory.h │ │ ├── wholememory_op.h │ │ └── wholememory_tensor.h ├── scripts │ └── run-cmake-format.sh ├── src │ ├── cuda_macros.cpp │ ├── cuda_macros.hpp │ ├── error.hpp │ ├── graph_ops │ │ ├── append_unique.cpp │ │ ├── append_unique_func.cuh │ │ ├── append_unique_impl.cu │ │ ├── append_unique_impl.h │ │ ├── csr_add_self_loop.cpp │ │ ├── csr_add_self_loop_func.cuh │ │ ├── csr_add_self_loop_impl.cu │ │ └── csr_add_self_loop_impl.h │ ├── logger.cpp │ ├── logger.hpp │ ├── net_utils.cpp │ ├── net_utils.h │ ├── nvml_wrap.cpp │ ├── nvml_wrap.h │ ├── nvshmem_plugins │ │ └── nvshmem_wholememory_bootstrap.cpp │ ├── parallel_utils.cpp │ ├── parallel_utils.hpp │ ├── wholegraph_ops │ │ ├── raft_random_gen.cu │ │ ├── sample_comm.cuh │ │ ├── unweighted_sample_without_replacement.cpp │ │ ├── unweighted_sample_without_replacement_func.cuh │ │ ├── unweighted_sample_without_replacement_impl.h │ │ ├── unweighted_sample_without_replacement_impl_mapped.cu │ │ ├── unweighted_sample_without_replacement_impl_nccl.cu │ │ ├── unweighted_sample_without_replacement_nccl_func.cuh │ │ ├── weighted_sample_without_replacement.cpp │ │ ├── weighted_sample_without_replacement_func.cuh │ │ ├── weighted_sample_without_replacement_impl.h │ │ └── weighted_sample_without_replacement_impl_mapped.cu │ ├── wholememory │ │ ├── communicator.cpp │ │ ├── communicator.hpp │ │ ├── embedding.cpp │ │ ├── embedding.hpp │ │ ├── embedding_cache.cpp │ │ ├── embedding_cache.hpp │ │ ├── embedding_optimizer.cpp │ │ ├── embedding_optimizer.hpp │ │ ├── env_func_ptrs.cpp │ │ ├── env_func_ptrs.hpp │ │ ├── file_io.cpp │ │ ├── file_io.h │ │ ├── global_reference.cpp │ │ ├── initialize.cpp │ │ ├── initialize.hpp │ │ ├── integer_utils.hpp │ │ ├── memory_handle.cpp │ │ ├── memory_handle.hpp │ │ ├── nccl_comms.cpp │ │ ├── nccl_comms.hpp │ │ ├── system_info.cpp │ │ ├── system_info.hpp │ │ ├── tensor_description.cpp │ │ ├── wholememory.cpp │ │ └── wholememory_tensor.cpp │ └── wholememory_ops │ │ ├── functions │ │ ├── bucket_ids_for_hierarchy_func.cu │ │ ├── bucket_ids_for_hierarchy_func.h │ │ ├── bucket_ids_func.cu │ │ ├── bucket_ids_func.h │ │ ├── embedding_cache_func.cu │ │ ├── embedding_cache_func.cuh │ │ ├── embedding_cache_func.h │ │ ├── embedding_optimizer_func.cu │ │ ├── embedding_optimizer_func.h │ │ ├── exchange_embeddings_nccl_func.cu │ │ ├── exchange_embeddings_nccl_func.h │ │ ├── exchange_ids_nccl_func.cu │ │ ├── exchange_ids_nccl_func.h │ │ ├── gather_cached_func.cu │ │ ├── gather_cached_func.h │ │ ├── gather_func.cu │ │ ├── gather_func_impl_floating_data_int32_indices.cu │ │ ├── gather_func_impl_floating_data_int64_indices.cu │ │ ├── gather_func_impl_integer_data_int32_indices.cu │ │ ├── gather_func_impl_integer_data_int64_indices.cu │ │ ├── gather_scatter_func.cuh │ │ ├── gather_scatter_func.h │ │ ├── map_indices_func.cu │ │ ├── map_indices_func.h │ │ ├── nvshmem_device_reference.cuh │ │ ├── nvshmem_gather_func_impl_floating_data_int32_indices.cu │ │ ├── nvshmem_gather_func_impl_floating_data_int64_indices.cu │ │ ├── nvshmem_gather_func_impl_integer_data_int32_indices.cu │ │ ├── nvshmem_gather_func_impl_integer_data_int64_indices.cu │ │ ├── nvshmem_gather_scatter_func.cuh │ │ ├── nvshmem_scatter_func_impl_floating_data_int32_indices.cu │ │ ├── nvshmem_scatter_func_impl_floating_data_int64_indices.cu │ │ ├── nvshmem_scatter_func_impl_integer_data_int32_indices.cu │ │ ├── nvshmem_scatter_func_impl_integer_data_int64_indices.cu │ │ ├── nvshmem_template.cuh │ │ ├── scatter_func.cu │ │ ├── scatter_func_impl_floating_data_int32_indices.cu │ │ ├── scatter_func_impl_floating_data_int64_indices.cu │ │ ├── scatter_func_impl_integer_data_int32_indices.cu │ │ ├── scatter_func_impl_integer_data_int64_indices.cu │ │ ├── sort_indices_func.cu │ │ ├── sort_indices_func.h │ │ ├── sort_unique_ids_for_hierarchy_func.cu │ │ ├── sort_unique_ids_for_hierarchy_func.h │ │ ├── sort_unique_indices_func.cu │ │ └── sort_unique_indices_func.h │ │ ├── gather_op.cpp │ │ ├── gather_op_impl.h │ │ ├── gather_op_impl_hierarchy.cu │ │ ├── gather_op_impl_mapped.cu │ │ ├── gather_op_impl_nccl.cu │ │ ├── gather_op_impl_nvshmem.cu │ │ ├── output_memory_handle.hpp │ │ ├── register.hpp │ │ ├── scatter_op.cpp │ │ ├── scatter_op_impl.h │ │ ├── scatter_op_impl.nvshmem.cu │ │ ├── scatter_op_impl_mapped.cu │ │ ├── scatter_op_impl_nccl.cu │ │ ├── temp_memory_handle.hpp │ │ ├── thrust_allocator.cpp │ │ ├── thrust_allocator.hpp │ │ └── wholememory_test_op.cu └── tests │ ├── CMakeLists.txt │ ├── graph_ops │ ├── append_unique_test_utils.cu │ ├── append_unique_test_utils.hpp │ ├── append_unique_tests.cu │ ├── csr_add_self_loop_tests.cu │ ├── csr_add_self_loop_utils.cu │ └── csr_add_self_loop_utils.hpp │ ├── parallel_utils_tests.cpp │ ├── wholegraph_ops │ ├── graph_sampling_test_utils.cu │ ├── graph_sampling_test_utils.hpp │ ├── wholegraph_csr_unweighted_sample_without_replacement_tests.cu │ └── wholegraph_csr_weighted_sample_without_replacement_tests.cu │ ├── wholememory │ ├── wholememory_comm_tests.cpp │ ├── wholememory_handle_tests.cpp │ ├── wholememory_tensor_tests.cpp │ └── wholememory_test_utils.hpp │ └── wholememory_ops │ ├── cacheset_tests.cu │ ├── embedding_test_utils.cu │ ├── embedding_test_utils.hpp │ ├── wholememory_embedding_gradient_apply_tests.cu │ ├── wholememory_embedding_tests.cu │ ├── wholememory_gather_tests.cu │ └── wholememory_scatter_tests.cu ├── datasets ├── get_test_data.sh └── karate.csv ├── dependencies.yaml ├── img ├── GDF_community.png ├── Scaling.png ├── cugraph-pyg-stack.png ├── cugraph-stack.png ├── cugraph_logo_2.png ├── cugraphops_context.png ├── gaas_img_1.png ├── gaas_img_2.png ├── gnn_blog.png ├── gnn_context.png ├── gnn_framework.png ├── goai_logo.png ├── pg_example.png ├── rapids_arrow.png └── rapids_logo.png ├── mg_utils ├── README.md ├── default-config.sh ├── functions.sh ├── run-dask-process.sh └── wait_for_workers.py ├── print_env.sh ├── pyproject.toml ├── python ├── .coveragerc ├── cugraph-pyg │ ├── LICENSE │ ├── README.md │ ├── conda │ │ ├── cugraph_pyg_dev_cuda-129_arch-aarch64.yaml │ │ ├── cugraph_pyg_dev_cuda-129_arch-x86_64.yaml │ │ ├── cugraph_pyg_dev_cuda-130_arch-aarch64.yaml │ │ └── cugraph_pyg_dev_cuda-130_arch-x86_64.yaml │ ├── cugraph_pyg │ │ ├── VERSION │ │ ├── __init__.py │ │ ├── _version.py │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── feature_store.py │ │ │ └── graph_store.py │ │ ├── examples │ │ │ ├── dist_gin_sg.py │ │ │ ├── fraud │ │ │ │ ├── README.md │ │ │ │ ├── bitcoin_mnmg.py │ │ │ │ └── bitcoin_rf.py │ │ │ ├── gcn_dist_mnmg.py │ │ │ ├── kg │ │ │ │ ├── renumber_kg.py │ │ │ │ └── run_renumber.sh │ │ │ ├── movielens_mnmg.py │ │ │ ├── plc │ │ │ │ └── pylibcugraph_mg.py │ │ │ ├── rgcn_link_class_mnmg.py │ │ │ └── taobao_mnmg.py │ │ ├── loader │ │ │ ├── __init__.py │ │ │ ├── link_loader.py │ │ │ ├── link_neighbor_loader.py │ │ │ ├── neighbor_loader.py │ │ │ ├── node_loader.py │ │ │ └── utils.py │ │ ├── sampler │ │ │ ├── __init__.py │ │ │ ├── distributed_sampler.py │ │ │ ├── io │ │ │ │ ├── __init__.py │ │ │ │ └── reader.py │ │ │ ├── sampler.py │ │ │ └── sampler_utils.py │ │ ├── tensor │ │ │ ├── __init__.py │ │ │ ├── dist_matrix.py │ │ │ ├── dist_tensor.py │ │ │ └── utils.py │ │ ├── tests │ │ │ ├── conftest.py │ │ │ ├── data │ │ │ │ ├── test_feature_store.py │ │ │ │ ├── test_feature_store_mg.py │ │ │ │ ├── test_graph_store.py │ │ │ │ └── test_graph_store_mg.py │ │ │ ├── loader │ │ │ │ ├── test_neighbor_loader.py │ │ │ │ └── test_neighbor_loader_mg.py │ │ │ ├── sampler │ │ │ │ └── test_distributed_sampler.py │ │ │ ├── tensor │ │ │ │ ├── test_dist_matrix_mg.py │ │ │ │ └── test_dist_tensor_mg.py │ │ │ └── test_version.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ └── imports.py │ ├── pyproject.toml │ └── pytest.ini ├── libwholegraph │ ├── CMakeLists.txt │ ├── README.md │ ├── libwholegraph │ │ ├── VERSION │ │ ├── __init__.py │ │ ├── _version.py │ │ └── load.py │ └── pyproject.toml └── pylibwholegraph │ ├── CMakeLists.txt │ ├── LICENSE │ ├── README.md │ ├── examples │ ├── node_classfication.py │ └── ogbn_papers100m_convert.py │ ├── pylibwholegraph │ ├── VERSION │ ├── __init__.py │ ├── _version.py │ ├── binding │ │ ├── CMakeLists.txt │ │ ├── __init__.py │ │ └── wholememory_binding.pyx │ ├── test_utils │ │ ├── __init__.py │ │ └── test_comm.py │ ├── tests │ │ ├── pylibwholegraph │ │ │ ├── test_wholememory_binding.py │ │ │ ├── test_wholememory_io.py │ │ │ └── test_wholememory_tensor.py │ │ ├── test_version.py │ │ └── wholegraph_torch │ │ │ └── ops │ │ │ ├── test_graph_add_csr_self_loop.py │ │ │ ├── test_graph_append_unique.py │ │ │ ├── test_wholegraph_gather_scatter.py │ │ │ ├── test_wholegraph_unweighted_sample_without_replacement.py │ │ │ ├── test_wholegraph_weighted_sample_without_replacement.py │ │ │ └── test_wholememory_cython_binding.py │ ├── torch │ │ ├── __init__.py │ │ ├── comm.py │ │ ├── common_options.py │ │ ├── data_loader.py │ │ ├── distributed_launch.py │ │ ├── dlpack_utils.py │ │ ├── embedding.py │ │ ├── gnn_model.py │ │ ├── graph_ops.py │ │ ├── graph_structure.py │ │ ├── initialize.py │ │ ├── tensor.py │ │ ├── utils.py │ │ ├── wholegraph_env.py │ │ ├── wholegraph_ops.py │ │ └── wholememory_ops.py │ ├── torch_cpp_ext │ │ ├── torch_env_func_ptrs.cpp │ │ ├── torch_env_func_ptrs.h │ │ ├── torch_utils.cpp │ │ ├── torch_utils.h │ │ └── wholegraph_torch_ext.cpp │ └── utils │ │ ├── __init__.py │ │ └── multiprocess.py │ └── pyproject.toml ├── readme_pages ├── CONTRIBUTING.md ├── PRTAGS.md ├── cugraph_pyg.md ├── gnn_support.md ├── news.md └── performance │ └── performance.md └── thirdparty └── LICENSES └── LICENSE.cugraph-ops.NVIDIA /.devcontainer/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.devcontainer/Dockerfile -------------------------------------------------------------------------------- /.devcontainer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.devcontainer/README.md -------------------------------------------------------------------------------- /.devcontainer/cuda12.9-conda/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.devcontainer/cuda12.9-conda/devcontainer.json -------------------------------------------------------------------------------- /.devcontainer/cuda12.9-pip/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.devcontainer/cuda12.9-pip/devcontainer.json -------------------------------------------------------------------------------- /.devcontainer/cuda13.0-conda/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.devcontainer/cuda13.0-conda/devcontainer.json -------------------------------------------------------------------------------- /.devcontainer/cuda13.0-pip/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.devcontainer/cuda13.0-pip/devcontainer.json -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/copy-pr-bot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.github/copy-pr-bot.yaml -------------------------------------------------------------------------------- /.github/ops-bot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.github/ops-bot.yaml -------------------------------------------------------------------------------- /.github/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.github/release.yml -------------------------------------------------------------------------------- /.github/workflows/build.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.github/workflows/build.yaml -------------------------------------------------------------------------------- /.github/workflows/pr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.github/workflows/pr.yaml -------------------------------------------------------------------------------- /.github/workflows/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.github/workflows/test.yaml -------------------------------------------------------------------------------- /.github/workflows/trigger-breaking-change-alert.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.github/workflows/trigger-breaking-change-alert.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/LICENSE -------------------------------------------------------------------------------- /RAPIDS_BRANCH: -------------------------------------------------------------------------------- 1 | main 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 26.02.00 2 | -------------------------------------------------------------------------------- /ci/build_cpp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/build_cpp.sh -------------------------------------------------------------------------------- /ci/build_docs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/build_docs.sh -------------------------------------------------------------------------------- /ci/build_python.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/build_python.sh -------------------------------------------------------------------------------- /ci/build_wheel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/build_wheel.sh -------------------------------------------------------------------------------- /ci/build_wheel_cugraph-pyg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/build_wheel_cugraph-pyg.sh -------------------------------------------------------------------------------- /ci/build_wheel_libwholegraph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/build_wheel_libwholegraph.sh -------------------------------------------------------------------------------- /ci/build_wheel_pylibwholegraph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/build_wheel_pylibwholegraph.sh -------------------------------------------------------------------------------- /ci/check_style.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/check_style.sh -------------------------------------------------------------------------------- /ci/notebook_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/notebook_list.py -------------------------------------------------------------------------------- /ci/release/update-version.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/release/update-version.sh -------------------------------------------------------------------------------- /ci/run_ctests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/run_ctests.sh -------------------------------------------------------------------------------- /ci/run_cugraph_pyg_pytests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/run_cugraph_pyg_pytests.sh -------------------------------------------------------------------------------- /ci/run_pylibwholegraph_pytests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/run_pylibwholegraph_pytests.sh -------------------------------------------------------------------------------- /ci/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/test.sh -------------------------------------------------------------------------------- /ci/test_cpp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/test_cpp.sh -------------------------------------------------------------------------------- /ci/test_notebooks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/test_notebooks.sh -------------------------------------------------------------------------------- /ci/test_python.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/test_python.sh -------------------------------------------------------------------------------- /ci/test_wheel_cugraph-pyg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/test_wheel_cugraph-pyg.sh -------------------------------------------------------------------------------- /ci/test_wheel_pylibwholegraph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/test_wheel_pylibwholegraph.sh -------------------------------------------------------------------------------- /ci/utils/git_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/utils/git_helpers.py -------------------------------------------------------------------------------- /ci/utils/nbtest.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/utils/nbtest.sh -------------------------------------------------------------------------------- /ci/utils/nbtestlog2junitxml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/utils/nbtestlog2junitxml.py -------------------------------------------------------------------------------- /ci/validate_wheel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/ci/validate_wheel.sh -------------------------------------------------------------------------------- /cmake/RAPIDS.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cmake/RAPIDS.cmake -------------------------------------------------------------------------------- /cmake/rapids_config.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cmake/rapids_config.cmake -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/codecov.yml -------------------------------------------------------------------------------- /conda/environments/all_cuda-129_arch-aarch64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/environments/all_cuda-129_arch-aarch64.yaml -------------------------------------------------------------------------------- /conda/environments/all_cuda-129_arch-x86_64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/environments/all_cuda-129_arch-x86_64.yaml -------------------------------------------------------------------------------- /conda/environments/all_cuda-130_arch-aarch64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/environments/all_cuda-130_arch-aarch64.yaml -------------------------------------------------------------------------------- /conda/environments/all_cuda-130_arch-x86_64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/environments/all_cuda-130_arch-x86_64.yaml -------------------------------------------------------------------------------- /conda/recipes/cugraph-pyg/recipe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/recipes/cugraph-pyg/recipe.yaml -------------------------------------------------------------------------------- /conda/recipes/libwholegraph/conda_build_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/recipes/libwholegraph/conda_build_config.yaml -------------------------------------------------------------------------------- /conda/recipes/libwholegraph/recipe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/recipes/libwholegraph/recipe.yaml -------------------------------------------------------------------------------- /conda/recipes/pylibwholegraph/conda_build_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/recipes/pylibwholegraph/conda_build_config.yaml -------------------------------------------------------------------------------- /conda/recipes/pylibwholegraph/recipe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/conda/recipes/pylibwholegraph/recipe.yaml -------------------------------------------------------------------------------- /cpp/.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/.clang-format -------------------------------------------------------------------------------- /cpp/.clang-tidy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/.clang-tidy -------------------------------------------------------------------------------- /cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/Doxyfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/Doxyfile -------------------------------------------------------------------------------- /cpp/bench/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/bench/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/bench/common/wholegraph_benchmark.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/bench/common/wholegraph_benchmark.cpp -------------------------------------------------------------------------------- /cpp/bench/common/wholegraph_benchmark.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/bench/common/wholegraph_benchmark.hpp -------------------------------------------------------------------------------- /cpp/bench/wholememory_ops/gather_scatter_bench.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/bench/wholememory_ops/gather_scatter_bench.cu -------------------------------------------------------------------------------- /cpp/cmake/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/cmake/config.json -------------------------------------------------------------------------------- /cpp/cmake/thirdparty/get_nccl.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/cmake/thirdparty/get_nccl.cmake -------------------------------------------------------------------------------- /cpp/cmake/thirdparty/get_nvshmem.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/cmake/thirdparty/get_nvshmem.cmake -------------------------------------------------------------------------------- /cpp/cmake/thirdparty/get_raft.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/cmake/thirdparty/get_raft.cmake -------------------------------------------------------------------------------- /cpp/docs/DEVELOPER_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/docs/DEVELOPER_GUIDE.md -------------------------------------------------------------------------------- /cpp/doxygen/header.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/doxygen/header.html -------------------------------------------------------------------------------- /cpp/doxygen/main_page.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/doxygen/main_page.md -------------------------------------------------------------------------------- /cpp/include/wholememory/device_reference.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/device_reference.cuh -------------------------------------------------------------------------------- /cpp/include/wholememory/embedding.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/embedding.h -------------------------------------------------------------------------------- /cpp/include/wholememory/env_func_ptrs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/env_func_ptrs.h -------------------------------------------------------------------------------- /cpp/include/wholememory/global_reference.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/global_reference.h -------------------------------------------------------------------------------- /cpp/include/wholememory/graph_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/graph_op.h -------------------------------------------------------------------------------- /cpp/include/wholememory/tensor_description.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/tensor_description.h -------------------------------------------------------------------------------- /cpp/include/wholememory/wholegraph_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/wholegraph_op.h -------------------------------------------------------------------------------- /cpp/include/wholememory/wholememory.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/wholememory.h -------------------------------------------------------------------------------- /cpp/include/wholememory/wholememory_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/wholememory_op.h -------------------------------------------------------------------------------- /cpp/include/wholememory/wholememory_tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/include/wholememory/wholememory_tensor.h -------------------------------------------------------------------------------- /cpp/scripts/run-cmake-format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/scripts/run-cmake-format.sh -------------------------------------------------------------------------------- /cpp/src/cuda_macros.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/cuda_macros.cpp -------------------------------------------------------------------------------- /cpp/src/cuda_macros.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/cuda_macros.hpp -------------------------------------------------------------------------------- /cpp/src/error.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/error.hpp -------------------------------------------------------------------------------- /cpp/src/graph_ops/append_unique.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/graph_ops/append_unique.cpp -------------------------------------------------------------------------------- /cpp/src/graph_ops/append_unique_func.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/graph_ops/append_unique_func.cuh -------------------------------------------------------------------------------- /cpp/src/graph_ops/append_unique_impl.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/graph_ops/append_unique_impl.cu -------------------------------------------------------------------------------- /cpp/src/graph_ops/append_unique_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/graph_ops/append_unique_impl.h -------------------------------------------------------------------------------- /cpp/src/graph_ops/csr_add_self_loop.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/graph_ops/csr_add_self_loop.cpp -------------------------------------------------------------------------------- /cpp/src/graph_ops/csr_add_self_loop_func.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/graph_ops/csr_add_self_loop_func.cuh -------------------------------------------------------------------------------- /cpp/src/graph_ops/csr_add_self_loop_impl.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/graph_ops/csr_add_self_loop_impl.cu -------------------------------------------------------------------------------- /cpp/src/graph_ops/csr_add_self_loop_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/graph_ops/csr_add_self_loop_impl.h -------------------------------------------------------------------------------- /cpp/src/logger.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/logger.cpp -------------------------------------------------------------------------------- /cpp/src/logger.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/logger.hpp -------------------------------------------------------------------------------- /cpp/src/net_utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/net_utils.cpp -------------------------------------------------------------------------------- /cpp/src/net_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/net_utils.h -------------------------------------------------------------------------------- /cpp/src/nvml_wrap.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/nvml_wrap.cpp -------------------------------------------------------------------------------- /cpp/src/nvml_wrap.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/nvml_wrap.h -------------------------------------------------------------------------------- /cpp/src/nvshmem_plugins/nvshmem_wholememory_bootstrap.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/nvshmem_plugins/nvshmem_wholememory_bootstrap.cpp -------------------------------------------------------------------------------- /cpp/src/parallel_utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/parallel_utils.cpp -------------------------------------------------------------------------------- /cpp/src/parallel_utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/parallel_utils.hpp -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/raft_random_gen.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/raft_random_gen.cu -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/sample_comm.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/sample_comm.cuh -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/unweighted_sample_without_replacement.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/unweighted_sample_without_replacement.cpp -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/unweighted_sample_without_replacement_func.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/unweighted_sample_without_replacement_func.cuh -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/unweighted_sample_without_replacement_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/unweighted_sample_without_replacement_impl.h -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/unweighted_sample_without_replacement_impl_mapped.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/unweighted_sample_without_replacement_impl_mapped.cu -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/unweighted_sample_without_replacement_impl_nccl.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/unweighted_sample_without_replacement_impl_nccl.cu -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/unweighted_sample_without_replacement_nccl_func.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/unweighted_sample_without_replacement_nccl_func.cuh -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/weighted_sample_without_replacement.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/weighted_sample_without_replacement.cpp -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/weighted_sample_without_replacement_func.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/weighted_sample_without_replacement_func.cuh -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/weighted_sample_without_replacement_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/weighted_sample_without_replacement_impl.h -------------------------------------------------------------------------------- /cpp/src/wholegraph_ops/weighted_sample_without_replacement_impl_mapped.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholegraph_ops/weighted_sample_without_replacement_impl_mapped.cu -------------------------------------------------------------------------------- /cpp/src/wholememory/communicator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/communicator.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/communicator.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/communicator.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/embedding.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/embedding.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/embedding.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/embedding.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/embedding_cache.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/embedding_cache.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/embedding_cache.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/embedding_cache.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/embedding_optimizer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/embedding_optimizer.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/embedding_optimizer.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/embedding_optimizer.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/env_func_ptrs.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/env_func_ptrs.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/env_func_ptrs.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/env_func_ptrs.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/file_io.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/file_io.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/file_io.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/file_io.h -------------------------------------------------------------------------------- /cpp/src/wholememory/global_reference.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/global_reference.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/initialize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/initialize.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/initialize.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/initialize.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/integer_utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/integer_utils.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/memory_handle.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/memory_handle.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/memory_handle.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/memory_handle.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/nccl_comms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/nccl_comms.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/nccl_comms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/nccl_comms.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/system_info.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/system_info.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/system_info.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/system_info.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory/tensor_description.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/tensor_description.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/wholememory.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/wholememory.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory/wholememory_tensor.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory/wholememory_tensor.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/bucket_ids_for_hierarchy_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/bucket_ids_for_hierarchy_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/bucket_ids_for_hierarchy_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/bucket_ids_for_hierarchy_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/bucket_ids_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/bucket_ids_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/bucket_ids_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/bucket_ids_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/embedding_cache_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/embedding_cache_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/embedding_cache_func.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/embedding_cache_func.cuh -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/embedding_cache_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/embedding_cache_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/embedding_optimizer_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/embedding_optimizer_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/embedding_optimizer_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/embedding_optimizer_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/exchange_embeddings_nccl_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/exchange_embeddings_nccl_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/exchange_embeddings_nccl_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/exchange_embeddings_nccl_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/exchange_ids_nccl_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/exchange_ids_nccl_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/exchange_ids_nccl_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/exchange_ids_nccl_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_cached_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_cached_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_cached_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_cached_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_func_impl_floating_data_int32_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_func_impl_floating_data_int32_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_func_impl_floating_data_int64_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_func_impl_floating_data_int64_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_func_impl_integer_data_int32_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_func_impl_integer_data_int32_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_func_impl_integer_data_int64_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_func_impl_integer_data_int64_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_scatter_func.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_scatter_func.cuh -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/gather_scatter_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/gather_scatter_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/map_indices_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/map_indices_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/map_indices_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/map_indices_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_device_reference.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_device_reference.cuh -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_gather_func_impl_floating_data_int32_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_gather_func_impl_floating_data_int32_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_gather_func_impl_floating_data_int64_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_gather_func_impl_floating_data_int64_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_gather_func_impl_integer_data_int32_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_gather_func_impl_integer_data_int32_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_gather_func_impl_integer_data_int64_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_gather_func_impl_integer_data_int64_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_gather_scatter_func.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_gather_scatter_func.cuh -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_scatter_func_impl_floating_data_int32_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_scatter_func_impl_floating_data_int32_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_scatter_func_impl_floating_data_int64_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_scatter_func_impl_floating_data_int64_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_scatter_func_impl_integer_data_int32_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_scatter_func_impl_integer_data_int32_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_scatter_func_impl_integer_data_int64_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_scatter_func_impl_integer_data_int64_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/nvshmem_template.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/nvshmem_template.cuh -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/scatter_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/scatter_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/scatter_func_impl_floating_data_int32_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/scatter_func_impl_floating_data_int32_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/scatter_func_impl_floating_data_int64_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/scatter_func_impl_floating_data_int64_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/scatter_func_impl_integer_data_int32_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/scatter_func_impl_integer_data_int32_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/scatter_func_impl_integer_data_int64_indices.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/scatter_func_impl_integer_data_int64_indices.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/sort_indices_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/sort_indices_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/sort_indices_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/sort_indices_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/sort_unique_ids_for_hierarchy_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/sort_unique_ids_for_hierarchy_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/sort_unique_ids_for_hierarchy_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/sort_unique_ids_for_hierarchy_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/sort_unique_indices_func.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/sort_unique_indices_func.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/functions/sort_unique_indices_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/functions/sort_unique_indices_func.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/gather_op.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/gather_op.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/gather_op_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/gather_op_impl.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/gather_op_impl_hierarchy.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/gather_op_impl_hierarchy.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/gather_op_impl_mapped.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/gather_op_impl_mapped.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/gather_op_impl_nccl.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/gather_op_impl_nccl.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/gather_op_impl_nvshmem.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/gather_op_impl_nvshmem.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/output_memory_handle.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/output_memory_handle.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/register.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/register.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/scatter_op.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/scatter_op.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/scatter_op_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/scatter_op_impl.h -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/scatter_op_impl.nvshmem.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/scatter_op_impl.nvshmem.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/scatter_op_impl_mapped.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/scatter_op_impl_mapped.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/scatter_op_impl_nccl.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/scatter_op_impl_nccl.cu -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/temp_memory_handle.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/temp_memory_handle.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/thrust_allocator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/thrust_allocator.cpp -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/thrust_allocator.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/thrust_allocator.hpp -------------------------------------------------------------------------------- /cpp/src/wholememory_ops/wholememory_test_op.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/src/wholememory_ops/wholememory_test_op.cu -------------------------------------------------------------------------------- /cpp/tests/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/tests/graph_ops/append_unique_test_utils.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/graph_ops/append_unique_test_utils.cu -------------------------------------------------------------------------------- /cpp/tests/graph_ops/append_unique_test_utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/graph_ops/append_unique_test_utils.hpp -------------------------------------------------------------------------------- /cpp/tests/graph_ops/append_unique_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/graph_ops/append_unique_tests.cu -------------------------------------------------------------------------------- /cpp/tests/graph_ops/csr_add_self_loop_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/graph_ops/csr_add_self_loop_tests.cu -------------------------------------------------------------------------------- /cpp/tests/graph_ops/csr_add_self_loop_utils.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/graph_ops/csr_add_self_loop_utils.cu -------------------------------------------------------------------------------- /cpp/tests/graph_ops/csr_add_self_loop_utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/graph_ops/csr_add_self_loop_utils.hpp -------------------------------------------------------------------------------- /cpp/tests/parallel_utils_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/parallel_utils_tests.cpp -------------------------------------------------------------------------------- /cpp/tests/wholegraph_ops/graph_sampling_test_utils.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholegraph_ops/graph_sampling_test_utils.cu -------------------------------------------------------------------------------- /cpp/tests/wholegraph_ops/graph_sampling_test_utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholegraph_ops/graph_sampling_test_utils.hpp -------------------------------------------------------------------------------- /cpp/tests/wholegraph_ops/wholegraph_csr_unweighted_sample_without_replacement_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholegraph_ops/wholegraph_csr_unweighted_sample_without_replacement_tests.cu -------------------------------------------------------------------------------- /cpp/tests/wholegraph_ops/wholegraph_csr_weighted_sample_without_replacement_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholegraph_ops/wholegraph_csr_weighted_sample_without_replacement_tests.cu -------------------------------------------------------------------------------- /cpp/tests/wholememory/wholememory_comm_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory/wholememory_comm_tests.cpp -------------------------------------------------------------------------------- /cpp/tests/wholememory/wholememory_handle_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory/wholememory_handle_tests.cpp -------------------------------------------------------------------------------- /cpp/tests/wholememory/wholememory_tensor_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory/wholememory_tensor_tests.cpp -------------------------------------------------------------------------------- /cpp/tests/wholememory/wholememory_test_utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory/wholememory_test_utils.hpp -------------------------------------------------------------------------------- /cpp/tests/wholememory_ops/cacheset_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory_ops/cacheset_tests.cu -------------------------------------------------------------------------------- /cpp/tests/wholememory_ops/embedding_test_utils.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory_ops/embedding_test_utils.cu -------------------------------------------------------------------------------- /cpp/tests/wholememory_ops/embedding_test_utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory_ops/embedding_test_utils.hpp -------------------------------------------------------------------------------- /cpp/tests/wholememory_ops/wholememory_embedding_gradient_apply_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory_ops/wholememory_embedding_gradient_apply_tests.cu -------------------------------------------------------------------------------- /cpp/tests/wholememory_ops/wholememory_embedding_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory_ops/wholememory_embedding_tests.cu -------------------------------------------------------------------------------- /cpp/tests/wholememory_ops/wholememory_gather_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory_ops/wholememory_gather_tests.cu -------------------------------------------------------------------------------- /cpp/tests/wholememory_ops/wholememory_scatter_tests.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/cpp/tests/wholememory_ops/wholememory_scatter_tests.cu -------------------------------------------------------------------------------- /datasets/get_test_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/datasets/get_test_data.sh -------------------------------------------------------------------------------- /datasets/karate.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/datasets/karate.csv -------------------------------------------------------------------------------- /dependencies.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/dependencies.yaml -------------------------------------------------------------------------------- /img/GDF_community.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/GDF_community.png -------------------------------------------------------------------------------- /img/Scaling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/Scaling.png -------------------------------------------------------------------------------- /img/cugraph-pyg-stack.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/cugraph-pyg-stack.png -------------------------------------------------------------------------------- /img/cugraph-stack.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/cugraph-stack.png -------------------------------------------------------------------------------- /img/cugraph_logo_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/cugraph_logo_2.png -------------------------------------------------------------------------------- /img/cugraphops_context.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/cugraphops_context.png -------------------------------------------------------------------------------- /img/gaas_img_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/gaas_img_1.png -------------------------------------------------------------------------------- /img/gaas_img_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/gaas_img_2.png -------------------------------------------------------------------------------- /img/gnn_blog.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/gnn_blog.png -------------------------------------------------------------------------------- /img/gnn_context.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/gnn_context.png -------------------------------------------------------------------------------- /img/gnn_framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/gnn_framework.png -------------------------------------------------------------------------------- /img/goai_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/goai_logo.png -------------------------------------------------------------------------------- /img/pg_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/pg_example.png -------------------------------------------------------------------------------- /img/rapids_arrow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/rapids_arrow.png -------------------------------------------------------------------------------- /img/rapids_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/img/rapids_logo.png -------------------------------------------------------------------------------- /mg_utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/mg_utils/README.md -------------------------------------------------------------------------------- /mg_utils/default-config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/mg_utils/default-config.sh -------------------------------------------------------------------------------- /mg_utils/functions.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/mg_utils/functions.sh -------------------------------------------------------------------------------- /mg_utils/run-dask-process.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/mg_utils/run-dask-process.sh -------------------------------------------------------------------------------- /mg_utils/wait_for_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/mg_utils/wait_for_workers.py -------------------------------------------------------------------------------- /print_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/print_env.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/pyproject.toml -------------------------------------------------------------------------------- /python/.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/.coveragerc -------------------------------------------------------------------------------- /python/cugraph-pyg/LICENSE: -------------------------------------------------------------------------------- 1 | ../../LICENSE -------------------------------------------------------------------------------- /python/cugraph-pyg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/README.md -------------------------------------------------------------------------------- /python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-129_arch-aarch64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-129_arch-aarch64.yaml -------------------------------------------------------------------------------- /python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-129_arch-x86_64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-129_arch-x86_64.yaml -------------------------------------------------------------------------------- /python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-130_arch-aarch64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-130_arch-aarch64.yaml -------------------------------------------------------------------------------- /python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-130_arch-x86_64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-130_arch-x86_64.yaml -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/VERSION: -------------------------------------------------------------------------------- 1 | ../../../VERSION -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/__init__.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/_version.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/data/__init__.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/data/feature_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/data/feature_store.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/data/graph_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/data/graph_store.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/dist_gin_sg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/dist_gin_sg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/fraud/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/fraud/README.md -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/fraud/bitcoin_mnmg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/fraud/bitcoin_mnmg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/fraud/bitcoin_rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/fraud/bitcoin_rf.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_mnmg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_mnmg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/kg/renumber_kg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/kg/renumber_kg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/kg/run_renumber.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/kg/run_renumber.sh -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/movielens_mnmg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/movielens_mnmg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/plc/pylibcugraph_mg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/plc/pylibcugraph_mg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/rgcn_link_class_mnmg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/rgcn_link_class_mnmg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/examples/taobao_mnmg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/examples/taobao_mnmg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/loader/__init__.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/loader/link_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/loader/link_loader.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/loader/link_neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/loader/link_neighbor_loader.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/loader/neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/loader/neighbor_loader.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/loader/node_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/loader/node_loader.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/loader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/loader/utils.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/sampler/__init__.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/sampler/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/sampler/distributed_sampler.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/sampler/io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/sampler/io/__init__.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/sampler/io/reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/sampler/io/reader.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/sampler/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/sampler/sampler.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/sampler/sampler_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/sampler/sampler_utils.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tensor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tensor/__init__.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tensor/dist_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tensor/dist_matrix.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tensor/dist_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tensor/dist_tensor.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tensor/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tensor/utils.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/conftest.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/data/test_feature_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/data/test_feature_store.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/data/test_feature_store_mg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/data/test_feature_store_mg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/data/test_graph_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/data/test_graph_store.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/data/test_graph_store_mg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/data/test_graph_store_mg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/loader/test_neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/loader/test_neighbor_loader.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/loader/test_neighbor_loader_mg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/loader/test_neighbor_loader_mg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/sampler/test_distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/sampler/test_distributed_sampler.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/tensor/test_dist_matrix_mg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/tensor/test_dist_matrix_mg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/tensor/test_dist_tensor_mg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/tensor/test_dist_tensor_mg.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/tests/test_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/tests/test_version.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/utils/__init__.py -------------------------------------------------------------------------------- /python/cugraph-pyg/cugraph_pyg/utils/imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/cugraph_pyg/utils/imports.py -------------------------------------------------------------------------------- /python/cugraph-pyg/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/pyproject.toml -------------------------------------------------------------------------------- /python/cugraph-pyg/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/cugraph-pyg/pytest.ini -------------------------------------------------------------------------------- /python/libwholegraph/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/libwholegraph/CMakeLists.txt -------------------------------------------------------------------------------- /python/libwholegraph/README.md: -------------------------------------------------------------------------------- 1 | ../../README.md 2 | -------------------------------------------------------------------------------- /python/libwholegraph/libwholegraph/VERSION: -------------------------------------------------------------------------------- 1 | ../../../VERSION -------------------------------------------------------------------------------- /python/libwholegraph/libwholegraph/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/libwholegraph/libwholegraph/__init__.py -------------------------------------------------------------------------------- /python/libwholegraph/libwholegraph/_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/libwholegraph/libwholegraph/_version.py -------------------------------------------------------------------------------- /python/libwholegraph/libwholegraph/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/libwholegraph/libwholegraph/load.py -------------------------------------------------------------------------------- /python/libwholegraph/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/libwholegraph/pyproject.toml -------------------------------------------------------------------------------- /python/pylibwholegraph/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/CMakeLists.txt -------------------------------------------------------------------------------- /python/pylibwholegraph/LICENSE: -------------------------------------------------------------------------------- 1 | ../../LICENSE -------------------------------------------------------------------------------- /python/pylibwholegraph/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/README.md -------------------------------------------------------------------------------- /python/pylibwholegraph/examples/node_classfication.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/examples/node_classfication.py -------------------------------------------------------------------------------- /python/pylibwholegraph/examples/ogbn_papers100m_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/examples/ogbn_papers100m_convert.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/VERSION: -------------------------------------------------------------------------------- 1 | ../../../VERSION -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/__init__.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/_version.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/binding/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/binding/CMakeLists.txt -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/binding/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/binding/__init__.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/binding/wholememory_binding.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/binding/wholememory_binding.pyx -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/test_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/test_utils/__init__.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/test_utils/test_comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/test_utils/test_comm.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/pylibwholegraph/test_wholememory_binding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/pylibwholegraph/test_wholememory_binding.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/pylibwholegraph/test_wholememory_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/pylibwholegraph/test_wholememory_io.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/pylibwholegraph/test_wholememory_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/pylibwholegraph/test_wholememory_tensor.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/test_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/test_version.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_graph_add_csr_self_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_graph_add_csr_self_loop.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_graph_append_unique.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_graph_append_unique.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_gather_scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_gather_scatter.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_unweighted_sample_without_replacement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_unweighted_sample_without_replacement.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_weighted_sample_without_replacement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholegraph_weighted_sample_without_replacement.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholememory_cython_binding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/tests/wholegraph_torch/ops/test_wholememory_cython_binding.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/__init__.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/comm.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/common_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/common_options.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/data_loader.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/distributed_launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/distributed_launch.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/dlpack_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/dlpack_utils.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/embedding.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/gnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/gnn_model.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/graph_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/graph_ops.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/graph_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/graph_structure.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/initialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/initialize.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/tensor.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/utils.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/wholegraph_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/wholegraph_env.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/wholegraph_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/wholegraph_ops.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch/wholememory_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch/wholememory_ops.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/torch_env_func_ptrs.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/torch_env_func_ptrs.cpp -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/torch_env_func_ptrs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/torch_env_func_ptrs.h -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/torch_utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/torch_utils.cpp -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/torch_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/torch_utils.h -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/wholegraph_torch_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/torch_cpp_ext/wholegraph_torch_ext.cpp -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/utils/__init__.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pylibwholegraph/utils/multiprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pylibwholegraph/utils/multiprocess.py -------------------------------------------------------------------------------- /python/pylibwholegraph/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/python/pylibwholegraph/pyproject.toml -------------------------------------------------------------------------------- /readme_pages/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/readme_pages/CONTRIBUTING.md -------------------------------------------------------------------------------- /readme_pages/PRTAGS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/readme_pages/PRTAGS.md -------------------------------------------------------------------------------- /readme_pages/cugraph_pyg.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/readme_pages/cugraph_pyg.md -------------------------------------------------------------------------------- /readme_pages/gnn_support.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/readme_pages/gnn_support.md -------------------------------------------------------------------------------- /readme_pages/news.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/readme_pages/news.md -------------------------------------------------------------------------------- /readme_pages/performance/performance.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/readme_pages/performance/performance.md -------------------------------------------------------------------------------- /thirdparty/LICENSES/LICENSE.cugraph-ops.NVIDIA: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rapidsai/cugraph-gnn/HEAD/thirdparty/LICENSES/LICENSE.cugraph-ops.NVIDIA --------------------------------------------------------------------------------