Skip to content

wangsiping97/ParallelGRU

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

48 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ParallelGRU

A parallel version of GRU (Gated Recurrent Unit) model using CUDA, by Siping Wang and Junyan Pu. Final project for CMU 15-618.

Project Page

https://wangsiping97.github.io/15618/

Usage

Note:

  1. The NVIDIA CUDA C/C++ Compiler (NVCC) needs to be added to PATH.
  2. The CUDA shared library must be loaded at runtime.

Training

$ cd training
$ make
$ ./cudaGRU -g <1 for GPU or 0 for CPU> -i <number of iterations>

Inference

$ cd inference
$ make
$ ./cudaInference -g <1 for GPU or 0 for CPU>

About

A parallel version of GRU (Gated Recurrent Unit) model using CUDA, by Siping Wang and Junyan Pu. Final project for CMU 15-618.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors