智慧物流>> cuda_gcn>> 返回
项目作者： hengdashi

项目描述 ：
GPU implementation of GCN (Graph Convolutional Networks)
高级语言： Cuda
项目地址: git://github.com/hengdashi/cuda_gcn.git
创建时间： 2019-11-20T20:24:26Z
项目社区：https://github.com/hengdashi/cuda_gcn
开源协议：
下载


Parallelization of Graph Convolutional Network

This is the class project repo for CS 259 (High Performance Computing) Fall 2019 taught by Glenn Reinman.
The sequential version of the code comes from here. Reddit dataset can be downloaded here.
This project is a GPU acceleration of the Graph Convolutional Network.

Team members are: Zongze Li, Yuanhao Jia, Hengda Shi, Jintao Jiang

Setup

Convert reddit data format (networkx, numpy, scipy are required)

  1. python3 reddit_preprocess.py

Run cora/citeseer/pubmed/reddit datasets on CPU

  1. make seq
  2. ./gcn-seq cora

Run cora/citeseer/pubmed/reddit datasets on GPU

  1. make cuda
  2. ./gcn-cuda cora

Clean up executable

  1. make clean
report_1649116215280.pdf