├── .gitignore ├── .gitmodules ├── Docker ├── cu102 │ ├── build.sh │ ├── dockerfile │ └── launch.sh └── cu110 │ ├── build.sh │ ├── dockerfile │ └── launch.sh ├── GNNAdvisor ├── 0_bench_GNNA_GCN.py ├── 0_bench_GNNA_GIN.py ├── 0_run_gcn.sh ├── 0_run_gin.sh ├── 1_log2csv.py ├── 2_run_study.sh ├── 2_study2csv.py ├── 3_single_spmm_bench.py ├── 4_verifying.py ├── GNNA_main.py ├── GNNConv │ ├── GNNAdvisor.cpp │ ├── GNNAdvisor_kernel.cu │ ├── build.sh │ ├── clean.sh │ └── setup.py ├── dataset.py ├── gnn_conv.py ├── param.py ├── s7-4_1_neighbor_partitioning.py ├── s7-4_2_dimension_partitiong.py ├── s7-4_3_node_renumbering.py ├── s7-5_1_hidden_dimension.py └── unitest.py ├── Gunrock ├── 0_bench_Gunrock.py ├── 1_log2csv.py ├── CMakeLists.txt ├── app │ └── spmm │ │ ├── spmm_app.cu │ │ ├── spmm_enactor.cuh │ │ ├── spmm_problem.cuh │ │ └── spmm_test.cuh ├── bench_gunrock.csv ├── bench_gunrock.log ├── build_spmm.sh ├── examples │ └── spmm │ │ ├── CMakeLists.txt │ │ ├── Makefile │ │ ├── test.sh │ │ └── test_spmm.cu ├── install_cmake.sh └── spmm ├── README.md ├── clean_build.sh ├── dgl_baseline ├── 0_bench_dgl_gcn.py ├── 0_bench_dgl_gin.py ├── 0_run_gcn.sh ├── 0_run_gin.sh ├── 1_log2csv.py ├── dataset.py ├── dgl_main.py ├── gcn.py └── gin.py ├── docs ├── _config.yml └── index.md ├── images ├── dimensionWorker.png ├── hiddemDimension.png ├── install_pkg.png ├── neighborPartition.png ├── nodeRenumbering.png └── nodeReordering.png ├── install_pkg_sm61.sh ├── install_pkg_sm70.sh ├── install_pkg_sm86.sh ├── pyg_baseline ├── 0_bench_pyg_gcn.py ├── 0_bench_pyg_gin.py ├── 0_run_gcn.sh ├── 0_run_gin.sh ├── 1_log2csv.py ├── dataset.py └── pyg_main.py └── rabbit_module ├── rabbit_order.py └── src ├── Makefile ├── edge_list.hpp ├── rabbit_order.hpp ├── reorder ├── reorder.cc ├── reorder.cpp └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/.gitmodules -------------------------------------------------------------------------------- /Docker/cu102/build.sh: -------------------------------------------------------------------------------- 1 | docker build -t osdi-ae . -------------------------------------------------------------------------------- /Docker/cu102/dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Docker/cu102/dockerfile -------------------------------------------------------------------------------- /Docker/cu102/launch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Docker/cu102/launch.sh -------------------------------------------------------------------------------- /Docker/cu110/build.sh: -------------------------------------------------------------------------------- 1 | docker build -t osdi-ae . -------------------------------------------------------------------------------- /Docker/cu110/dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Docker/cu110/dockerfile -------------------------------------------------------------------------------- /Docker/cu110/launch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Docker/cu110/launch.sh -------------------------------------------------------------------------------- /GNNAdvisor/0_bench_GNNA_GCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/0_bench_GNNA_GCN.py -------------------------------------------------------------------------------- /GNNAdvisor/0_bench_GNNA_GIN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/0_bench_GNNA_GIN.py -------------------------------------------------------------------------------- /GNNAdvisor/0_run_gcn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/0_run_gcn.sh -------------------------------------------------------------------------------- /GNNAdvisor/0_run_gin.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/0_run_gin.sh -------------------------------------------------------------------------------- /GNNAdvisor/1_log2csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/1_log2csv.py -------------------------------------------------------------------------------- /GNNAdvisor/2_run_study.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/2_run_study.sh -------------------------------------------------------------------------------- /GNNAdvisor/2_study2csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/2_study2csv.py -------------------------------------------------------------------------------- /GNNAdvisor/3_single_spmm_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/3_single_spmm_bench.py -------------------------------------------------------------------------------- /GNNAdvisor/4_verifying.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/4_verifying.py -------------------------------------------------------------------------------- /GNNAdvisor/GNNA_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/GNNA_main.py -------------------------------------------------------------------------------- /GNNAdvisor/GNNConv/GNNAdvisor.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/GNNConv/GNNAdvisor.cpp -------------------------------------------------------------------------------- /GNNAdvisor/GNNConv/GNNAdvisor_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/GNNConv/GNNAdvisor_kernel.cu -------------------------------------------------------------------------------- /GNNAdvisor/GNNConv/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/GNNConv/build.sh -------------------------------------------------------------------------------- /GNNAdvisor/GNNConv/clean.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/GNNConv/clean.sh -------------------------------------------------------------------------------- /GNNAdvisor/GNNConv/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/GNNConv/setup.py -------------------------------------------------------------------------------- /GNNAdvisor/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/dataset.py -------------------------------------------------------------------------------- /GNNAdvisor/gnn_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/gnn_conv.py -------------------------------------------------------------------------------- /GNNAdvisor/param.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/param.py -------------------------------------------------------------------------------- /GNNAdvisor/s7-4_1_neighbor_partitioning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/s7-4_1_neighbor_partitioning.py -------------------------------------------------------------------------------- /GNNAdvisor/s7-4_2_dimension_partitiong.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/s7-4_2_dimension_partitiong.py -------------------------------------------------------------------------------- /GNNAdvisor/s7-4_3_node_renumbering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/s7-4_3_node_renumbering.py -------------------------------------------------------------------------------- /GNNAdvisor/s7-5_1_hidden_dimension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/s7-5_1_hidden_dimension.py -------------------------------------------------------------------------------- /GNNAdvisor/unitest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/GNNAdvisor/unitest.py -------------------------------------------------------------------------------- /Gunrock/0_bench_Gunrock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/0_bench_Gunrock.py -------------------------------------------------------------------------------- /Gunrock/1_log2csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/1_log2csv.py -------------------------------------------------------------------------------- /Gunrock/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/CMakeLists.txt -------------------------------------------------------------------------------- /Gunrock/app/spmm/spmm_app.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/app/spmm/spmm_app.cu -------------------------------------------------------------------------------- /Gunrock/app/spmm/spmm_enactor.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/app/spmm/spmm_enactor.cuh -------------------------------------------------------------------------------- /Gunrock/app/spmm/spmm_problem.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/app/spmm/spmm_problem.cuh -------------------------------------------------------------------------------- /Gunrock/app/spmm/spmm_test.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/app/spmm/spmm_test.cuh -------------------------------------------------------------------------------- /Gunrock/bench_gunrock.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/bench_gunrock.csv -------------------------------------------------------------------------------- /Gunrock/bench_gunrock.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/bench_gunrock.log -------------------------------------------------------------------------------- /Gunrock/build_spmm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/build_spmm.sh -------------------------------------------------------------------------------- /Gunrock/examples/spmm/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/examples/spmm/CMakeLists.txt -------------------------------------------------------------------------------- /Gunrock/examples/spmm/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/examples/spmm/Makefile -------------------------------------------------------------------------------- /Gunrock/examples/spmm/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/examples/spmm/test.sh -------------------------------------------------------------------------------- /Gunrock/examples/spmm/test_spmm.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/examples/spmm/test_spmm.cu -------------------------------------------------------------------------------- /Gunrock/install_cmake.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/install_cmake.sh -------------------------------------------------------------------------------- /Gunrock/spmm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/Gunrock/spmm -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/README.md -------------------------------------------------------------------------------- /clean_build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/clean_build.sh -------------------------------------------------------------------------------- /dgl_baseline/0_bench_dgl_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/0_bench_dgl_gcn.py -------------------------------------------------------------------------------- /dgl_baseline/0_bench_dgl_gin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/0_bench_dgl_gin.py -------------------------------------------------------------------------------- /dgl_baseline/0_run_gcn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/0_run_gcn.sh -------------------------------------------------------------------------------- /dgl_baseline/0_run_gin.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/0_run_gin.sh -------------------------------------------------------------------------------- /dgl_baseline/1_log2csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/1_log2csv.py -------------------------------------------------------------------------------- /dgl_baseline/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/dataset.py -------------------------------------------------------------------------------- /dgl_baseline/dgl_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/dgl_main.py -------------------------------------------------------------------------------- /dgl_baseline/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/gcn.py -------------------------------------------------------------------------------- /dgl_baseline/gin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/dgl_baseline/gin.py -------------------------------------------------------------------------------- /docs/_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/docs/_config.yml -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/docs/index.md -------------------------------------------------------------------------------- /images/dimensionWorker.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/images/dimensionWorker.png -------------------------------------------------------------------------------- /images/hiddemDimension.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/images/hiddemDimension.png -------------------------------------------------------------------------------- /images/install_pkg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/images/install_pkg.png -------------------------------------------------------------------------------- /images/neighborPartition.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/images/neighborPartition.png -------------------------------------------------------------------------------- /images/nodeRenumbering.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/images/nodeRenumbering.png -------------------------------------------------------------------------------- /images/nodeReordering.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/images/nodeReordering.png -------------------------------------------------------------------------------- /install_pkg_sm61.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/install_pkg_sm61.sh -------------------------------------------------------------------------------- /install_pkg_sm70.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/install_pkg_sm70.sh -------------------------------------------------------------------------------- /install_pkg_sm86.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/install_pkg_sm86.sh -------------------------------------------------------------------------------- /pyg_baseline/0_bench_pyg_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/pyg_baseline/0_bench_pyg_gcn.py -------------------------------------------------------------------------------- /pyg_baseline/0_bench_pyg_gin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/pyg_baseline/0_bench_pyg_gin.py -------------------------------------------------------------------------------- /pyg_baseline/0_run_gcn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/pyg_baseline/0_run_gcn.sh -------------------------------------------------------------------------------- /pyg_baseline/0_run_gin.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/pyg_baseline/0_run_gin.sh -------------------------------------------------------------------------------- /pyg_baseline/1_log2csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/pyg_baseline/1_log2csv.py -------------------------------------------------------------------------------- /pyg_baseline/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/pyg_baseline/dataset.py -------------------------------------------------------------------------------- /pyg_baseline/pyg_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/pyg_baseline/pyg_main.py -------------------------------------------------------------------------------- /rabbit_module/rabbit_order.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/rabbit_module/rabbit_order.py -------------------------------------------------------------------------------- /rabbit_module/src/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/rabbit_module/src/Makefile -------------------------------------------------------------------------------- /rabbit_module/src/edge_list.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/rabbit_module/src/edge_list.hpp -------------------------------------------------------------------------------- /rabbit_module/src/rabbit_order.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/rabbit_module/src/rabbit_order.hpp -------------------------------------------------------------------------------- /rabbit_module/src/reorder: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/rabbit_module/src/reorder -------------------------------------------------------------------------------- /rabbit_module/src/reorder.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/rabbit_module/src/reorder.cc -------------------------------------------------------------------------------- /rabbit_module/src/reorder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/rabbit_module/src/reorder.cpp -------------------------------------------------------------------------------- /rabbit_module/src/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukeWang96/GNNAdvisor_OSDI21/HEAD/rabbit_module/src/setup.py --------------------------------------------------------------------------------