-
Notifications
You must be signed in to change notification settings - Fork 5.7k
添加paddle在cudnn v6环境下cudnn convolution对double类型的支持 #954
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Comments
@hedaoyuan Is this issue already fixed? |
从PR内容看起来主要是,修改 如果只是修改convolution支持double的话,修改一下ISSUE的描述,当前标题跟内容不是特别相符,我昨天看到PR的标题的时候一开始也没看明白准备写什么。对cudnn v6的其他特性支持可以另外新建一个ISSUE来描述。 |
另外,我看你的ISSUE描述应该是测试了Tesla P4,CUDNN V6等,如果有一些测试对比的性能数据的话也可以新建ISSUE贴上来。 |
@ccx0912 cudnn v6是在哪出现的?有链接么 |
* update api_cn; fix the error of reverting layers_cn to 1.4; include previous changes * update layers_cn; add code example to elementwise_mod * delete recordio_writer_cn; revert index_cn * update api_cn after review * fix io_cn.rst * update layers_cn * update io_cn
Uh oh!
There was an error while loading. Please reload this page.
在测试paddle的在nvidia tesla P4上运行时,为了尽可能的发挥Pascal架构的优势,需要使用到cudnn v6和cuda 8.0。检查cudnn v5与v6的区别,发现在
paddle/cuda/src/hl_cuda_cudnn.cc
文件中使用的cudnn函数,只有函数cudnnSetConvolution2dDescriptor
有所修改,cudnn v6后cudnn convolution可以支持double类型,所以函数的输入需要添加一个变量cudnnDataType_t
变量来指定计算的数据类型,当前的PR #949 中也仅修改了这个函数,而对于cudnn v6中的其他特征并未考虑,如CUDNN_DATA_INT8
等);cudnnReduceTensor
等);The text was updated successfully, but these errors were encountered: