Commit Graph

34 Commits (27cd3efdd140bb156505b573aa222cfb690d169b)

Author SHA1 Message Date
Xin Pan 5eb87506bc add per kernel config and remove const_cast.
6 years ago
chengduo 46d01d798e
Revert "Revert "Remove workspace_handle in conv_cudnn (#15186)"" (#15290)
6 years ago
chengduozh 358e657f68 Revert "Remove workspace_handle in conv_cudnn (#15186)"
6 years ago
chengduo 064512aa47
Remove workspace_handle in conv_cudnn (#15186)
6 years ago
Wu Yi fd85418329
[Feature] support mix precision training for resnet (#14899)
6 years ago
minqiyang 53433d7f2e Revert the changes of VLOG
6 years ago
qingqing01 fd7e643153
Convolution fusion operator. (#14449)
6 years ago
Dang Qingqing d219818434 Fix compiling in cuDNN v5.
6 years ago
Qiyang Min 698698f2fa
Merge branch 'develop' into fix_vlog
6 years ago
qingqing01 abe209234f
Exhaustive search for cuDNN conv. (#14286)
6 years ago
minqiyang 0c3227a523 Change the origin VLOG level to 10 times
6 years ago
chengduo a9b5d42dd4
Add fp16 backward support (#14202)
6 years ago
qingqing01 db8c52da5e Revert " Exhaustive search for cuDNN conv. (#14043)"
6 years ago
qingqing01 ce7d9b0799
Exhaustive search for cuDNN conv. (#14043)
6 years ago
sneaxiy 5be6f762d0 remove_lock_in_some_ops
6 years ago
fengjiayi 82a1b35b9b Revert "Revert "Add CudnnHolder and use it in Conv and ConvTranspose op""
7 years ago
guochaorong 151e169eb7
Revert "Add CudnnHolder and use it in Conv and ConvTranspose op"
7 years ago
fengjiayi b0aca8824d make CudnnHolder thread safe
7 years ago
fengjiayi 407ff0bdbc use CudnnHolder in conv_cudnn_op
7 years ago
dzhwinter 4069262f0e
Revert ""cherry picked operators changes" (#12184)" (#12747)
7 years ago
dzhwinter bf3c34960f
"cherry picked operators changes" (#12184)
7 years ago
Yu Yang 040fc1c39b Fix bug in cudnn_determistic
7 years ago
Wu Yi 73fcfc06ec
refine conv cudnn enforce (#12353)
7 years ago
dzhwinter 7971d4a310
Feature/deterministic (#11205)
7 years ago
Kexin Zhao 8b16927230 add fp16 support to conv3d
7 years ago
chengduoZH c5774e3282 add FLAGS_use_deterministic_algo
7 years ago
Kexin Zhao 187ba08789 enable tensor core for conv cudnn
7 years ago
Kexin Zhao 8ebfc153dd update
7 years ago
Kexin Zhao e967d19b0a add more tests
7 years ago
Kexin Zhao a13ec3432a fix test error
7 years ago
Kexin Zhao e4de5dc347 add conv2d fp16 support
7 years ago
qingqing01 24509f4af9 Fix the grammar in copyright. (#8403)
7 years ago
Yi Wang fc374821dd Correct #include path
7 years ago
Yi Wang 90648f336d Move file to fluid/; Edit CMakeLists.txt
7 years ago