Move DropoutLayer random number generator to GPU

PrincetonVision / marvin

Marvin: A Minimalist GPU-only N-Dimensional ConvNets Framework

MIT License

422 stars 137 forks source link

Move DropoutLayer random number generator to GPU #7

Closed danielsuo closed 8 years ago

danielsuo commented 8 years ago

Currently we rely on the CPU for random numbers. To make this slightly less painfully slow, we create a new thread, but this is slower than using the GPU and also causes problems in Windows (#4).

vinjn commented 8 years ago

cuRAND might be a good candidate. https://developer.nvidia.com/curand