提交 d740f88e authored 作者: --global's avatar --global

Change name of flags to control CuDNN convolution implementation

上级 2389c490
...@@ -340,15 +340,26 @@ class GpuDnnConvDesc(GpuOp): ...@@ -340,15 +340,26 @@ class GpuDnnConvDesc(GpuOp):
AddConfigVar('dnn.conv.workmem', AddConfigVar('dnn.conv.workmem',
"Default value for the workmem attribute of cudnn convolutions.", "This flag is deprecated; use dnn.conv.algo_fwd.",
EnumStr('small', 'none', 'large', 'fft', 'guess', EnumStr(''),
'guess_once', 'time', 'time_once'),
in_c_key=False) in_c_key=False)
AddConfigVar('dnn.conv.workmem_bwd', AddConfigVar('dnn.conv.workmem_bwd',
"Default value for the workmem attribute of cudnn gradient " "This flag is deprecated; use dnn.conv.algo_bwd.",
"convolutions.", EnumStr(''),
EnumStr('none', 'deterministic', 'fft', 'guess', 'guess_once'), in_c_key=False)
AddConfigVar('dnn.conv.algo_fwd',
"Default implementation to use for CuDNN forward convolution.",
EnumStr('small', 'none', 'large', 'fft', 'guess_once',
'guess_on_shape_change', 'time_once',
'time_on_shape_change'),
in_c_key=False)
AddConfigVar('dnn.conv.algo_bwd',
"Default implementation to use for CuDNN backward convolution.",
EnumStr('none', 'deterministic', 'fft', 'guess_once',
'guess_on_shape_change'),
in_c_key=False) in_c_key=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论