dynamic-sparse-graph

December 7, 2018 ยท View on GitHub

This is an implementation of the paper "Dynamic Sparse Graph for Efficient Deep Learning" using CUDA.

Installation

  1. Edit Makefile so that NVCCFLAGS contains the correct value of --gpu-architecture and --gpu-code corresponding to your GPU. A list of the correct values for each generation can be found here: https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#gpu-feature-list.
  2. Run 'make'.

Usage

./mlp_driver

Contents

  1. 1Installation
  2. 2Usage