PrincetonVision / marvin

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Move DropoutLayer random number generator to GPU

danielsuo opened this issue · comments

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).

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