├── .clang-format ├── .gitignore ├── .gitmodules ├── CMakeLists.txt ├── LICENSE ├── README.md ├── cfgs ├── commnet_cora.cfg ├── gat_cora.cfg ├── gcn_citeseer.cfg ├── gcn_cora.cfg ├── gcn_cora_cluster.cfg ├── gcn_cora_sample.cfg ├── gcn_pubmed.cfg ├── gcn_reddit.cfg ├── gcn_reddit_full.cfg ├── gcn_sample.cfg ├── gcn_sample_cache.cfg ├── gcn_sample_demo.cfg ├── gcn_sample_gpu.cfg ├── gcn_sample_gpu_arxiv.cfg ├── gcn_sample_gpu_arxiv_cache.cfg ├── gcn_sample_gpu_arxiv_trans.cfg ├── gin_cora.cfg └── test_batch_dist.cfg ├── comm ├── logger.h ├── network.cpp └── network.h ├── copy_all.sh ├── core ├── FullyRepGraph.hpp ├── GraphSegment.cpp ├── GraphSegment.h ├── MetisPartition.hpp ├── NtsEdgeTensor.hpp ├── NtsScheduler.hpp ├── PartitionedGraph.hpp ├── coocsc.hpp ├── graph.hpp ├── neutronstar.hpp ├── ntsBaseOp.hpp ├── ntsCPUFusedGraphOp.hpp ├── ntsContext.hpp ├── ntsDataloador.hpp ├── ntsDistCPUGraphOp.hpp ├── ntsDistGPUFusedGraphOp.hpp ├── ntsDistGPUGraphOp.hpp ├── ntsGraphOp.hpp ├── ntsMiniBatchGraphOp.hpp ├── ntsPeerRPC.hpp ├── ntsSampler.hpp ├── ntsSingleCPUGraphOp.hpp ├── ntsSingleGPUFusedGraphOp.hpp └── ntsSubLinearNNOP.hpp ├── cuda ├── CMakeLists.txt ├── cuda_type.h ├── graphs │ ├── common.h │ └── csr_graph.h ├── ntsCUDA.hpp ├── ntsCUDADistKernel.cuh ├── ntsCUDAFuseKernel.cuh ├── ntsCUDAGraphOP.cu └── ntsCUDATransferKernel.cuh ├── data ├── README.md ├── generate_nts_dataset.py └── requirements.txt ├── dep └── gemini │ ├── atomic.hpp │ ├── bitmap.hpp │ ├── constants.hpp │ ├── filesystem.hpp │ ├── math.hpp │ ├── mpi.hpp │ ├── time.hpp │ └── type.hpp ├── exp ├── Partition │ ├── README.md │ ├── partition │ │ ├── README.md │ │ ├── bytegnn_partition.cpp │ │ ├── bytegnn_partition.py │ │ ├── bytegnn_partition.sh │ │ ├── dgl_partition.py │ │ ├── draw.py │ │ ├── exp-time-cost.py │ │ ├── hash_partition.py │ │ ├── main.py │ │ ├── metis_partition.py │ │ ├── myPartition.py │ │ ├── pagraph_partition.py │ │ ├── partition_graph.py │ │ ├── run-exp.py │ │ ├── train_dist.py │ │ └── utils.py │ └── requirements.txt ├── README.md ├── cpu-gpu-usage.py ├── draw-gpu-cache.py ├── draw.py ├── exp-batch-size │ ├── draw-bs-fig10.py │ ├── draw-mix-fig11.py │ ├── exp-batch-size.py │ ├── ogbn-products-mix.cfg │ ├── reddit-mix.cfg │ └── run_nts.sh ├── exp-batch-type │ └── draw-batch-type-fig12.py ├── exp-diff-optim │ ├── draw-diff-optim-fig14.py │ ├── draw-explicit.py │ ├── draw-unified-zero.py │ └── exp-diff-optim.py ├── exp-fanout │ ├── draw-fanout-fig13a.py │ └── draw-sample-fig13b.py ├── exp-gpu-cache │ ├── draw-cache-fig18.py │ └── exp-gpu-cache.py ├── exp-hybrid-trans │ ├── draw-hybrid-trans-fig17.py │ └── exp-hybrid-trans.py ├── exp-partition │ ├── README.md │ ├── draw-partition-time-fig6.py │ └── exp-partition.py ├── exp-pipeline │ ├── cpu-gpu-usage.py │ ├── draw-pipe-ablasion.py │ ├── draw-usage.py │ ├── exp-diff-pipe.py │ └── exp-usage.py ├── exp-pipeline_motivation.py ├── exp-sample-rate │ ├── draw-sample-fig13b.py │ └── sample-rate.py ├── exp2-batch-size.py ├── exp3-gpu-cache.py ├── exp3-gpu-cache.sh ├── exp4-hybrid-trans.py ├── hostfile ├── utils.py └── utils.py.rm ├── format ├── clang_format_exclusions.txt ├── format-py.sh ├── format.sh ├── requirements.txt └── run_clang_format.py ├── kill.sh ├── logo ├── NeutronBench.png └── logo.png ├── make_and_copy.sh ├── run.sh ├── run_nts.sh ├── run_nts_dist.sh ├── test ├── test.cfg ├── testcsr.cpp └── testsample.cpp ├── toolkits ├── CMakeLists.txt ├── COMMNET_GPU.hpp ├── GAT_CPU.hpp ├── GAT_CPU_DIST.hpp ├── GAT_CPU_DIST_OPTM.hpp ├── GAT_GPU_DIST.hpp ├── GCN.hpp ├── GCN_ASYNC_GPU_sample.hpp ├── GCN_CPU.hpp ├── GCN_CPU_CLUSTER.hpp ├── GCN_CPU_EAGER.hpp ├── GCN_CPU_GRAPH.hpp ├── GCN_CPU_LAYER.hpp ├── GCN_CPU_NEIGHBOR.hpp ├── GCN_EAGER.hpp ├── GCN_EAGER_single.hpp ├── GCN_GPU_NEIGHBOR-bak.hpp ├── GCN_GPU_NEIGHBOR-cache-dgl-exp.hpp ├── GCN_GPU_NEIGHBOR-cache-exp.hpp ├── GCN_GPU_NEIGHBOR-exp3.hpp ├── GCN_GPU_NEIGHBOR-partition.hpp ├── GCN_GPU_NEIGHBOR-sample-rate.hpp ├── GCN_GPU_NEIGHBOR-trans-exp.hpp ├── GCN_GPU_NEIGHBOR.hpp ├── GGCN_CPU.hpp ├── GIN_CPU.hpp ├── GIN_GPU.hpp ├── main.cpp ├── old.cpp ├── test_batch_distributed.hpp ├── test_getdepneighbor_cpu.hpp └── test_getdepneighbor_gpu.hpp └── utils ├── cuda_memory.hpp ├── rand.hpp ├── torch_func.hpp └── utils.hpp /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/.clang-format -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/.gitmodules -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/commnet_cora.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/commnet_cora.cfg -------------------------------------------------------------------------------- /cfgs/gat_cora.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gat_cora.cfg -------------------------------------------------------------------------------- /cfgs/gcn_citeseer.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_citeseer.cfg -------------------------------------------------------------------------------- /cfgs/gcn_cora.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_cora.cfg -------------------------------------------------------------------------------- /cfgs/gcn_cora_cluster.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_cora_cluster.cfg -------------------------------------------------------------------------------- /cfgs/gcn_cora_sample.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_cora_sample.cfg -------------------------------------------------------------------------------- /cfgs/gcn_pubmed.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_pubmed.cfg -------------------------------------------------------------------------------- /cfgs/gcn_reddit.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_reddit.cfg -------------------------------------------------------------------------------- /cfgs/gcn_reddit_full.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_reddit_full.cfg -------------------------------------------------------------------------------- /cfgs/gcn_sample.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_sample.cfg -------------------------------------------------------------------------------- /cfgs/gcn_sample_cache.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_sample_cache.cfg -------------------------------------------------------------------------------- /cfgs/gcn_sample_demo.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_sample_demo.cfg -------------------------------------------------------------------------------- /cfgs/gcn_sample_gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_sample_gpu.cfg -------------------------------------------------------------------------------- /cfgs/gcn_sample_gpu_arxiv.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_sample_gpu_arxiv.cfg -------------------------------------------------------------------------------- /cfgs/gcn_sample_gpu_arxiv_cache.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_sample_gpu_arxiv_cache.cfg -------------------------------------------------------------------------------- /cfgs/gcn_sample_gpu_arxiv_trans.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gcn_sample_gpu_arxiv_trans.cfg -------------------------------------------------------------------------------- /cfgs/gin_cora.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/gin_cora.cfg -------------------------------------------------------------------------------- /cfgs/test_batch_dist.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cfgs/test_batch_dist.cfg -------------------------------------------------------------------------------- /comm/logger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/comm/logger.h -------------------------------------------------------------------------------- /comm/network.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/comm/network.cpp -------------------------------------------------------------------------------- /comm/network.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/comm/network.h -------------------------------------------------------------------------------- /copy_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/copy_all.sh -------------------------------------------------------------------------------- /core/FullyRepGraph.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/FullyRepGraph.hpp -------------------------------------------------------------------------------- /core/GraphSegment.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/GraphSegment.cpp -------------------------------------------------------------------------------- /core/GraphSegment.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/GraphSegment.h -------------------------------------------------------------------------------- /core/MetisPartition.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/MetisPartition.hpp -------------------------------------------------------------------------------- /core/NtsEdgeTensor.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/NtsEdgeTensor.hpp -------------------------------------------------------------------------------- /core/NtsScheduler.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/NtsScheduler.hpp -------------------------------------------------------------------------------- /core/PartitionedGraph.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/PartitionedGraph.hpp -------------------------------------------------------------------------------- /core/coocsc.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/coocsc.hpp -------------------------------------------------------------------------------- /core/graph.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/graph.hpp -------------------------------------------------------------------------------- /core/neutronstar.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/neutronstar.hpp -------------------------------------------------------------------------------- /core/ntsBaseOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsBaseOp.hpp -------------------------------------------------------------------------------- /core/ntsCPUFusedGraphOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsCPUFusedGraphOp.hpp -------------------------------------------------------------------------------- /core/ntsContext.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsContext.hpp -------------------------------------------------------------------------------- /core/ntsDataloador.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsDataloador.hpp -------------------------------------------------------------------------------- /core/ntsDistCPUGraphOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsDistCPUGraphOp.hpp -------------------------------------------------------------------------------- /core/ntsDistGPUFusedGraphOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsDistGPUFusedGraphOp.hpp -------------------------------------------------------------------------------- /core/ntsDistGPUGraphOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsDistGPUGraphOp.hpp -------------------------------------------------------------------------------- /core/ntsGraphOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsGraphOp.hpp -------------------------------------------------------------------------------- /core/ntsMiniBatchGraphOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsMiniBatchGraphOp.hpp -------------------------------------------------------------------------------- /core/ntsPeerRPC.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsPeerRPC.hpp -------------------------------------------------------------------------------- /core/ntsSampler.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsSampler.hpp -------------------------------------------------------------------------------- /core/ntsSingleCPUGraphOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsSingleCPUGraphOp.hpp -------------------------------------------------------------------------------- /core/ntsSingleGPUFusedGraphOp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsSingleGPUFusedGraphOp.hpp -------------------------------------------------------------------------------- /core/ntsSubLinearNNOP.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/core/ntsSubLinearNNOP.hpp -------------------------------------------------------------------------------- /cuda/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/CMakeLists.txt -------------------------------------------------------------------------------- /cuda/cuda_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/cuda_type.h -------------------------------------------------------------------------------- /cuda/graphs/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/graphs/common.h -------------------------------------------------------------------------------- /cuda/graphs/csr_graph.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/graphs/csr_graph.h -------------------------------------------------------------------------------- /cuda/ntsCUDA.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/ntsCUDA.hpp -------------------------------------------------------------------------------- /cuda/ntsCUDADistKernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/ntsCUDADistKernel.cuh -------------------------------------------------------------------------------- /cuda/ntsCUDAFuseKernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/ntsCUDAFuseKernel.cuh -------------------------------------------------------------------------------- /cuda/ntsCUDAGraphOP.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/ntsCUDAGraphOP.cu -------------------------------------------------------------------------------- /cuda/ntsCUDATransferKernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/cuda/ntsCUDATransferKernel.cuh -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/data/README.md -------------------------------------------------------------------------------- /data/generate_nts_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/data/generate_nts_dataset.py -------------------------------------------------------------------------------- /data/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/data/requirements.txt -------------------------------------------------------------------------------- /dep/gemini/atomic.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/dep/gemini/atomic.hpp -------------------------------------------------------------------------------- /dep/gemini/bitmap.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/dep/gemini/bitmap.hpp -------------------------------------------------------------------------------- /dep/gemini/constants.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/dep/gemini/constants.hpp -------------------------------------------------------------------------------- /dep/gemini/filesystem.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/dep/gemini/filesystem.hpp -------------------------------------------------------------------------------- /dep/gemini/math.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/dep/gemini/math.hpp -------------------------------------------------------------------------------- /dep/gemini/mpi.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/dep/gemini/mpi.hpp -------------------------------------------------------------------------------- /dep/gemini/time.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/dep/gemini/time.hpp -------------------------------------------------------------------------------- /dep/gemini/type.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/dep/gemini/type.hpp -------------------------------------------------------------------------------- /exp/Partition/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/README.md -------------------------------------------------------------------------------- /exp/Partition/partition/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/README.md -------------------------------------------------------------------------------- /exp/Partition/partition/bytegnn_partition.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/bytegnn_partition.cpp -------------------------------------------------------------------------------- /exp/Partition/partition/bytegnn_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/bytegnn_partition.py -------------------------------------------------------------------------------- /exp/Partition/partition/bytegnn_partition.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/bytegnn_partition.sh -------------------------------------------------------------------------------- /exp/Partition/partition/dgl_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/dgl_partition.py -------------------------------------------------------------------------------- /exp/Partition/partition/draw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/draw.py -------------------------------------------------------------------------------- /exp/Partition/partition/exp-time-cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/exp-time-cost.py -------------------------------------------------------------------------------- /exp/Partition/partition/hash_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/hash_partition.py -------------------------------------------------------------------------------- /exp/Partition/partition/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/main.py -------------------------------------------------------------------------------- /exp/Partition/partition/metis_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/metis_partition.py -------------------------------------------------------------------------------- /exp/Partition/partition/myPartition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/myPartition.py -------------------------------------------------------------------------------- /exp/Partition/partition/pagraph_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/pagraph_partition.py -------------------------------------------------------------------------------- /exp/Partition/partition/partition_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/partition_graph.py -------------------------------------------------------------------------------- /exp/Partition/partition/run-exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/run-exp.py -------------------------------------------------------------------------------- /exp/Partition/partition/train_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/train_dist.py -------------------------------------------------------------------------------- /exp/Partition/partition/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/partition/utils.py -------------------------------------------------------------------------------- /exp/Partition/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/Partition/requirements.txt -------------------------------------------------------------------------------- /exp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/README.md -------------------------------------------------------------------------------- /exp/cpu-gpu-usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/cpu-gpu-usage.py -------------------------------------------------------------------------------- /exp/draw-gpu-cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/draw-gpu-cache.py -------------------------------------------------------------------------------- /exp/draw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/draw.py -------------------------------------------------------------------------------- /exp/exp-batch-size/draw-bs-fig10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-batch-size/draw-bs-fig10.py -------------------------------------------------------------------------------- /exp/exp-batch-size/draw-mix-fig11.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-batch-size/draw-mix-fig11.py -------------------------------------------------------------------------------- /exp/exp-batch-size/exp-batch-size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-batch-size/exp-batch-size.py -------------------------------------------------------------------------------- /exp/exp-batch-size/ogbn-products-mix.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-batch-size/ogbn-products-mix.cfg -------------------------------------------------------------------------------- /exp/exp-batch-size/reddit-mix.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-batch-size/reddit-mix.cfg -------------------------------------------------------------------------------- /exp/exp-batch-size/run_nts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-batch-size/run_nts.sh -------------------------------------------------------------------------------- /exp/exp-batch-type/draw-batch-type-fig12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-batch-type/draw-batch-type-fig12.py -------------------------------------------------------------------------------- /exp/exp-diff-optim/draw-diff-optim-fig14.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-diff-optim/draw-diff-optim-fig14.py -------------------------------------------------------------------------------- /exp/exp-diff-optim/draw-explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-diff-optim/draw-explicit.py -------------------------------------------------------------------------------- /exp/exp-diff-optim/draw-unified-zero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-diff-optim/draw-unified-zero.py -------------------------------------------------------------------------------- /exp/exp-diff-optim/exp-diff-optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-diff-optim/exp-diff-optim.py -------------------------------------------------------------------------------- /exp/exp-fanout/draw-fanout-fig13a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-fanout/draw-fanout-fig13a.py -------------------------------------------------------------------------------- /exp/exp-fanout/draw-sample-fig13b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-fanout/draw-sample-fig13b.py -------------------------------------------------------------------------------- /exp/exp-gpu-cache/draw-cache-fig18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-gpu-cache/draw-cache-fig18.py -------------------------------------------------------------------------------- /exp/exp-gpu-cache/exp-gpu-cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-gpu-cache/exp-gpu-cache.py -------------------------------------------------------------------------------- /exp/exp-hybrid-trans/draw-hybrid-trans-fig17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-hybrid-trans/draw-hybrid-trans-fig17.py -------------------------------------------------------------------------------- /exp/exp-hybrid-trans/exp-hybrid-trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-hybrid-trans/exp-hybrid-trans.py -------------------------------------------------------------------------------- /exp/exp-partition/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-partition/README.md -------------------------------------------------------------------------------- /exp/exp-partition/draw-partition-time-fig6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-partition/draw-partition-time-fig6.py -------------------------------------------------------------------------------- /exp/exp-partition/exp-partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-partition/exp-partition.py -------------------------------------------------------------------------------- /exp/exp-pipeline/cpu-gpu-usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-pipeline/cpu-gpu-usage.py -------------------------------------------------------------------------------- /exp/exp-pipeline/draw-pipe-ablasion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-pipeline/draw-pipe-ablasion.py -------------------------------------------------------------------------------- /exp/exp-pipeline/draw-usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-pipeline/draw-usage.py -------------------------------------------------------------------------------- /exp/exp-pipeline/exp-diff-pipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-pipeline/exp-diff-pipe.py -------------------------------------------------------------------------------- /exp/exp-pipeline/exp-usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-pipeline/exp-usage.py -------------------------------------------------------------------------------- /exp/exp-pipeline_motivation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-pipeline_motivation.py -------------------------------------------------------------------------------- /exp/exp-sample-rate/draw-sample-fig13b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-sample-rate/draw-sample-fig13b.py -------------------------------------------------------------------------------- /exp/exp-sample-rate/sample-rate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp-sample-rate/sample-rate.py -------------------------------------------------------------------------------- /exp/exp2-batch-size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp2-batch-size.py -------------------------------------------------------------------------------- /exp/exp3-gpu-cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp3-gpu-cache.py -------------------------------------------------------------------------------- /exp/exp3-gpu-cache.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp3-gpu-cache.sh -------------------------------------------------------------------------------- /exp/exp4-hybrid-trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/exp4-hybrid-trans.py -------------------------------------------------------------------------------- /exp/hostfile: -------------------------------------------------------------------------------- 1 | localhost 2 | -------------------------------------------------------------------------------- /exp/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/utils.py -------------------------------------------------------------------------------- /exp/utils.py.rm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/exp/utils.py.rm -------------------------------------------------------------------------------- /format/clang_format_exclusions.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /format/format-py.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/format/format-py.sh -------------------------------------------------------------------------------- /format/format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/format/format.sh -------------------------------------------------------------------------------- /format/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/format/requirements.txt -------------------------------------------------------------------------------- /format/run_clang_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/format/run_clang_format.py -------------------------------------------------------------------------------- /kill.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/kill.sh -------------------------------------------------------------------------------- /logo/NeutronBench.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/logo/NeutronBench.png -------------------------------------------------------------------------------- /logo/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/logo/logo.png -------------------------------------------------------------------------------- /make_and_copy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/make_and_copy.sh -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/run.sh -------------------------------------------------------------------------------- /run_nts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/run_nts.sh -------------------------------------------------------------------------------- /run_nts_dist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/run_nts_dist.sh -------------------------------------------------------------------------------- /test/test.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/test/test.cfg -------------------------------------------------------------------------------- /test/testcsr.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/test/testcsr.cpp -------------------------------------------------------------------------------- /test/testsample.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/test/testsample.cpp -------------------------------------------------------------------------------- /toolkits/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/CMakeLists.txt -------------------------------------------------------------------------------- /toolkits/COMMNET_GPU.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/COMMNET_GPU.hpp -------------------------------------------------------------------------------- /toolkits/GAT_CPU.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GAT_CPU.hpp -------------------------------------------------------------------------------- /toolkits/GAT_CPU_DIST.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GAT_CPU_DIST.hpp -------------------------------------------------------------------------------- /toolkits/GAT_CPU_DIST_OPTM.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GAT_CPU_DIST_OPTM.hpp -------------------------------------------------------------------------------- /toolkits/GAT_GPU_DIST.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GAT_GPU_DIST.hpp -------------------------------------------------------------------------------- /toolkits/GCN.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN.hpp -------------------------------------------------------------------------------- /toolkits/GCN_ASYNC_GPU_sample.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_ASYNC_GPU_sample.hpp -------------------------------------------------------------------------------- /toolkits/GCN_CPU.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_CPU.hpp -------------------------------------------------------------------------------- /toolkits/GCN_CPU_CLUSTER.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_CPU_CLUSTER.hpp -------------------------------------------------------------------------------- /toolkits/GCN_CPU_EAGER.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_CPU_EAGER.hpp -------------------------------------------------------------------------------- /toolkits/GCN_CPU_GRAPH.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_CPU_GRAPH.hpp -------------------------------------------------------------------------------- /toolkits/GCN_CPU_LAYER.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_CPU_LAYER.hpp -------------------------------------------------------------------------------- /toolkits/GCN_CPU_NEIGHBOR.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_CPU_NEIGHBOR.hpp -------------------------------------------------------------------------------- /toolkits/GCN_EAGER.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_EAGER.hpp -------------------------------------------------------------------------------- /toolkits/GCN_EAGER_single.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_EAGER_single.hpp -------------------------------------------------------------------------------- /toolkits/GCN_GPU_NEIGHBOR-bak.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_GPU_NEIGHBOR-bak.hpp -------------------------------------------------------------------------------- /toolkits/GCN_GPU_NEIGHBOR-cache-dgl-exp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_GPU_NEIGHBOR-cache-dgl-exp.hpp -------------------------------------------------------------------------------- /toolkits/GCN_GPU_NEIGHBOR-cache-exp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_GPU_NEIGHBOR-cache-exp.hpp -------------------------------------------------------------------------------- /toolkits/GCN_GPU_NEIGHBOR-exp3.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_GPU_NEIGHBOR-exp3.hpp -------------------------------------------------------------------------------- /toolkits/GCN_GPU_NEIGHBOR-partition.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_GPU_NEIGHBOR-partition.hpp -------------------------------------------------------------------------------- /toolkits/GCN_GPU_NEIGHBOR-sample-rate.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_GPU_NEIGHBOR-sample-rate.hpp -------------------------------------------------------------------------------- /toolkits/GCN_GPU_NEIGHBOR-trans-exp.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_GPU_NEIGHBOR-trans-exp.hpp -------------------------------------------------------------------------------- /toolkits/GCN_GPU_NEIGHBOR.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GCN_GPU_NEIGHBOR.hpp -------------------------------------------------------------------------------- /toolkits/GGCN_CPU.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GGCN_CPU.hpp -------------------------------------------------------------------------------- /toolkits/GIN_CPU.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GIN_CPU.hpp -------------------------------------------------------------------------------- /toolkits/GIN_GPU.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/GIN_GPU.hpp -------------------------------------------------------------------------------- /toolkits/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/main.cpp -------------------------------------------------------------------------------- /toolkits/old.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/old.cpp -------------------------------------------------------------------------------- /toolkits/test_batch_distributed.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/test_batch_distributed.hpp -------------------------------------------------------------------------------- /toolkits/test_getdepneighbor_cpu.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/test_getdepneighbor_cpu.hpp -------------------------------------------------------------------------------- /toolkits/test_getdepneighbor_gpu.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/toolkits/test_getdepneighbor_gpu.hpp -------------------------------------------------------------------------------- /utils/cuda_memory.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/utils/cuda_memory.hpp -------------------------------------------------------------------------------- /utils/rand.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/utils/rand.hpp -------------------------------------------------------------------------------- /utils/torch_func.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/utils/torch_func.hpp -------------------------------------------------------------------------------- /utils/utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iDC-NEU/NeutronBench/HEAD/utils/utils.hpp --------------------------------------------------------------------------------