Skip to content

5D Conv & Pooling Primitives in Cuda

License

Notifications You must be signed in to change notification settings

ViktorYastrebov/kuDNN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

kudnn

kuDNN exposes Convolution/Pooling Forward and Backward primitives for 5D data (3D images) which was not available at the time (cuDNN 2). It has the same interface with NVIDIA's [cuDNN] (https://developer.nvidia.com/cudnn). Currently in use by [KNet.jl] (https://github.com/denizyuret/Knet.jl)

To create yourself a shared object file: cd src; make libkudnn.so

About

5D Conv & Pooling Primitives in Cuda

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Cuda 81.3%
  • Julia 12.0%
  • C 6.2%
  • Makefile 0.5%