search

PrincetonVision / marvin

Marvin: A Minimalist GPU-only N-Dimensional ConvNets Framework
MIT License
422 stars 137 forks source link

cudnnBatchNormalization FP16 in cudnn rc5 #37

Open hengck23 opened 8 years ago

hengck23 commented 8 years ago

it is mentioned that cudnnBatchNormalization FP16 cannot work in cudnn v4. How about cudnn rc5? I have an error "CUDNN_STATUS_BAD_PARAM". https://github.com/PrincetonVision/marvin/commit/9861e29ea6c924095d88ad0ace7870bc4a9d0134