vuanvin / vdnn-plus-plus

Implementation of vDNN++; an improvement over vDNN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

vdnn-plus-plus

vDNN++ is an improvement in design over vDNN (https://arxiv.org/abs/1602.08124). This repository contain my implementation of vDNN and vDNN++. Currently, this supports linear networks.

vDNN++ removes synchronization at the end of computation of each layer. It support different heuristics and reduces memory fragmentation. It demonstrates the feasibility of compression on CPU-side for reducing pinned memory usage.

About

Implementation of vDNN++; an improvement over vDNN


Languages

Language:Cuda 62.9%Language:C++ 31.6%Language:C 3.9%Language:Shell 0.8%Language:CMake 0.7%Language:Python 0.0%